Java单元测试Junit（五）使用Spring TestContext测试Spring应用

什么是Spring TestContext? 

      Spring TestContext是Spring提供的一套基于注解的Test框架,Spring TestContext有非常好的兼容性,可以无缝兼容JUnit,TestNG等单元测试框架,而且在其基础上增加更多的功能 

Spring TestContext为我们带来什么? 

      在Spring应用大行其道的今天,使用Spring来构建应用已经是再普通不过的事情,但当使用JUnit,TestNG等传统的单元测试技术和Spring进行结合的时候,就会出现很多并不如意的事情 

      1.Spring容器初始化问题:此问题已于JUnit4中被解决,通过使用@BeforeClass 

         可以有效防止Spring容器被多次初始化的问题 

      2.硬编码获取Bean:此问题是由于JUnit并不兼容Spring,所以当单元测试运行 

         的时候,无法解释Spring独有的注解,从而需要使用硬编码来获取Bean 

      3.数据现场破坏:JUnit当中可以使用DBUnit来进行数据现场维护的解决方案, 

         详情可以通过笔者的DBUnit使用了解DBUnit,Spring TestContext通过AOP 

         声明式事务来对单元测试进行回滚,有效的解决了数据现场的问题 

      4.事务:通常我们的单元测试都需要和数据库进行交互,但传统的JUnit的 

        组成单元为TestCase,并不存在事务的概念,而我们大多数情况下都需要 

        观察事务的执行过程或总体的性能,特别是对长事务模块的测试, 

        Spring TestContext允许单元测试支持事务的控制 

Spring TestContext使用: 

      首先为Maven增加Spring的依赖,由于Spring有众多依赖,这里就不给出代码了 

      以下为Spring TestContext的代码 

package com.accentrix.ray;  
 
import org.junit.After;  
import org.junit.Before;  
import org.junit.Test;  
import org.springframework.beans.factory.annotation.Autowired;  
import org.springframework.test.annotation.ExpectedException;  
import org.springframework.test.annotation.Repeat;  
import org.springframework.test.annotation.Rollback;  
import org.springframework.test.annotation.Timed;  
import org.springframework.test.context.ContextConfiguration;  
import org.springframework.test.context.junit4.AbstractJUnit4SpringContextTests;  
import org.springframework.test.context.transaction.AfterTransaction;  
import org.springframework.test.context.transaction.BeforeTransaction;  
 
import com.accentrix.vtc.hpp.entity.planning.User;  
import com.accentrix.vtc.hpp.unit.dao.UserDao;  
 
//指定Spring配置文件位置  
@ContextConfiguration(locations = "/applicationContext-test.xml")  
/*  
 *  配置事务是使用哪个事务管理器和默认是否回滚,通常继承  
 *  AbstractTransactionalJUnit4SpringContextTests后不需配置  
 *  @TransactionConfiguration(defaultRollback = true,transactionManager="test")  
 */ 
public class SpringTest extends AbstractTransactionalJUnit4SpringContextTests{  
 
    // 可以直接使用Spring Annotation的方式注入  
    @Autowired 
    private UserDao userDao;  
 
    private User user;  
 
    // 当有某些操作不想在包含在事务,希望在事务前执行时,使用@BeforeTransaction  
    @BeforeTransaction 
    public void beforeTransaction() {  
        System.out.println("Hello ");  
    }  
 
    /*  
     * 由于继承了AbstractTransactionalJUnit4SpringContextTests,  
     * 此时JUnit就拥有了事务控制,@Before和@After都会包含在事务当中  
     */ 
    @Before 
    public void setUp() {  
        System.out.println(userDao);  
 
        // 在开始前先插入测试数据  
        user = new User(1, "Ray", "123");  
        userDao.add(user);  
        /*  
         * 此时使用Spring Test的就需要注意,因为 Hibernate会将HQL或SQL语句进行缓存,所以  
         * 这里若然DAO是使用Hibenrate,需要使用flush  
         */ 
        userDao.getSession().flush();  
    }  
 
      
    @Test 
    @ExpectedException(RuntimeException.class) // ①  
    @Timed(millis = 1000) // ②  
    @Repeat(100) // ③  
    @Rollback(false) // ④  
    public void testMain() {  
        throw new RuntimeException();  
    }  
 
      
    @After 
    public void tearDown() {  
        // 最后删除测试数据  
        userDao.delete(user);  
    }  
 
    // 当有某些操作不想在包含在事务,希望在事务后执行时,使用@AfterTransaction  
    @AfterTransaction 
    public void afterTrasaction() {  
        System.out.println("World");  
    }  
}  
 

      请留意上面代码中testMain上的4个注解 

@ExpectedException:此形式和@Test(expected=RuntimeException.class)一样,表示该单元测试应该抛出RuntimeException的错误,否者测试不通过,但建议使用JUnit提供的 

@Timed:Timed和@Test(timeout=1000)有所不同的,@Test(timeout=1000)代表被注解的方法单体执行超过1秒就会测试失败,但@Timed()为测试整个事务从开始到执行testMain再结束的总耗时,而且会把@Repeat的重复也计算在总耗时内 

@Repeat:Repeat代表此方法将会重复执行100次,注意,会重复执行@Before和@After方法 

@Rollback:这里是Spring TestContext我认为最大的魅力所在,只需一行简单的注解就可以实现控制回滚操作,此时可以覆盖基类中回滚默认为true的限制,可以控制是否回滚事务, true为回滚,false为不回滚 

@NotTransaction:请注意这个注解,在2.5版本此注解为声明该单元测试不需要事务支持,但在3.0版本后认为单元测试当中应该都要有事务控制,所以此注解已被Deprecated(不建议再使用) 

      除此以外Spring TestContext还提供了很多方便的操作让我们使用 

public void testMain() {  
        //获取ApplicationContext对象  
        ApplicationContext ac = super.applicationContext;  
          
        //获取SimpleJdbcTemplate实现  
        SimpleJdbcTemplate  simpleJdbcTemplate= super.simpleJdbcTemplate;  
          
        //获取某table的总行数  
        super.countRowsInTable("t_user");  
          
        //执行某份外部的sql文件,true为当发生错误时仍然继续执行  
        super.executeSqlScript("D:/execute.sql", true);  
          
        //快速删除某张表的全部记录  
        super.deleteFromTables("t_test");  
    }  
 

Spring
TestContext使用极其简单,通过简单的注解就能扩展JUnit单元测试,弥补了JUnit在测试Spring应用时的不足和空白,还可以使用Spring自身的注解对依赖进行注入,简化了此前每次获取Bean的烦恼,在Spring应用中请毫不犹疑的使用Spring Test,在Spring TestContext当中除了支持JUnit4以外,还支持JUnit3.8,TestNG等,只需将单元测试类分别继承AbstractTransactionalJUnit38SpringContextTests和AbstractTransactionalTestNGSpringContextTests即可,