如何存根使用@InjectMocks注释的类的方法?

时间:2017-03-02 16:26:48

标签: java unit-testing junit mockito

MyDictionary.get方法下方通过调用map使用注入的map.get

只是出于好奇,我把MyDictionary.get方法打断了,就像我一直对嘲笑和间谍一样,所以我覆盖了注射。

但这仅在MyDictionary.get确实调用map.get时有效。如果map.get返回一些字符串(此处为空字符串),则存根Mockito.when不起作用。行为就好像它不存在一样。在断言行中,dictionary.get("key")等于空字符串。这是我不明白的。

@RunWith(MockitoJUnitRunner.class)
public class MockitoTest {

    @Mock
    Map<String, String>map;

    @InjectMocks
    MyDictionary dictionary;

    @Test
    public void testMyDictionary(){
        Mockito.when(dictionary.get("key")).thenReturn("value");
        Assert.assertEquals("value", dictionary.get("key"));
    }

    private static class MyDictionary{
        private Map<String, String> map; 
        public String get(String key){
            return map.get(key);
// or,
            return "";
        }
    }
}

1 个答案:

答案 0 :(得分:3)

当您尝试存根不是@Mock或@Spy的对象的方法时,我很惊讶您没有得到MissingMethodInvocationException

dictionary实例只是Mockito未代理的类的常规实例(因为只使用了@InjectMocks注释)。

另一个惊喜是,当null被触发时,你没有得到map.get,因为字符串返回方法的默认返回值是null

反正..

如果你想要存储`dictionary'实例的方法,你必须按如下方式配置你的测试类:

@InjectMocks
@Spy
MyDictionary dictionary;

@Test
public void testMyDictionary(){
    doReturn("value").when(dictionary).get("key);
    Assert.assertEquals("value", dictionary.get("key"));
}