Question

我试图从MySQL数据库中检索一些ResultSet，并希望为我的类的公共方法（search＆amp; add）编写JUnit测试。

如何为数据库相关类编写JUnit测试？

public class MetropolisModel implements MetropolisControl {

    private int rowsCount = 0;
    private int columesCount = 0;
    private ResultSet lastResult = null;

    @Override
    public ResultSet search(String metropolis, String continent, String population, boolean populationLargerThan, boolean exactMatch) {
        ResultSet rs = null;

        try {

            Connection c = MyDB.getConnection();
            String query = createSearchQuery(metropolis, continent, population, populationLargerThan, exactMatch);

            PreparedStatement ps = c.prepareStatement(query);

            int position = 1;

            if (!metropolis.isEmpty()) {
                ps.setString(position, addPercentageSign(metropolis, exactMatch));
                position++;
            }
            if (!continent.isEmpty()) {
                ps.setString(position, addPercentageSign(continent, exactMatch));
                position++;
            }
            if (!population.isEmpty()) {
                ps.setString(position, population);
            }

            rs = ps.executeQuery();

            setTable(rs);

            System.out.println("after " + ps + "%nFound records: " + rowsCount);

        } catch (SQLException e) {
            e.printStackTrace();
        }

        return rs;

    }

    @Override
    public void add(String metropolis, String continent, String population) {

        try {

            Connection c = MyDB.getConnection();

            String query = "insert into metropolises values(?,?,?);";

            PreparedStatement ps = c.prepareStatement(query);

            ps.setString(1, metropolis);
            ps.setString(2, continent);

            if (population.isEmpty()) {
                ps.setNull(3, Types.BIGINT);
            } else {
                ps.setString(3, population);
            }

            ps.executeUpdate();

            search(metropolis, continent, population, true, true);

        } catch (SQLException ex) {
            ex.printStackTrace();
        }
    }

    private String createSearchQuery(String metropolis, String continent, String population, boolean populationLargerThan, boolean exactMatch) {
        String query = "select * from metropolises";

        if (!metropolis.isEmpty() || !continent.isEmpty() || !population.isEmpty()) {
            query += " where ";
            String operatorForString = exactMatch ? " = " : " like ";
            String operatorForInt = exactMatch ? " = " : populationLargerThan ? " > " : " < ";

            query += metropolis.isEmpty() ? "" : " metropolis " + operatorForString + " ?";
            query += continent.isEmpty() ? "" : addANDKeyword(" continent " + operatorForString + " ? ", !metropolis.isEmpty());
            query += population.isEmpty() ? "" : addANDKeyword(" population " + operatorForInt + " ? ", !metropolis.isEmpty() || !continent.isEmpty());
        }

        query += ";";
        System.out.println(query);
        return query;
    }

    private String addANDKeyword(String input, boolean needToAddKeyword) {
        if (!needToAddKeyword) {
            return input;
        } else {
            input = " and " + input;
            return input;
        }
    }

    private String addPercentageSign(String input, boolean exactMatch) {
        if (exactMatch) {
            return input;
        } else {
            return "%" + input + "%";
        }
    }

    private void setTable(ResultSet rs) throws SQLException {
        lastResult = rs;
        ResultSetMetaData rsmd = rs.getMetaData();
        columesCount = rsmd.getColumnCount();
        rs.last();
        rowsCount = rs.getRow();
    }
}

Answer 1

使用HSQLDB的内存实例来提高速度。（当然，如果您的数据库访问机制保护您不必担心数据库语法的差异，那么您只能这样做，因为您使用的任何数据库的语法可能与HSQLDB语法略有不同。）
每个测试类首先创建架构并使用示例数据填充数据库。
每个测试类使用自己的全名（getClass（）。getName（），可能在用下划线替换点后）作为数据库模式名称，以保证一个测试不会试图弄乱另一个测试的模式
每种测试方法都执行以下操作：
1. 开始交易
2. 可能会添加更多样本数据
3. 执行查询
4. 收到结果
5. 确保结果符合预期
6. ROLLS返回事务，因此每个测试方法都完全按照发现的方式离开数据库。

如何处理数据库查询的JUNIT测试

1 个答案: