模拟Guava CacheLoader回调方法

时间:2017-10-26 16:07:21

标签: java junit mockito guava

我一直试图嘲笑回调但没有成功。 这是一个示例代码:

public class TestService {
    private UtilService utilService;

    private LoadingCache<String, String> cache= CacheBuilder.newBuilder()
            .build(new CacheLoader<String, String>() {
                public String load(String key) throws Exception {
                    System.out.println("key = " + key);
                    return getKey(key);
                }
            });

    public String getkeyFromCache(String key) throws ExecutionException {
        return cache.get(key);
    }

    @VisibleForTesting
    public String getKey(String key) {
        return utilService.getKey(key);
    }
}

这样的测试用例:

@RunWith(MockitoJUnitRunner.class)
public class TestServiceTest {

    public static final String MYKEY = "Mykey";
    @Spy
    TestService testService=new TestService();

    @Before
    public void before() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testCache() throws ExecutionException {
        doReturn(MYKEY).when(testService).getKey(MYKEY);
        String result=testService.getkeyFromCache(MYKEY);
        String result2nd=testService.getkeyFromCache(MYKEY);
        verify(testService,times(1)).getKey(MYKEY);
    }

}

但看起来它没有被调用

Wanted but not invoked:
testService.getKey("Mykey");
-> at utils.TestServiceTest.testCache(TestServiceTest.java:38)

However, there were other interactions with this mock:
-> at utils.TestServiceTest.testCache(TestServiceTest.java:36)
-> at utils.TestServiceTest.testCache(TestServiceTest.java:37)

    at utils.TestServiceTest.testCache(TestServiceTest.java:38)
    at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
    at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
    at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
    at java.lang.reflect.Method.invoke(Method.java:498)
    at org.junit.runners.model.FrameworkMethod$1.runReflectiveCall(FrameworkMethod.java:50)
    at org.junit.internal.runners.model.ReflectiveCallable.run(ReflectiveCallable.java:12)
    at org.junit.runners.model.FrameworkMethod.invokeExplosively(FrameworkMethod.java:47)
    at org.junit.internal.runners.statements.InvokeMethod.evaluate(InvokeMethod.java:17)
    at org.junit.internal.runners.statements.RunBefores.evaluate(RunBefores.java:26)
    at org.junit.runners.ParentRunner.runLeaf(ParentRunner.java:325)
    at org.junit.runners.BlockJUnit4ClassRunner.runChild(BlockJUnit4ClassRunner.java:78)
    at org.junit.runners.BlockJUnit4ClassRunner.runChild(BlockJUnit4ClassRunner.java:57)
    at org.junit.runners.ParentRunner$3.run(ParentRunner.java:290)
    at org.junit.runners.ParentRunner$1.schedule(ParentRunner.java:71)
    at org.junit.runners.ParentRunner.runChildren(ParentRunner.java:288)
    at org.junit.runners.ParentRunner.access$000(ParentRunner.java:58)
    at org.junit.runners.ParentRunner$2.evaluate(ParentRunner.java:268)
    at org.junit.runners.ParentRunner.run(ParentRunner.java:363)
    at org.mockito.internal.runners.JUnit45AndHigherRunnerImpl.run(JUnit45AndHigherRunnerImpl.java:37)
    at org.mockito.runners.MockitoJUnitRunner.run(MockitoJUnitRunner.java:62)
    at org.junit.runner.JUnitCore.run(JUnitCore.java:137)
    at com.intellij.junit4.JUnit4IdeaTestRunner.startRunnerWithArgs(JUnit4IdeaTestRunner.java:68)
    at com.intellij.rt.execution.junit.IdeaTestRunner$Repeater.startRunnerWithArgs(IdeaTestRunner.java:47)
    at com.intellij.rt.execution.junit.JUnitStarter.prepareStreamsAndStart(JUnitStarter.java:242)
    at com.intellij.rt.execution.junit.JUnitStarter.main(JUnitStarter.java:70)

我想测试缓存功能但不测试UtilService,所以我想模拟它。 我想不出一种方法可以使用Answer或Captor来进行这项测试,或者我会想到。

2 个答案:

答案 0 :(得分:2)

你的问题是你自己创造了间谍。所以回调将与真实方法联系在一起。允许Mockito处理对象创建解决了这个问题。 (我正在使用mockito 1.10.19)

我重构了你的测试代码:

@RunWith(MockitoJUnitRunner.class)
public class TestServiceTest {

    public static final String MYKEY = "Mykey";
    @Spy
    TestService testService;

    @Test
    public void testCache() throws ExecutionException {
        doReturn(MYKEY).when(testService).getKey(MYKEY);

        String result=testService.getkeyFromCache(MYKEY);
        String result2nd=testService.getkeyFromCache(MYKEY);

        verify(testService).getKey(MYKEY);
    }

}

注意:当您使用Mockito跑步者时 - 您不需要自己初始化Mockito。此外,时间(1)是验证呼叫的默认值。

答案 1 :(得分:1)

当在load()方法中调用getKey()方法时,它在TestService类中调用,而不是通过mock-proxy调用。这就是为什么

doReturn(MYKEY).when(testService).getKey(MYKEY);

不起作用。但是,您可能并且应该模拟UtilService,并且只能访问一次。