单元测试之Junit和spring集成

使用 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，如下：

<beans xmlns="http://www.Springframework.org/schema/beans"
	 xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
 xsi:schemaLocation="http://www.Springframework.org/schema/beans 
 http://www.Springframework.org/schema/beans/Spring-beans-3.2.xsd"> 
 	 <beans profile="test"> 
 <bean id="datasource" 
 class="org.Springframework.jdbc.datasource.DriverManagerDataSource"> 
 <property name="driverClassName" value="org.hsqldb.jdbcDriver" /> 
	 <property name="url" value="jdbc:hsqldb:hsql://localhost" /> 
	 <property name="username" value="sa"/> 
	 <property name="password" value=""/> 
	 </bean> 
</beans> 
	
<beans profile="production"> 
 <bean id="datasource" 
 class="org.Springframework.jdbc.datasource.DriverManagerDataSource"> 
	 <property name="driverClassName" value="org.hsqldb.jdbcDriver" /> 
	 <property name="url" value="jdbc:hsqldb:hsql://localhost/prod" /> 
	 <property name="username" value="sa"/> 
	 <property name="password" value=""/> 
	 </bean> 
 </beans> 
 <beans profile="test,production"> 
 <bean id="transactionManager" 
     class="org.Springframework.jdbc.datasource.DataSourceTransactionManager"> 
	 <property name="dataSource" ref="datasource"></property> 
	 </bean> 
	 <bean id="initer" init-method="init" class="service.Initializer"> 
	 </bean> 
 <bean id="accountDao" depends-on="initer" class="DAO.AccountDao"> 
	 		 <property name="dataSource" ref="datasource"/> 
	 	 </bean> 
	 	
	 	 <bean id="accountService" class="service.AccountService"> 
	 	 </bean> 
	 	 <bean id="envSetter" class="EnvSetter"/> 
 	 </beans> 
 </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("退出，资源释放");
	}
}