如何在忽略 LocalDateTime.now() 产生的时间差异的情况下断言查询？

Question

我在断言 LocalDateTime.now() 方法产生的时间时遇到问题，我想断言整个查询而忽略它产生的时间。 代码如下：

 @Test
public void getLastDrawResult_lotteryIdPresent() {
    Optional<Long> lotteryId = Optional.of(3L);
    Optional<String> name = Optional.empty();
    Optional<String> byName = Optional.empty();
    Optional<String> byDate = Optional.empty();
    Optional<Boolean> jackpotOnly = Optional.empty();

    String query = QUERY_MAIN_BLOCK.replace("/*statement0*/", LocalDateTime.now().plusMinutes(buyingLock).toString()) + CONDITION_SEARCH_BY_LAST_DRAW_RESULT;
    String updatedQuery = query.replace("/*statement1*/", " WHERE d.lottery_info_id = 3") + " AND d.lottery_info_id = 3";

    DrawResultDto drawResultDto = DrawResultDto.builder().id(3L).build();
    List<DrawResultDto> expectedDrawResultDto = singletonList(drawResultDto);

    when(jdbcTemplate.query(updatedQuery, drawResultDtoQueryBuilder.queryMapper)).thenReturn(expectedDrawResultDto);
    List<DrawResultDto> actualDrawResultDto = drawResultDtoQueryBuilder.getLastDrawResult(lotteryId, name, byName, byDate, jackpotOnly);

    verify(jdbcTemplate).query(updatedQuery, drawResultDtoQueryBuilder.queryMapper);
    assertEquals(expectedDrawResultDto, actualDrawResultDto);

这是比较失败的详细信息，因此您可以了解我需要忽略的内容：Comparison Failure Window Screenshot.

这是我正在测试的方法：

public List<DrawResultDto> getLastDrawResult(Optional<Long> lotteryId, Optional<String> name, Optional<String> byName, Optional<String> byDate, Optional<Boolean> jackpotOnly) {
    String query = QUERY_MAIN_BLOCK.replace("/*statement0*/", LocalDateTime.now().plusMinutes(buyingLock).toString()) +
            CONDITION_SEARCH_BY_LAST_DRAW_RESULT;
    StringBuilder builder = new StringBuilder();
    if (lotteryId.isPresent()) {
        builder.append(query.replace("/*statement1*/", " WHERE d.lottery_info_id = " + lotteryId.get()))
                .append(String.format(" AND d.lottery_info_id = %d", lotteryId.get()));
    } else {
        builder.append(query);
        name.ifPresent(nameValue -> builder.append(" AND lower(li.name) LIKE '%").append(SqlUtils.escapeLike(nameValue).toLowerCase()).append("%' "));
        jackpotOnly.ifPresent(jackpotOnlyValue -> {
            if (jackpotOnlyValue) {
                builder.append(" AND ").append(CONDITION_SEARCH_BY_JACKPOT_WIN);
            }
        });
        byName.ifPresent(s -> builder.append(" ORDER BY li.name ").append(SqlUtils.escapeLike(s)));
        byDate.ifPresent(s -> builder.append(" ORDER BY nfd.nearestDrawDate ").append(SqlUtils.escapeLike(s)));
        if (!byName.isPresent() && !byDate.isPresent()){
            builder.append(" ORDER BY li.name desc ");
        }
    }
    return jdbcTemplate.query(builder.toString(), queryMapper);
}

我试图从 Mockito 做matches()，没有运气。我似乎无法让正则表达式工作。就像匹配所有内容，但以下正则表达式：(\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}.\d{3})。其中选择了我需要忽略的部分。

Answer 1

这里的最佳实践包括您注入一个Clock，这将使您能够通过覆盖now 方法访问您需要的任何对象（包括LocalDateTime）。

public List<DrawResultDto> getLastDrawResult(
    Optional<Long> lotteryId,
    Optional<String> name,
    Optional<String> byName,
    Optional<String> byDate,
    Optional<Boolean> jackpotOnly,
    Clock clock) {
  LocalDateTime localDateTime = LocalDateTime.now(clock);
  /* ... */
}

与Ash's answer 一样，您可以使用当前签名创建一个新 SystemClock 的方法重载，然后测试接受时钟的方法。

/** Your original method signature. No changes to any calling code. */
public List<DrawResultDto> getLastDrawResult(
    Optional<Long> lotteryId,
    Optional<String> name,
    Optional<String> byName,
    Optional<String> byDate,
    Optional<Boolean> jackpotOnly) {
  return getLastDrawResult(lotteryId, name, byName, byDate, jackpotOnly,
      Clock.systemDefaultZone());
}

/** Your original method implementation. Test this one. */
public List<DrawResultDto> getLastDrawResult(
    Optional<Long> lotteryId,
    Optional<String> name,
    Optional<String> byName,
    Optional<String> byDate,
    Optional<Boolean> jackpotOnly,
    Clock clock) {
  LocalDateTime localDateTime = LocalDateTime.now(clock);
  /* ... */
}

这样，在您的测试中，您可以传入来自Clock.fixed 的值。

另外，您应该强烈考虑将此方法切换为参数化查询，JdbcTemplate supports natively。你甚至可以support named parameters using related classes。

Answer 2

似乎您是在编写代码之后编写测试：p（否则您永远不会遇到这样的问题）。

我要做的是将 localedatetime 传递给函数，以便您可以正确地对其进行单元测试：

public List<DrawResultDto> getLastDrawResult(LocalDateTime t, Optional<Long> lotteryId, Optional<String> name, Optional<String> byName, Optional<String> byDate, Optional<Boolean> jackpotOnly) {
    String query = QUERY_MAIN_BLOCK.replace("/*statement0*/", t.plusMinutes(buyingLock).toString()) +
    [...]
}

为了与现有代码保持一致，请保留实际签名：

public List<DrawResultDto> getLastDrawResult(Optional<Long> lotteryId, Optional<String> name, Optional<String> byName, Optional<String> byDate, Optional<Boolean> jackpotOnly) {
    return getLastDrawResult(LocaleDateTime.now(), Optional<Long> lotteryId, Optional<String> name, Optional<String> byName, Optional<String> byDate, Optional<Boolean> jackpotOnly);
}

现在您可以正确地对第一种方法进行单元测试（通过传递一个内置在测试方法中的 localedatetime），对第二种方法进行单元测试很简单。