2

我正在使用带有 AbstractTransactionalJUnit4SpringContextTests 的 JUnit4，并且想创建简单的 CRUD 单元测试 - 如下所示。但是，对 SQL 更新的调用导致事务管理器为我的测试实例生成一个新 ID。由于在更新过程中没有传回新 ID，因此我不再拥有我正在测试的行的主键 - 这阻碍了测试的其余部分。

有没有办法阻止事务管理器在调用更新时为书生成新 ID？（如果没有更好的方法来测试 CRUD？）

@Test
public void testCRUDBook() {
    Book b1 = new Book(title, author);
    BookFactory factory = database.getBookFactory();
    int id = factory.createBook(b1);

    Book b2 = factory.readBook(id);        
    assertEquals(b1.getTitle(), b2.getTitle());
    assertEquals(b1.getAuthor(), b2.getAuthor());

    b2.setTitle("title 2");
    b2.setAuthor("author 2");
    assertTrue(factory.updateBook(b2));

    // The problem arises here as updating the book record causes a new Id 
    // to be generated so querying by Id is no longer possible.
    Book b3 = factory.readBook(b2.getId());        
    assertEquals(b3.getTitle(), "title 2");
    assertEquals(b3.getAuthor(), "author 2");    

    assertTrue(factory.deleteBook(b3));
}

书是这样的：

public class Book {
    private int id;
    private String title;
    private String author;

    public Book() {}

    // NEW
    public Book(String title, String author) {
        this.title = title;
        this.author = author;
    }

    // READ
    public Book(int id, String title, String author) {
        this.id = id;
        this.title = title;
        this.author = author;
    }

    // GENERIC ACCESSORS (left out for brevity)
}

为了完整性 - 工厂：

public class BookFactory extends BaseDatabaseFactory {

        @Transactional
    public int createBook(final Book b) {
        final String insertSQL = "INSERT INTO book (author, title) VALUES (?, ?)";
        KeyHolder keyHolder = new GeneratedKeyHolder();
        try {
            int update = jdbcTemplate.update(
                    new PreparedStatementCreator() {
                        @Override
                        public PreparedStatement createPreparedStatement(Connection connection) throws SQLException {
                            PreparedStatement ps = connection.prepareStatement(insertSQL, PreparedStatement.RETURN_GENERATED_KEYS);
                            ps.setString(1, b.getTitle());
                            ps.setString(2, b.getAuthor());
                            return ps;
                        }
                    },
                    keyHolder);
            if (update != 1) {
                throw new IllegalStateException("Adding book record to database resulted in " + update + " records.");
            }
            return keyHolder.getKey().intValue();
        } catch (DataAccessException ex) {
            String msg = "Falied to create new book:" + b.toString() + "ex:" + ex.getMessage();
            throw new RuntimeException(msg);
        }
    }

    private static class BookRowMapper implements RowMapper {
        @Override
        public Object mapRow(ResultSet rs, int rowNum) throws SQLException {
            Book b = new Book();
            b.setId(rs.getInt("id"));
            b.setTitle(rs.getString("title"));
            b.setAuthor(rs.getString("author"));
            return b;
        }
    }

    @Transactional
    public Book readBook(int id) {
        Book b = null;
        try {
            String sql = "SELECT * FROM book WHERE id = " + id;
            b = (Book) jdbcTemplate.queryForObject(sql, new Object[]{}, new BookRowMapper());
        } catch (DataAccessException ex) {
            throw new RuntimeException("Falied to locate book.", ex);
        }
        return b;
    }    

    @Transactional
    public boolean updateBook(final Book b) {
        final String updateSQL = "UPDATE book SET author = ?, title = ? WHERE id = ?";
        try {
            jdbcTemplate.update(
                    new PreparedStatementCreator() {
                        @Override
                        public PreparedStatement createPreparedStatement(Connection connection) throws SQLException {
                            PreparedStatement ps = connection.prepareStatement(updateSQL);
                            ps.setString(1, b.getAuthor());
                            ps.setString(2, b.getTitle());
                            ps.setInt(3, b.getId());
                            return ps;
                        }
                    });
        } catch (DataAccessException ex) {
            log.error("Falied to update Book:" + b.toString(), ex);
            return false;
        }
        return true;
    }

    @Transactional
    public boolean deleteBook(final Book b) {
        try {
            jdbcTemplate.update("DELETE FROM book WHERE id = ?", b.getId());
        } catch (DataAccessException ex) {
            log.error("Falied to delete Book:" + b.toString(), ex);
            return false;
        }
        return true;
    }    
}
4

2 回答 2

1

Can you make the id an Integer type instead? It looks as if the factory causes a new instance to be created when you do the update to b2. id should be nullable.

于 2012-07-11T21:42:50.440 回答
0

在尝试记录问题时修复了问题（通过粘贴整洁的工厂代码）。createBook(book) 和 readBook(int) 工厂方法缺少 @Transactional 注释；我曾认为事务是从调用测试容器继承的——显然情况并非如此。跨事务拆分测试必须创建该行的多个实例。

我希望这可以帮助遇到同样问题的其他人。

于 2012-07-11T22:09:44.620 回答