在进行数据访问的单元测试,一般都会新增数据记录的情况,那么在设置数据集一般会是:
<?xml version="1.0" encoding="UTF-8"?>
<dataset>
</dataset>
然后在测试方法中,通过验证新增的记录数或者新增记录的内容,确保被测试方法正确执行:
@Test
@DatabaseSetup("user/UserService-addUser-dataset.xml")
public void testAddUser(){
User user = new User();
userService.addUser(user);
List<User> userList = (List<User>)userDao.findAll();
assertEquals(1, userList.size());
}
上面的测试方法,在多数情况会正常运行,不会出现问题,但是却留下一个隐患:如果同时存在包含了user的数据集,验证记录数的断言很可能会失败。比如,存在一个测试中使用的数据集:
<?xml version="1.0" encoding="UTF-8"?>
<dataset>
<user id="1" name="Test One"/>
<user id="2" name="Test Two"/>
</dataset>
这是testAddUser方法很可能出现"Expect: 1,Actual: 3"的失败信息。要彻底解决这个问题也很简单,只需要将会新增数据的表设置为空就行了:
<?xml version="1.0" encoding="UTF-8"?>
<dataset>
<user/>
</dataset>
问题的原因:SpringTestDbunit在执行测试之前会检查DatabaseSetup注解,对数据集中指定的表进行CLEAN_INSERT操作,而数据集中没有提到的表则会保持不变。所以要想彻底清空一个表,必须要指定一个空记录。