单元测试应该遵循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)注释掉。

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐