13

我有几个 DAO 对象用于从数据库中检索信息，我真的很想为它们编写一些自动化测试，但我很难弄清楚如何去做。

我正在使用 SpringJdbcTemplate运行实际查询（通过准备好的语句）并将结果映射到模型对象（通过RowMapper类）。

如果我要编写单元测试，我不确定我将/应该如何模拟对象。例如，由于只有读取，我会使用实际的数据库连接而不是模拟 jdbcTemplate，但我不确定这是否正确。

这是批处理中最简单的 DAO 的（简化）代码：

/**
 * Implementation of the {@link BusinessSegmentDAO} interface using JDBC.
 */
public class GPLBusinessSegmentDAO implements BusinessSegmentDAO {
    private JdbcTemplate jdbcTemplate;

    private static class BusinessSegmentRowMapper implements RowMapper<BusinessSegment>  {
        public BusinessSegment mapRow(ResultSet rs, int arg1) throws SQLException { 
            try {
                return new BusinessSegment(rs.getString(...));
            } catch (SQLException e) {
                return null;
            }
        }
    }

    private static class GetBusinessSegmentsPreparedStatementCreator 
        implements PreparedStatementCreator {
        private String region, cc, ll;
        private int regionId;

        private GetBusinessSegmentsPreparedStatementCreator(String cc, String ll) {
            this.cc = cc;
            this.ll = ll;
        }

        public PreparedStatement createPreparedStatement(Connection connection)
                throws SQLException {           
            String sql = "SELECT ...";

            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1, cc);
            ps.setString(2, ll);
            return ps;
        }
    }

    public GPLBusinessSegmentDAO(DataSource dataSource) {
        jdbcTemplate = new JdbcTemplate(dataSource);
    }

    public Collection<BusinessSegment> getBusinessSegments(String cc, String ll) {
        return jdbcTemplate.query(
                new GetBusinessSegmentsPreparedStatementCreator(cc, ll), 
                new BusinessSegmentRowMapper());
    }

}

任何想法将不胜感激。

谢谢！

4

3 回答 3

11

请查看以下链接：

  1. 使用 Spring 和 DbUnit 测试 SQL 查询
  2. 使用 JdbcTemplate 测试代码的 MockObjects 或 DBUnit

希望有帮助。

编辑：

这里是RowMapperTests 的 GitHub 版本，方便参考。

于 2011-03-23T10:28:46.743 回答
5

JdbcTemplate我建议打破你对类的依赖，JdbcOperations而是使用接口，例如

public class GPLBusinessSegmentDAO implements BusinessSegmentDAO {
    private final JdbcOperations jdbc;

    public GPLBusinessSegmentDAO(DataSource dataSource) {
        this(new JdbcTemplate(dataSource));
    }

    public GPLBusinessSegmentDAO(JdbcOperations jdbc) {
        this.jdbc = jdbc;
    }

    // ... DAO methods here
}

您的单元测试可以调用第二个构造函数，传入一个模拟JdbcOperations对象。由于所有数据库操作都是通过jdbc对象执行的，因此您可以轻松地模拟它。

您的实时代码可以像以前一样调用第一个构造函数。

于 2011-03-23T10:59:11.067 回答
2

要为此编写真正的单元测试，您不会接触到真正的数据库。但是，您可能会发现将真实的 DataSource 传递给您的底层数据库更为实用，并根据您传入的 cc 和 ll 值测试 getBusinessSegments() 方法返回 0、1 和许多结果。

另一个值得研究的选项是传入嵌入式Java DB的 DataSource，该数据源在 setUp/@Before 方法中使用您的模式初始化。我想您真正想要测试的是 SELECT... 查询正确映射到架构，因此这样的测试将捕获运行时出现的任何错误，例如，架构发生变化。

于 2011-03-23T10:30:32.837 回答