模拟方法

Question

public class MyClass { 

    public int result(){ 
    int result = calculate(new ValueProvider()); 
      return result; 
     } 

    public int calculate(ValueProvider provider){ 
      return provider.getSalary(); 
     } 

} 

public class ValueProvider { 

    public int getSalary(){ 
     return 10000000; 
    } 
} 

我需要测试方法result()，但不得不嘲笑第二种方法计算其应返回默认值。

Answer 1

使用Mockito spy创建部分模拟。

例如：

public class TestMyClass { 

    @Test 
    public void aTest() { 
     MyClass spy = Mockito.spy(MyClass.class); 

     // this will cause MyClass.calculate() to return 5 when it 
     // is invoked with *any* instance of ValueProvider 
     Mockito.doReturn(5).when(spy).calculate(Mockito.any(ValueProvider.class)); 

     Assert.assertEquals(123, spy.result()); 
    } 
} 

在这种情况下，测试在calculate该调用嘲笑但result调用是真实的。从the docs：

您可以创建真实对象的间谍。当你使用间谍，那么真正的方法被调用（除非方法被扼杀）。