如何使用dbUnit将数据库恢复到初始状态?

时间:2010-09-28 14:17:00

标签: java unit-testing testing dbunit

我是自动化测试和dbUnit的新手。所以我很感激你的意见。

我打算创建一个测试套件,它将按以下方式运行:

  • 创建内存H2数据库
  • 运行DDL脚本以创建表格
  • 运行dbUnit以插入将由所有测试使用的初始数据(我们称之为 STATE0 )。
  • 运行测试

直到那里它对我来说很好看,但我不明白的是,如何在测试运行后将数据库恢复为 STATE0 并更改数据?

我可以用dbUnit做吗?
还是用其他东西?
我应该在每次测试之前重新创建数据库吗?

简单的不在测试中提交事务不适合我,因为测试最终会运行多个事务,可能会有多个数据库连接。

2 个答案:

答案 0 :(得分:7)

如果您正确编写@BeforeClass@Before@After方法,DBUnit可以自动完成四项工作。例如。在我们的项目中,使用Derby,一个这样的测试用例看起来像

public class MyTest {
    protected static IDataSet getDataSet() throws Exception {
        URL url = MyTest.class.getClassLoader().getResource("MyDataSet.xml");
        return new XmlDataSet(new FileInputStream(url.getPath()));
    }

    private static JdbcDatabaseTester databaseTester;

    @BeforeClass
    public static void setUpClass() throws Exception {
        // Init test environment, session etc.
        databaseTester = new JdbcDatabaseTester(
                "org.apache.derby.jdbc.ClientDriver",
                "jdbc:derby://localhost:1527/myschema", 
                "username", "password");
        databaseTester.setDataSet(getDataSet());
    }

    @AfterClass
    public static void tearDownClass() {
        // Close session etc.
    }

    @Before
    public void setUp() throws Exception {
        databaseTester.onSetup();
    }

    @After
    public void tearDown() throws Exception {
        databaseTester.onTearDown();
    }

    @Test
    public void test() throws Exception { ... }
}

此代码在每次测试后将数据库架构(的一个子集)放回MyDataSet.xml定义的状态。 (注意,正如@Pascal评论的那样,重置可能并不总是满的 - 如果测试修改了不在数据集中的表,它将不受@Before / @After方法的影响。)

答案 1 :(得分:6)

要将数据库初始化为初始数据集,只需在测试用例中实现这些方法:

@Override
protected DatabaseOperation getSetUpOperation() throws Exception
{
    return DatabaseOperation.CLEAN_INSERT; // by default (will do DELETE_ALL + INSERT)
}

@Override
protected DatabaseOperation getTearDownOperation() throws Exception
{
    return DatabaseOperation.NONE; // by default
}

如果某些测试在空表中插入行(例如,未在初始数据集中定义),则可能存在外键约束。

只需在数据集中添加此空表而不添加任何行:

<mydb_mypopulatedtable id="1" name="toto" alias="funky"/>
<mydb_mypopulatedtable id="2" name="titi" alias="groovy"/>
<mydb_mypopulatedtable id="3" name="tutu" alias="creepy"/>

<mydb_myemptytable />

在这里,myemptytable有mypopulatedtable的外键。如果未定义myemptytable,DBUnit将尝试删除mypopulatedtable但由于约束而失败。如果已定义,DBUnit将删除myemptytable行。