是否可以验证在Mockito中以不同线程运行的模拟方法?

时间:2013-06-20 06:17:55

标签: java mocking mockito powermock

我有类似以下的方法,

public void generateCSVFile(final Date billingDate) {
    asyncTaskExecutor.execute(new Runnable() {
        public void run() {
            try {
                accessService.generateCSVFile(billingDate);
            } catch (Exception e) {
                LOG.error(e.getMessage());
            }
        }
    });
}

我嘲笑过:

PowerMockito.doNothing().when(accessService).generateCSVFile(billingDate);

但是当我确认:

verify(rbmPublicViewAccessService, timeout(100).times(1)).generateCSVFile(billingDate);

它让我没有被调用。这是因为它是通过单独的线程调用的,是否可以验证在不同线程中调用的方法?

3 个答案:

答案 0 :(得分:34)

当您验证调用时,Runnable很可能尚未执行asyncTaskExecutor,从而导致单元测试中出现验证错误。

解决此问题的最佳方法是在验证调用之前加入生成的线程并等待执行。

如果你无法获得线程的实例,可能的解决方法是模拟asyncTaskExecutor并实现它,以便它直接执行runnable。

private ExecutorService executor;

@Before
public void setup() {
    executor = mock(ExecutorService.class);
    implementAsDirectExecutor(executor);
}

protected void implementAsDirectExecutor(ExecutorService executor) {
    doAnswer(new Answer<Object>() {
        public Object answer(InvocationOnMock invocation) throws Exception {
            ((Runnable) invocation.getArguments()[0]).run();
            return null;
        }
    }).when(executor).submit(any(Runnable.class));
}

答案 1 :(得分:9)

我遇到了同样的问题并且使用了超时参数 http://javadoc.io/page/org.mockito/mockito-core/latest/org/mockito/Mockito.html#22 但是在

中使用参数0
verify(someClass, timeout(0)).someMethod(any(someParameter.class));

它有效。我假设测试线程产生,因此其他线程有机会完成其工作,适当地调用模拟。 它仍然闻起来像黑客。

答案 2 :(得分:1)

要进一步迭代Tom的答案-使用Java 8 Lambda,您现在可以使用以下代码来模拟Executor,这更加简洁:

    doAnswer((Answer<Void>)invocation -> {
        ((Runnable)invocation.getArgument(0)).run();
        return null;
    }).when(executorService).submit(any(Runnable.class));