如何在 JUnit 测试中初始化数据

Question

我的任务是为服务层编写压力（负载）测试。主要有CRUD操作。我们使用 JUnit 作为测试框架，使用 JUnitPerf 构建负载测试，使用 Spring 注入服务 bean，使用 hibernate 访问数据库。

压力测试类似于：读取实体-更新实体-保存-再次读取并比较。但是为了构建测试，我需要数据库中的一些测试数据，因此我需要在测试之前创建这些数据并在测试之后删除它。所需的流程：创建测试数据-在多个线程中运行测试-所有线程执行完成后删除测试数据。测试数据很多，所以最好使用一些测试转储sql文件来获取它。所以我需要的是：将数据从文件加载到数据库中-执行压力测试-删除所有加载的数据。

我使用 SchemaExport 加载数据。我遇到了以下异常：

org.hibernate.HibernateException: No local DataSource found for configuration - 'dataSource' property must be set on LocalSessionFactoryBean
at org.springframework.orm.hibernate3.LocalDataSourceConnectionProvider.configure(LocalDataSourceConnectionProvider.java:49)
at org.hibernate.connection.ConnectionProviderFactory.newConnectionProvider(ConnectionProviderFactory.java:124)
at org.hibernate.connection.ConnectionProviderFactory.newConnectionProvider(ConnectionProviderFactory.java:56)
at org.hibernate.tool.hbm2ddl.ManagedProviderConnectionHelper.prepare(ManagedProviderConnectionHelper.java:27)
at org.hibernate.tool.hbm2ddl.SchemaExport.execute(SchemaExport.java:180)
at org.hibernate.tool.hbm2ddl.SchemaExport.create(SchemaExport.java:133)
    .................

这是我的 SessionFactory bean 的定义：

  <bean id="sessionFactory" class="org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean">
    <property name="dataSource" ref="dataSource"/>
    <property name="hibernateProperties">
        <value>
            hibernate.dialect=${hibernate.dialect}
            hibernate.show.sql=${hibernate.show.sql}
        </value>
    </property>
    <property name="annotatedClasses">
        <list>
            ...
            my classes
            ...
        </list>
    </property>
</bean>

我用以下方式初始化测试：

 @BeforeClass
public static void createTestData() throws AccessDeniedException, AccountException, SQLException {
    ClassPathXmlApplicationContext appCtx = new ClassPathXmlApplicationContext("classpath:/applicationContext_test.xml");
    AnnotationSessionFactoryBean sessionFactoryBean = (AnnotationSessionFactoryBean) appCtx.getBean("sessionFactory");
    org.hibernate.cfg.Configuration configuration = sessionFactoryBean.getConfiguration();
    SchemaExport schemaExport = new SchemaExport(configuration);
    schemaExport.drop(false, true);
    schemaExport.create(false, true);
    if (schemaExport.getExceptions().size() > 0) {
        for (Object exception : schemaExport.getExceptions()) {
            if (exception instanceof Throwable) {
                ((Throwable) exception).printStackTrace();
            }
        }
        throw new IllegalStateException();
    }
}

我提到我需要负载测试来明确我不能在测试块中包含数据初始化。

我有两个问题： 1）如何在负载测试之前初始化数据并在负载测试之后删除它？ 2）我走的路对吗？或者也许我应该改用另一种技术进行压力测试？

Answer 1

您可以使用 DBUnit 来为您加载数据。或者使用 Unitils 将 DBUnit 与 Spring 和 Hibernate 集成。

问候，

Answer 2

如果您使用 Spring 进行测试，您应该考虑不使用 Spring 测试支持，而不是自己加载应用程序上下文并查找 bean。

您可以在测试类上执行类似的操作并自动注入 bean。

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations={"applicationContext_test.xml"})

关于测试后的“清理”，我一直考虑使用 测试范围的 Spring 事务管理。应该可以将测试开始声明为@Transactional，并在测试后以编程方式回滚事务。不需要通过这种方式显式删除数据。

到目前为止这只是一个想法。一定要自己尝试一下...