在模擬類中使用Java8 lambda進行參數匹配

Question

我正在進行單元測試的類需要一個依賴項並調用該依賴項的函數。該函數將一個複雜的對象作爲參數，併產生一個結果。現在，我想嘲笑依賴關係，並根據傳入的參數返回一些內容。下面給出了一個簡化的工作版本。

我可以在when方法中使用Java 8 lambda表達式來消除ArgHasNext類嗎？像下面的註釋代碼（不編譯）。

class ArgHasNext implements ArgumentMatcher<Arg> { 
    public boolean matches(Arg arg) { 
     return arg.hasNext(); 
    } 

    @Override 
    public boolean matches(Object o) { 
     return matches((Arg)o); 
    } 
} 

@RunWith(MockitoJUnitRunner.class) 
public class ArgumentMatcherTest { 

    @Mock 
    private Dependency dep = mock(Dependency.class); 

    @Test 
    public void test() { 

     when(dep.func(argThat(new ArgHasNext()))).thenReturn(true); 
     // when(dep.func(argThat((Arg a) -> a.hasNext()))).thenReturn(true); 
     // when(dep.func(argThat((Arg a) -> !a.hasNext()))).thenReturn(false); 
     Sut sut = new Sut(dep); 
     assertEquals(sut.method(new Arg(true)), "True"); 
     assertEquals(sut.method(new Arg(false)), "False"); 
    } 
} 

class Arg { 
    private boolean hasNext; 

    public Arg(boolean hasNext) { 
     this.hasNext = hasNext; 
    } 

    public boolean hasNext() { 
     return this.hasNext; 
    } 
} 

class Sut { 
    private Dependency dep; 

    public Sut(Dependency dep) { 
     this.dep = dep; 
    } 

    public String method(Arg arg) { 

     if (dep.func(arg)) { 
      return "True"; 
     } 
     else { 
      return "False"; 
     } 
    } 

} 

class Dependency { 
    public boolean func(Arg arg) { 
     if (arg.hasNext()) { 
      return true; 
     } 
     return false; 
    } 
} 

我正在使用Mockito核心版本2.0.54-beta。

編輯 好吧，也許我過分簡化了這個例子。在實際情況中，依賴項func方法返回在被測試方法返回之前在SUT中處理的分頁查詢結果。根據Arg，我希望依賴項func在第一次調用結果時返回第1頁結果，第2次結果返回結果。我只能做到這一點，當我可以根據傳遞給函數調用的參數模擬返回不同的值。這可能在Mockito中使用lambdas？現在

Answer 1

，我想嘲笑的依賴，有它基於傳入的參數返回的東西。

你不應該這樣做擺在首位。 任何機會讓你的嘲笑返回定義好的常量值。

原因是您應該保持測試代碼儘可能簡單，以減少由於測試代碼錯誤而導致測試失敗的風險。

---

你的問題的解決方案可能是mockitos Answer接口：

doAnswer(new Answer<YourReturnType>(){ 
    public YourReturnType answer(InvocationOnMock invocation) { 
     YourParameterType parameter = (YourParameterType)invocation.getArguments()[0]; 
     // calculate your return value 
     return yourCalculatedReturnValue; 
    } 
}).when(yourMock).theMethod(any(YourParameterType.class)); 

---

需要明確的是，我的模擬返回一個恆定值。該模擬被多次調用，我希望它第二次返回一個不同的值。 - MvdD

你應該寫過你的問題。解決方案很簡單，因爲它聽起來：

doReturn(VALUE_FOR_FIRST_CALL). 
    thenReturn(VALUE_FOR_SECOND_CALL). 
    thenReturn(VALUE_FOR_ANY_FURTHER_CALL). 
    when(mock).theMethod(); 

或者，如果你喜歡呆在少健談：

doReturn(VALUE_FOR_FIRST_CALL, VALUE_FOR_SECOND_CALL, VALUE_FOR_ANY_FURTHER_CALL). 
    when(mock).theMethod(); 

Answer 2

我想你會走上錯誤的兔子洞。含義：模擬規格是模擬規格。他們不應該什麼其他。

我的意思是：我會建議你寫這樣的測試：

@Test 
public void testForTrue() { 
    when(dep.func(any()).thenReturn(true); 
    Sut sut = new Sut(dep); 
    assertEquals(sut.method(new Arg(true)), "True"); 
} 

@Test 
public void testForFalse() { 
    when(dep.func(any()).thenReturn(false); 
    Sut sut = new Sut(dep); 
    assertEquals(sut.method(new Arg(false)), "False"); 
} 

如果在所有的，我會再重構爲：

private void testFor(boolean depReturnValue, expectedResult) { 
    when(dep.func(any()).thenReturn(depReturnValue); 
    Sut sut = new Sut(dep); 
    assertEquals(sut.method(new Arg(depReturnValue)), expectedResult); 
} 

，並呼籲從@Test方法，該方法如上所述。

換句話說：不要花費大量的時間和精力來製造「智能」嘲笑規格。而是專注於編寫簡單的直接測試。

當你可以編寫一個測試而不需要對傳遞給模擬調用的參數進行復雜的分析 - 然後去解決這個問題。

Answer 3

回答我自己的問題。由於模擬方法被多次調用了一個不同的複雜參數對象，我太專注於使模擬模型返回不同的東西，具體取決於傳入的參數。

解決方案是在隨後的調用中返回不同的結果，這很簡單，使用Mockito。

@RunWith(MockitoJUnitRunner.class) 
public class ArgumentMatcherTest { 

    @Mock 
    private Dependency dep = mock(Dependency.class); 

    @Test 
    public void test() { 

     // return true on first invocation, false on second. 
     when(dep.func(anyObject())).thenReturn(true).thenReturn(false); 

     Sut sut = new Sut(dep); 
     assertEquals(sut.method(new Arg(true)), "True"); 
     assertEquals(sut.method(new Arg(false)), "False"); 
    } 
}