数据库单元测试框架？

Question

在我的项目中，我使用了 spring、jpa 和 PostgreSQL DB， 我在数据库中有很多表，我需要对所有表进行单元测试。

是否有任何框架在每次测试完成后回滚所有事务，以便每个测试都将有新鲜/相同的数据库数据进行测试。这样在所有测试执行之后，数据库模式的数据将保持原样。

对此有何建议？

我对 DBUnit 有一些想法，但我需要为每个测试的每个输入数据编写 .xml 文件，并且需要在 setup() 中插入数据并在 TeaDown() 中清除/删除数据，但似乎没有更好的策略大部头书。

任何建议表示赞赏。 谢谢。

Answer 1

正如@Ryan所指出的......Spring参考的 测试部分应查阅手册。

一些启动技巧...

我们已经使用 Spring 的 AbstractTransactionalJUnit4SpringContextTests 处理了这个问题。

例如，我们定义一个抽象超类：

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration("file:WebContent/WEB-INF/testconfig/test-web-application-config.xml")
@TransactionConfiguration()
@Transactional
public abstract class OurAbstractTransactionalSpringContextTest extends AbstractTransactionalJUnit4SpringContextTests {
}

然后将需要附加上下文的各个子类定义为：

@ContextConfiguration("classpath:path/to/config/ConfigForTestCase.xml")
public class TestOurFunction extends OurAbstractTransactionalSpringContextTest {
    @Test
    public void testOurMethod() {
    }

}

请注意：

1. 并非所有测试类都需要附加上下文，请跳过特定子类上的@ContextConfiguration。
2. 我们通过 ant 执行并在 junit 任务上使用 forkmode="perBatch" 属性。这可确保所有测试都使用相同的上下文配置运行（无需为每个测试重新加载 Spring 上下文）。您可以使用 @DirtiesContext 来指示应在方法/类之后刷新上下文。
3. 使用@Test注释标记每个方法。 Spring 框架不会使用 Junit 的 public void testXXX() 约定来选取方法。

Answer 2

是否有任何框架可以在每次测试完成后回滚所有事务，以便每个测试都将有新鲜/相同的数据库数据进行测试。这样在所有测试执行之后，数据库模式的数据将保持原样。

来自我的其他答案之前发布的天，是的，这可以使用 DbUnit 实现。 （根据您的编辑，您不需要这个；我的答案的后续部分说明了我使用 DbUnit 的原因以及何时不使用它）。

以下代码片段演示了如何执行每个测试的设置：

@Before
public void setUp() throws Exception
{
    logger.info("Performing the setup of test {}", testName.getMethodName());
    IDatabaseConnection connection = null;
    try
    {
        connection = getConnection();
        IDataSet dataSet = getDataSet();
        //The following line cleans up all DbUnit recognized tables and inserts and test data before every test.
        DatabaseOperation.CLEAN_INSERT.execute(connection, dataSet);
    }
    finally
    {
        // Closes the connection as the persistence layer gets it's connection from elsewhere
        connection.close();
    }
}

private IDatabaseConnection getConnection() throws Exception
{
    @SuppressWarnings({ "rawtypes", "unused" })
    Class driverClass = Class.forName("org.apache.derby.jdbc.ClientDriver");
    Connection jdbcConnection = DriverManager.getConnection(jdbcURL, "XXX",
            "YYY");
    IDatabaseConnection databaseConnection = new DatabaseConnection(jdbcConnection);
    return databaseConnection;
}

private IDataSet getDataSet() throws Exception
{
    ClassLoader classLoader = this.getClass().getClassLoader();
    return new FlatXmlDataSetBuilder().build(classLoader.getResourceAsStream("database-test-setup.xml"));
}

database-test-setup.xml 文件包含将为每个测试插入数据库的数据。在 setup 方法中使用 DatabaseOperation.CLEAN_INSERT 可确保清除文件中指定的所有表（通过删除所有行），然后插入测试数据文件中的指定数据。

避免 DbUnit

我使用上述方法专门在每次测试开始之前清除序列，因为应用程序使用 JPA 提供程序在单独的事务中更新序列。如果您的应用程序没有执行类似的操作，那么您可以在 setup() 方法中简单地启动一个事务，并在测试后在拆卸时发出回滚。如果我的应用程序不使用序列（并且如果我不想重置它们），那么我的设置例程将非常简单：

@Before
public void setUp() throws Exception
{
    logger.info("Performing the setup of test {}", testName.getMethodName());
    // emf is created in the @BeforeClass annotated method
    em = emf.createEntityManager();
    // Starts the transaction before every test
    em.getTransaction.begin();
}

@After
public void tearDown() throws Exception
{
    logger.info("Performing the teardown of test {}", testName.getMethodName());
    if (em != null)
    {
        // Rolls back the transaction after every test
        em.getTransaction().rollback();
        em.close();
    }
}

此外，我使用 dbdeploy 与 Maven 一起使用，但这主要是为了使测试数据库与版本化数据模型保持最新。

Answer 3

Spring 的测试框架 正是为你做的。

Answer 4

我是按照以下方式处理的。

当项目处于测试模式时。我使用引导数据来使用 dbdeploy 进行测试
您可以断言的固定数据。并直接使用dao来测试应用程序的DAO和DB层。

希望它对

更新

有所帮助，例如您的系统中有一个名为Person的实体，现在您可以在其上测试基本的CRUD操作。

• 运行引导数据脚本来加载数据，
• 从数据库中检索所有人员并对其进行断言。同样，查看所有 CRUD

来回滚您可以标记的事务，

@TransactionConfiguration(transactionManager = "transactionManager", defaultRollback = true)

这样它将回滚数据库内容