[JUnit & Mockito] Return Multiple Values & Argument Matchers

🚀 Return Multiple Values

보통은 when() 과 thenReturn() 을 사용할 때

private final List<String> mock = mock(List.class);

@Test
public void returnSingleValue() {
    when(mock.size()).thenReturn(5);
    assertEquals(5, mock.size());
}

이런 식으로 하나의 값만을 리턴하는 형태로 Mock Object 를 사용하지만, 연속적으로 리턴값을 확인을 할 때는 다음과 같이 thenReturn() 을 필요한 만큼 더 붙이면 됩니다.

@Test
public void returnMultipleValues() {
    when(mock.size()).thenReturn(5).thenReturn(10);
    assertEquals(5, mock.size());
    assertEquals(10, mock.size());
}

🚀 Argument Matchers

다음으로는 해당 Mock Object에 파라미터와 함께 메소드를 사용할 때에는 다음과 같이 작성하시면 되는데, 만약 정해놓은 파라미터를 사용하지 않고 Mock Object를 사용하게 되면 전달한 값에 해당하는 디폴트값이 반환됩니다.

@Test
public void returnWithParameters() {
    when(mock.get(0)).thenReturn("in28minutes");
    assertEquals("in28minutes", mock.get(0));
    assertNull(mock.get(1));
}

그리고 위와 같이 값을 하드 코딩하지 않고 동적으로 파라미터 값을 설정하고 싶으시다면, Mockito의 Argument Matchers 클래스를 사용하시면 됩니다.

@Test
public void returnWithGenericParameters() {
    when(mock.get(anyInt())).thenReturn("in28minutes");
    assertEquals("in28minutes", mock.get(0));
    assertEquals("in28minutes", mock.get(1));
}