SpringBoot单元测试,@Rollback事务回滚避免脏数据

单元测试应该遵循AIR原则，避免产生脏数据是一项基本要求。

junit使用回滚注解@Rollback，配合事务注解@Transactional，来回滚事务。

RollBack可以定义在测试类上，对整个类起作用，也可以对方法单独定义。

定义在类上，这样每个方法默认都是要回滚的。方法执行完单元测试，数据库不会产生脏数据。

@SpringBootTest
@Transactional
@Rollback
public class UserServiceTest {
    @Test
	void testInserUser() {
        //向数据库增加一条用户信息
		inserUser(User user);	
	}

   @Test
	void testDeleteUser() {
        //删除一条用户信息
		deleteUser(int userId);	
	}

}

定义在方法上，这样对每个方法单独生效。

    @Test
	@Rollback
	void testMethod() {
		//向数据库增加一条用户信息
		inserUser(User user);		
	}

如果测试的时候想要产生真实数据，可以将@Rollback设置为无效

    @Test
	@Rollback(false)
	void testMethod() {
		//向数据库增加一条用户信息
		inserUser(User user);		
	}

对于整个项目应该怎样避免脏数据呢？可以新建一个基础测试类BaseTest.java，默认事务都是要回滚的，其他测试类都继承这个类。

BaseTest.java

@SpringBootTest
@Transactional
@Rollback
public class BaseTest {

}

UserServiceTest.java 继承BaseTest.java

@SpringBootTest
public class UserServiceTest extends BaseTest {
    @Test
	void testInserUser() {
        //向数据库增加一条用户信息
		inserUser(User user);	
	}

   @Test
	void testDeleteUser() {
        //删除一条用户信息
		deleteUser(int userId);	
	}
}

如果测试某个类或者方法时不需要回滚，只需要单独的显示设置@Rollback(false)就可以了。

UserServiceTest .java 增加@Rollback(false)，每个方法都回滚失效

@SpringBootTest
@Rollback(false)
public class UserServiceTest extends BaseTest {
    @Test
	void testInserUser() {
        //向数据库增加一条用户信息
		inserUser(User user);	
	}

   @Test
	void testDeleteUser() {
        //删除一条用户信息
		deleteUser(int userId);	
	}
}

UserServiceTest .java 的方法testInserUser() 增加@Rollback(false)，这个方法回滚失效。其他方法依然继承父类BaseTest的默认设置@Rollback有效。

@SpringBootTest
public class UserServiceTest extends BaseTest {
    @Test
    @Rollback(false)
	void testInserUser() {
        //向数据库增加一条用户信息
		inserUser(User user);	
	}

   @Test
	void testDeleteUser() {
        //删除一条用户信息
		deleteUser(int userId);	
	}
}

测试完成后，一定记得将@Rollback(false)注释掉。