java - 如何模拟在构造函数中创建的类？

Question

假设我有一个 bean JdbcTemplate，并且在只有一次类中我需要显式 NamedParameterJdbcTemplate。它是在构造函数中创建的。

问题:如何在 junit 测试期间模拟它？

@Service
public class QueryService {
    private final NamedParameterJdbcTemplate namedJdbc;

    public BookingExportService(JdbcTemplate jdbc) {
        this.namedJdbc = new NamedParameterJdbcTemplate(jdbc);
    }

    public void sql() {
        namedJdbc.query(sql1, mapSqlParameterSource, resultSetExtractor);
        namedJdbc.query(sql2, mapSqlParameterSource, resultSetExtractor);
    } 
}

在这里定义模拟很困难，因为我必须模拟底层的 JdbcTemplate 调用。这可以按如下方式工作:

@MockBean
private JdbcTemplate jdbc;

@Test
public void testQuery() {
    when(jdbc.query(any(PreparedStatementCreator.class), any(ResultSetExtractor.class))).thenReturn(..);
}

但是:如果我有不同的 sql 查询应该产生不同的结果(如上面的 sql1 和 sql2)，那么我就不走运了，在模拟期间不能区分 sql。

我的机会有多大？

Answer 1

这里:

this.namedJdbc = new NamedParameterJdbcTemplate(jdbc);

这只会让您的代码难以测试。基本上你在这里绕过了依赖注入(inject)，好吧，这意味着:你对该字段内容没有任何控制权。

三个选项:

• 转向像 PowerMock(ito) 或 JMockit 这样的模拟框架，它们允许您控制对 new() 的调用(不推荐)
• 更改您的代码，例如使用构造函数伸缩...这样您就可以简单地传递该类的一个实例(然后可以轻松地使用 Mockito 等“普通”框架进行模拟)
• 因为你已经在使用一个有自己的依赖注入(inject)想法的框架(例如@Autowired):退后一步，做一些框架暗示它要做的事情。