如何对收到正确的 PreparedStatement 进行单元测试？

Question

我想将合适的对象传递给 verify 方法，而不仅仅是 any()。

有办法吗？

我不能只是采用和复制 Lambda 方法并将结果传递给验证。这不起作用，因为无法直接测试 Lambda。

我的单元测试显然还没有接近测试任何东西：

    @Test
public void testRunTrigger() {
    campaignTrigger.updateCampaignStatus();

    verify(jdbcTemplate).update(any(PreparedStatementCreator.class));
    assertEquals("UPDATE campaign SET state = 'FINISHED'  WHERE state IN ('PAUSED','CREATED','RUNNING') AND campaign_end < ? ", campaignTrigger.UPDATE_CAMPAIGN_SQL);
}

这是我正在测试的课程：

@Component
@Slf4j
public class CampaignTrigger {
final String UPDATE_CAMPAIGN_SQL = String.format("UPDATE campaign SET state = '%s' " +
                " WHERE state IN (%s) AND campaign_end < ? ", FINISHED,
        Stream.of(PAUSED, CREATED, RUNNING)
                .map(CampaignState::name)
                .collect(Collectors.joining("','", "'", "'")));

@Autowired
private JdbcTemplate jdbcTemplate;

@Scheduled(cron = "${lotto.triggers.campaign}")
@Timed
void updateCampaignStatus() {
    jdbcTemplate.update(con -> {
        PreparedStatement callableStatement = con.prepareStatement(UPDATE_CAMPAIGN_SQL);
        callableStatement.setTimestamp(1,  Timestamp.valueOf(LocalDateTime.now()));
        log.debug("Updating campaigns statuses.");
        return callableStatement;
    });
}

任何建议或理论知识，这不是这样做的方法，我将不胜感激。

Answer 1

您不应该模拟您无法控制的代码。仅模拟您进行测试的代码，因为在模拟时您假设您知道（即您定义）模拟类的工作方式。

在这里，你不知道jdbcTemplate 是如何工作的，也不知道用一些 lambda 调用它是否真的如你所想。

使用您无法控制的代码测试您的代码是集成测试的重点。 IE。您应该将CampaignTrigger 与真实数据库（或内存中的数据库）一起测试，而不是模拟jdbcTemplate。

Answer 2

您可以通过捕获用于该调用的对象来试试运气，请参阅here。这允许编写这样的代码：

ArgumentCaptor<Person> argument = ArgumentCaptor.forClass(Person.class);
verify(mock).doSomething(argument.capture());
assertEquals("John", argument.getValue().getName());

让您完全访问传递给您的方法调用的对象！请注意，mockito 最近引入了一个 @Captor 注释，使事情变得更容易使用。

编辑；鉴于@Morfic 的cmets：他所说的绝对合理。

这个答案给出了“立即”的提示，你可以如何解决这个特定的问题。

Beyond：合理的方法总是总是将“待测单元”切片......尽可能小！

您的类/方法应该只负责一项职责；然后你确保可以用最简单的方法测试实现。

所以：如果问题是：“我应该使用参数捕获器还是应该更好地修改我的生产代码” - 然后修改你的生产代码。