单元测试后的回滚序列

Question

我正在使用spring test运行JUnit测试，我的代码如下所示：

@RunWith(SpringJUnit4ClassRunner.class)  
@ContextConfiguration(locations = {})
@Transactional (propagation = Propagation.REQUIRED)
@Rollback
public class MyTest {

    @Autowired
    private MyRepository repository;

    @Before
    public void before() {
       //clean repository
    }

    @Test
    public void test_1() {
       //add new entity
    }   

    @Test
    public void test_2() {
      //add new entity
    } 
    ...  
}

我想在所有测试之前回滚处于状态的数据库。表会回滚，但id生成的顺序会随着每次测试而增加。

请帮我找到在每次测试前设置起始值顺序的方法。

我正在使用Spring、Hibernate和HsqlDb

Answer 1

如果您使用@Transactional调用方法，您将无法进行回滚。您可以做的是使用@After以删除所有数据库，并在@前再次创建它。我认为这不是一个很好的主意，您应该删除这些值以隔离每个测试。我建议您使用H2数据库，以便能够在内存存储中使用，因此您不必担心id号。

必须将此依赖项添加到pom中：

<dependency>
    <groupId>com.h2database</groupId>
    <artifactId>h2</artifactId>
    <version>1.4.193</version>
</dependency>

以及内部的配置。xml应该是这样的：

<hibernate-configuration>
    <session-factory>
        <property name="hibernate.connection.driver_class">org.h2.Driver</property>
        <property name="hibernate.connection.username">sa</property>
        <property name="hibernate.connection.password"></property>
        <property name="hibernate.connection.url">jdbc:h2:mem:testDB;DATABASE_TO_UPPER=false;DB_CLOSE_DELAY=-1</property>
        <property name="hibernate.dialect">org.hibernate.dialect.H2Dialect</property>
        <property name="show_sql">false</property>
        <property name="hbm2ddl.auto">create</property>
    </session-factory>
</hibernate-configuration>

当然，你应该让它适应你的系统，但你必须保留用户名和密码，才能成功连接。

如果不想自动创建表，可以在连接URL上添加以下参数：

INIT=RUNSCRIPT FROM 'classpath:scripts/create.sql

这将在测试前运行您的脚本。

然后在你的测试中，你可以做一些链接：

@RunWith(SpringJUnit4ClassRunner.class)  
@ContextConfiguration(locations = {})
@Transactional (propagation = Propagation.REQUIRED)
public class MyTest {

    @Autowired
    private MyRepository repository;

    @Test
    public void test_1() {
       //add new entity
    }   

    @Test
    public void test_2() {
      //add new entity
    } 

    @After
    public void tearDown(){
      repository.clean();
    }
}

清洁方法可以是这样的

public void clean(){
    for (T obj: this.findall()) //replacing T for the type if you don't use generics
        session.delete(obj); //your entity manager or session, depending on how you do the queries
}