单元测试之Junit和spring集成

悠悠 2022-08-05 02:43 193阅读 0赞

使用 Spring 测试注释来进行常见的 Junit4 或者 TestNG 的单元测试，同时支持访问 Spring 的 beanFactory 和进行自动化的事务管理。  
**一、spring测试注解标签  
1.@ContextConfiguration 和 @Configuration 的使用**

Spring 3.0 新提供的特性 @Configuration，这个注释标签允许您用 Java 语言来定义 bean 实例。

package config; 
    
    import org.Springframework.beans.factory.annotation.Autowired; 
    import org.Springframework.context.annotation.Bean; 
    import org.Springframework.context.annotation.Configuration; 
    import org.Springframework.jdbc.datasource.DriverManagerDataSource; 
    
    import service.AccountService; 
    import service.Initializer; 
    import DAO.AccountDao; 
    
    @Configuration 
    public class SpringDb2Config { 
     private @Autowired DriverManagerDataSource datasource; 
     @Bean 
     public Initializer initer() { 
    	 return new Initializer(); 
     } 
    
     @Bean 
     public AccountDao accountDao() { 
    AccountDao DAO = new AccountDao(); 
    DAO.setDataSource(datasource); 
    return DAO; 
     } 
    
     @Bean 
     public AccountService accountService() { 
    return new AccountService(); 
     } 
    }

通过@ContextConfiguration指定配置文件，Spring test framework 会自动加载 XML 文件，也是我们采用的方式。  
**2.@DirtiesContext**  
缺省情况下，Spring 测试框架一旦加载 applicationContext 后，将一直缓存，不会改变，但是，  
由于 Spring 允许在运行期修改 applicationContext 的定义，例如在运行期获取 applicationContext，然后调用 registerSingleton 方法来动态的注册新的 bean，这样的情况下，如果我们还使用 Spring 测试框架的被修改过 applicationContext，则会带来测试问题，我们必须能够在运行期重新加载 applicationContext，这个时候，我们可以在测试类或者方法上注释：@DirtiesContext，作用如下：  
如果定义在类上（缺省），则在此测试类运行完成后，重新加载 applicationContext  
如果定义在方法上，即表示测试方法运行完成后，重新加载 applicationContext  
**3.@Transactional、@TransactionConfiguration 和 @Rollback**  
缺省情况下，Spring 测试框架将事务管理委托到名为 transactionManager 的 bean 上，如果您的事务管理器不是这个名字，那需要指定 transactionManager 属性名称，还可以指定 defaultRollback 属性，缺省为 true，即所有的方法都 rollback，您可以指定为 false，这样，在一些需要 rollback 的方法，指定注释标签 @Rollback（true）即可。事务的注解可以具体到方法  
**4.@Repeat**  
通过 @Repeat，您可以轻松的多次执行测试用例，而不用自己写 for 循环，使用方法：  
 @Repeat(3)   
 @Test(expected=IllegalArgumentException.class)   
 public void testInsertException() \{   
service.insertIfNotExist(null);   
 \}  
这样，testInsertException 就能被执行 3 次。  
**5.@ActiveProfiles **  
从 Spring 3.2 以后，Spring 开始支持使用 @ActiveProfiles 来指定测试类加载的配置包，比如您的配置文件只有一个，但是需要兼容生产环境的配置和单元测试的配置，那么您可以使用 profile 的方式来定义 beans，如下：

上面的定义，我们看到：  
在 XML 头中我们引用了 Spring 3.2 的 beans 定义，因为只有 Spring 3.2+ 才支持基于 profile 的定义  
在 <beans> 根节点下可以嵌套 <beans> 定义，要指定 profile 属性，这个配置中，我们定义了两个 datasource，一个属于 test profile，一个输入 production profile，这样，我们就能在测试程序中加载 test profile，不影响 production 数据库了  
在下面定义了一些属于两个 profile 的 beans，即 <beans profile=”test,production”> 这样方便重用一些 bean 的定义，因为这些 bean 在两个 profile 中都是一样的

@RunWith(SpringJUnit4ClassRunner.class) 
     @ContextConfiguration("/config/Spring-db.xml") 
     @Transactional 
     @ActiveProfiles("test") 
     public class AccountServiceTest { 
     ... 
     }

注意上面的 @ActiveProfiles，可以指定一个或者多个 profile，这样我们的测试类就仅仅加载这些名字的 profile 中定义的 bean 实例。

下面看一个配置实例：

添加依赖：

<dependency>
    			<groupId>junit</groupId>
    			<artifactId>junit</artifactId>
    			<version>4.10</version>
    		</dependency>
    		<dependency>
    			<groupId>org.springframework</groupId>
    			<artifactId>spring-test</artifactId>
    			<version>4.0.1.RELEASE</version>
    		</dependency>
    		<dependency>
    			<groupId>org.springframework</groupId>
    			<artifactId>spring-context</artifactId>
    			<version>4.0.1.RELEASE</version>
    		</dependency>
    		<dependency>
    			<groupId>org.springframework</groupId>
    			<artifactId>spring-tx</artifactId>
    			<version>4.0.1.RELEASE</version>
    		</dependency>

spring配置：

applicationContext.xml

<?xml version="1.0" encoding="UTF-8"?>
    <beans xmlns="http://www.springframework.org/schema/beans"
           xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
           xmlns:context="http://www.springframework.org/schema/context"
           xsi:schemaLocation="http://www.springframework.org/schema/beans
    						http://www.springframework.org/schema/beans/spring-beans-3.0.xsd 
    						http://www.springframework.org/schema/context 
    						http://www.springframework.org/schema/context/spring-context-3.0.xsd"
           default-lazy-init="false">
    
    	<description>Spring公共配置 </description>
    	<context:component-scan base-package="cn.slimsmart.unit.test.demo" />
    </beans>

AddServiceImpl添加@service注解

测试类：

package cn.slimsmart.unit.test.demo.junit;
    
    import static org.junit.Assert.assertTrue;
    
    import org.junit.After;
    import org.junit.Before;
    import org.junit.Test;
    import org.junit.runner.RunWith;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.test.context.ActiveProfiles;
    import org.springframework.test.context.ContextConfiguration;
    import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
    import org.springframework.test.context.transaction.TransactionConfiguration;
    import org.springframework.transaction.annotation.Transactional;
    
    //用来说明此测试类的运行者，这里用了 SpringJUnit4ClassRunner
    @RunWith(SpringJUnit4ClassRunner.class) 
    //指定 Spring 配置信息的来源，支持指定 XML 文件位置或者 Spring 配置类名
    @ContextConfiguration(locations = { "classpath*:/applicationContext.xml" })
    //表明此测试类的事务启用，这样所有的测试方案都会自动的 rollback,
    //@Transactional
    //defaultRollback，是否回滚，默认为true
    //transactionManager：指定事务管理器一般在spring配置文件里面配置
    //@TransactionConfiguration(defaultRollback=true,transactionManager="transactionManager")
    //但是需要兼容生产环境的配置和单元测试的配置，那么您可以使用 profile 的方式来定义 beans，
    //@ActiveProfiles("test")
    public class JunitSpringTest {
    	
    	@Autowired
    	private AddService addService;
    	
    	@Before
    	public void setUp(){
    		System.out.println("初始化");
    	}
    	
    	@Test
    	public void testAdd() {
    		assertTrue(addService.add(1, 1) == 2);
    	}
    	
    	@After
    	public void destroy() {
    		System.out.println("退出，资源释放");
    	}
    }