模拟外部服务返回null

时间:2018-10-11 14:39:54

标签: java unit-testing mockito junit4

我有一个非常简单的Dictionary类,可以调用外部API。

 public class Dictionary {
        protected ExternalService service = new ExternalService();       

        public String getValue(String key, String dictionaryName) {
            ExternalServiceInput input = new ExternalServiceInput();
            ExternalServiceOutput output = new ExternalServiceOutput();

            input.setKey(key);
            input.setDictionaryName(dictionaryName);

            try {
                output = service.invoke(input);
            } catch (Exception e) {         
                return null;
            }       

            return output.getValue();       
        }
    }

它工作正常,但是我想为此编写单元测试,所以我决定需要模拟service.invoke()

    @Mock   
    private ExternalService service;        

    @InjectMocks
    private Dictionary dictionary;      
    @InjectMocks
    private ExternalServiceOutput output;
    @InjectMocks
    private ExternalServiceInput input;


    @Before
    public void setUp() throws Exception {      
        MockitoAnnotations.initMocks(this);

        input.setKey("testKey");
        input.setDictionaryName("testDictionary");  
        output.setValue("testValue");           
    }

    @Test
    public void shouldReturnValue() throws Exception {
        when(service.invoke(input)).thenReturn(output);     
        assertEquals(output.getValue(), dictionary.getValue(input.getKey(), input.getDictionaryName()));        
    }

我尝试使用InputOutput作为常规字段,或者使用setUp方法对其进行初始化,所有内容最终都以NullPointerException类的Dictionary结尾

return output.getValue();

有人可以指出我做错了什么吗?

1 个答案:

答案 0 :(得分:2)

您应该在ExternalServiceInput类中重写equals和hashCode,或更改您的模拟以接受ExternalServiceInput的任何对象

when(service.invoke(Mockito.any(ExternalServiceInput.class))).thenReturn(output);