java - 如何模拟 DataSourceTransactionManager ？

Question

我有以下我想模拟的代码。我基本上对模拟 DataSourceTransactionManager 感兴趣。

    @Autowired
    @Qualifier("nesTransactionManager")
    DataSourceTransactionManager mDataSourceTransactionManager;

   ...............................

    DefaultTransactionDefinition def = new DefaultTransactionDefinition();
            def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRES_NEW);
       TransactionStatus status = mDataSourceTransactionManager.getTransaction(def);

       try {
          <-doing some DB operations here>
          mDataSourceTransactionManager.commit(status);
       } catch (Exception e) {
          mDataSourceTransactionManager.rollback(status);
       }
.............

到目前为止我已经尝试过:

    @Mock
    private DataSourceTransactionManager mDataSourceTransactionManager;

   @Before
    public void runBeforeEachTest() {
        when(mDataSourceTransactionManager.getTransaction(any(DefaultTransactionDefinition.class))).thenReturn(null);
        doNothing().when(mDataSourceTransactionManager).commit(any(TransactionStatus.class));
        doNothing().when(mDataSourceTransactionManager).rollback(any(TransactionStatus.class));
}

还有这个:

    @Mock
    private DataSourceTransactionManager mDataSourceTransactionManager;

   @Before
    public void runBeforeEachTest() {
DefaultTransactionDefinition def = new DefaultTransactionDefinition();
        def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRES_NEW);

        when(mDataSourceTransactionManager.getTransaction(eq(def))).thenReturn(null);
        doNothing().when(mDataSourceTransactionManager).commit(any(TransactionStatus.class));
        doNothing().when(mDataSourceTransactionManager).rollback(any(TransactionStatus.class));

但我总是遇到这个错误:

org.mockito.exceptions.misusing.InvalidUseOfMatchersException: 
Invalid use of argument matchers!
0 matchers expected, 1 recorded:
-> at com.nuance.entrd.mc.nes.jobs.JobPollerTest.runBeforeEachTest(JobPollerTest.java:43)

This exception may occur if matchers are combined with raw values:
    //incorrect:
    someMethod(anyObject(), "raw String");
When using matchers, all arguments have to be provided by matchers.
For example:
    //correct:
    someMethod(anyObject(), eq("String by matcher"));

For more info see javadoc for Matchers class.

    at org.springframework.transaction.support.AbstractPlatformTransactionManager.getTransaction(AbstractPlatformTransactionManager.java:337)

在此上下文中没有任何意义的错误。知道如何解决这个问题吗？

谢谢

Answer 1

我解决问题的方法是使用 PowerMokito :

@RunWith(PowerMockRunner.class)
@PrepareForTest({DataSourceTransactionManager.class})
public class JobPollerTest {
.....................
 private DataSourceTransactionManager mDataSourceTransactionManager;
......................

    mDataSourceTransactionManager = PowerMockito.mock(DataSourceTransactionManager.class);

     DefaultTransactionDefinition def = new DefaultTransactionDefinition();
            def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRES_NEW);

            Mockito.when(mDataSourceTransactionManager.getTransaction(def)).thenReturn(null);