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("退出，资源释放");
    }
}