如何使用 Mockito 和 JUnit 检查方法中的 if 语句？

Question

我有我应该测试的方法。代码（当然删减了部分）：

 public class FilterDataController {

    public static final String DATE_FORMAT = "yyyy-MM-dd";

    @Autowired
    private FilterDataProvider filterDataProvider;

    @ApiOperation(value = "Get possible filter data",response = ResponseEntity.class)
    @ApiResponses(value = {
            @ApiResponse(...),
            @ApiResponse(...)})
    @RequestMapping(path = "...", method = RequestMethod.GET)
    public ResponseEntity<Object> getPossibleFilterData(
            @RequestParam(value = "startDate") @DateTimeFormat(pattern=DATE_FORMAT) final Date startDate,
            @RequestParam(value = "endDate") @DateTimeFormat(pattern=DATE_FORMAT) final Date endDate) {
        if (endDate.compareTo(startDate) == -1){
            throw new ValueNotAllowedException("End date should be after or equal start date");
        }
        else {
            Date newEndDate = endDate;
            if (startDate.equals(endDate)){
                newEndDate = new Date(endDate.getTime() + TimeUnit.DAYS.toMillis(1) - 1);
            }

            List<String> possibleCountries = Lists.newArrayList(filterDataProvider.getPossibleCountries(startDate, newEndDate));

            return new ResponseEntity<>(new FilterResponse(possibleCountries),HttpStatus.OK);
        }
    }
}

问题：如何使用 Mockito 和 JUnit 检查方法 getPossibleFilterData 中的 if 语句？我想将相等的日期传递给方法，然后检查我的 if 语句是否正常工作。

原文由 Woland 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

如果你真的想要一个 纯 单元测试而不是集成测试，你可以依靠注释 @Mock 来模拟你的服务 FilterDataProvider 和 @InjectMocks to-84813-ject your mock into-to-84813-您的实例 FilterDataController 。

然后你可以提出 3 个测试：

1. 日期正确但不同的一项测试，
2. 另一个日期正确但相等的日期
3. 最后一个日期不正确的日期将抛出 ValueNotAllowedException 可以使用 @Test(expected = ValueNotAllowedException.class) 开箱即用地进行测试。

如果您需要确保 filterDataProvider.getPossibleCountries(startDate, newEndDate) 已使用您需要使用的预期参数调用 verify 。

代码将是这样的：

 @RunWith(MockitoJUnitRunner.class)
public class FilterDataControllerTest {
    @Mock
    FilterDataProvider filterDataProvider;
    @InjectMocks
    FilterDataController controller;

    @Test(expected = ValueNotAllowedException.class)
    public void testGetPossibleFilterDataIncorrectDates() {
        controller.getPossibleFilterData(new Date(1L), new Date(0L));
    }

    @Test
    public void testGetPossibleFilterDataCorrectDates() {
        // Make the mock returns a list of fake possibilities
        Mockito.when(
            filterDataProvider.getPossibleCountries(
                Mockito.anyObject(), Mockito.anyObject()
            )
        ).thenReturn(Arrays.asList("foo", "bar"));
        ResponseEntity<Object> response = controller.getPossibleFilterData(
            new Date(0L), new Date(1L)
        );
        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());
        // Make sure that
        // filterDataProvider.getPossibleCountries(new Date(0L), new Date(1L))
        // has been called as expected
        Mockito.verify(filterDataProvider).getPossibleCountries(
            new Date(0L), new Date(1L)
        );
        // Test response.getBody() here
    }

    @Test
    public void testGetPossibleFilterDataEqualDates() {
        // Make the mock returns a list of fake possibilities
        Mockito.when(
            filterDataProvider.getPossibleCountries(
                Mockito.anyObject(), Mockito.anyObject()
            )
        ).thenReturn(Arrays.asList("foo", "bar"));
        // Call the controller with the same dates
        ResponseEntity<Object> response = controller.getPossibleFilterData(
            new Date(1L), new Date(1L)
        );
        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());
        Mockito.verify(filterDataProvider).getPossibleCountries(
            new Date(1L), new Date(TimeUnit.DAYS.toMillis(1))
        );
        // Test response.getBody() here
    }
}

原文由 Nicolas Filotto 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

您将不得不模拟 FilterDataProvider 然后使用 InjectMocks 将其注入您的测试类。

getPossibleFilterData 将被测试的方法，所以选择任何特定的日期（使用 Calendar.set(...) ，然后 Calendar.getTime() 作为）并发送开始日期和结束日期。

现在，在 getPossibleFilterData 完成后，您可以验证调用 filterDataProvider.getPossibleCountries 的结束日期是否比开始日期早一毫秒。这可以通过 Calendar.getTimeInMillis() 在模拟类的方法中完成，或者通过使用比最初指定的日期多一毫秒的日期使用 Mockito 进行验证。

编辑：提供的代码示例：

 public class FilterDataControllerTest {
    @Test
    public void testSameDate() {
        FilterDataProvider provider = Mockito.mock(FilterDataProvider.class);
        FilterDataController controller = new FilterDataController(provider);

        Date startDate = new GregorianCalendar(2016, Calendar.JANUARY, 11).getTime();
        Date endDate = new GregorianCalendar(2016, Calendar.JANUARY, 11).getTime();
        Date expectedEndDate = new Date(endDate.getTime() + TimeUnit.DAYS.toMillis(1) - 1);

        controller.getPossibleFilterData(startDate, endDate);

        Mockito.verify(provider).getPossibleCountries(Mockito.eq(startDate), Mockito.eq(expectedEndDate));
    }
}

原文由 mdewit 发布，翻译遵循 CC BY-SA 3.0 许可协议