Question

我有一个spock / spring测试，它修改了一些数据库内容，我想知道如何回滚该记录。

目前我执行原始sql语句，保存字段数据，并在成功测试后恢复该数据。但我的猜测是，这可以通过更简单的方式完成吗？

@ContextConfiguration(locations = "classpath*:applicationContext-test.xml")
class RepositoryTest extends Specification {

    @Shared sql = Sql.newInstance("jdbc:sqlserver://...")
    ResourceRepository resourceRepository;

    def "Save test"() {
        setup:
        // copy the row before we do changes! we need to restore this later on!
        def row = sql.firstRow("select id, content, status from table-name where id = 12345")

        when:
        ...

        then:
        ..

        cleanup:
        sql.execute("""
                    update table-name set content = ${row.content}, status = ${row.status}
                    where id = ${row.id}
                    """)
    }
}

Answer 1

我发现这样做的最好方法是：

开始测试
启动交易
（可选）使用DBUnit
运行测试
回滚交易

请注意，所有数据库操作都发生在同一事务中。因为此事务在测试结束时回滚（即使抛出异常），所以数据库在测试结束时应始终处于与开始时相同的状态。

Spring提供了一些非常有用的类，它们将负责为每个测试启动和回滚事务。如果你正在使用Spring＆amp; JUnit4并不介意你的测试类必须扩展Spring类，最简单的选择可能是扩展AbstractTransactionalJUnit4SpringContextTests

// Location of the Spring config file(s)
@ContextConfiguration(locations = {"/application-context-test.xml", "classpath*:/application-context-persistence-test.xml"})

// Transaction manager bean name
@TransactionConfiguration(transactionManager = "hsqlTransactionManager", defaultRollback = true)
@Transactional(propagation=Propagation.REQUIRES_NEW)
public class MyTransactionalTests extends AbstractTransactionalJUnit4SpringContextTests {

    @Test
    public void thisWillRunInATransactionThatIsAutomaticallyRolledBack() {}
}

如果您不想扩展Spring类，可以使用注释来配置测试运行器。 Spring还支持许多其他主要的单元测试框架和旧版本的JUnit。

Answer 2

如果你在测试中将Groovy的Sql与Spring的注释（@ RunWith，@ ContextConfiguration，@ Transaction，@ Rollback，...）混合在一起，你可能想要包装带有org.springframework.jdbc.datasource.TransactionAwareDataSourceProxy的dataSource。

<bean id="db-dataSourceReal" 
   class="org.springframework.jndi.JndiObjectFactoryBean">
  <property name="jndiName" value="jdbc/foo" />
  <property name="resourceRef" value="true" />
  <property name="lookupOnStartup" value="true" />
</bean>

<bean id="db-dataSource"
   class="org.springframework.jdbc.datasource.TransactionAwareDataSourceProxy">
  <constructor-arg ref="db-dataSourceReal" />
</bean>

然后使用TransactionAwareDataSourceProxy作为Groovy Sql的dataSource。例如，在Test类中（在本例中使用maven的故障安全插件），

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations=[
        "classpath*:applicationContext-test.xml"
])
class SimplePojoDaoIT {

    @Resource(name="dao-pojoDao")
    PojoDao pojoDao

    @Test
    @Transactional("transactionManager")
    @Rollback
    public void testShouldPersistToDB(){

      SomePojo pojo = new SomePojo()
      pojo.with{
        id = 5
        name = 'foo'
      }

      pojoDao.persist(pojo)

      def sql = new Sql(pojoDao.dataSource)
      sql.rows("select * from POJO_TBL where id = :id", [['id':pojo.id]]).each{ row ->
//      println row
        pojo.with{
          assertEquals(id, row.ID.longValue())
          assertEquals(name, row.NAME)
        }
      }
    }
}

Answer 3

CREATE TABLE table_name
(
   id        NUMBER,
   content   NUMBER,
   status    NUMBER
);

INSERT INTO table_name
     VALUES (1, 2, 3);

INSERT INTO table_name
     VALUES (4, 5, 6);

INSERT INTO table_name
     VALUES (7, 8, 9);

COMMIT;

在运行测试之前，将此SELECT产生的字符串存储在VARCHAR2（4000）变量中，测试后只需执行字符串：

SELECT    'UPDATE TABLE_NAME SET CONTENT = '
       || CONTENT
       || ', STATUS = '
       || STATUS
       || ' WHERE ID = '
       || ID
  FROM TABLE_NAME
 WHERE ID = 1;

在我之前的例子中，我认为备份记录的ID = 1。在此示例中，字符串包含以下UPDATE语句：

UPDATE TABLE_NAME SET CONTENT = 2, STATUS = 3 WHERE ID = 1

如何在测试后回滚特定记录？

3 个答案: