JUnit —— TestSuite 的使用

首先说一下，suite ，中文是 一套，一组 的意思。

那么，TestSuite，顾名思义，就是用来运行一组测试的。

如何使用呢 ？

假设有个这样的测试类 StudentDAOTest ，代码如下：

package org.young.junit.testsuite;

import junit.framework.TestCase;

import org.young.junit.testsuite.dao.StudentDAO;
import org.young.junit.testsuite.dao.StudentDAOImpl;
import org.young.junit.testsuite.entity.Student;

public class StudentDAOTest extends TestCase {
    
    private StudentDAO dao;
    
    /**
     * 创建 DAO 实例
     */
    public void setUp() {
        dao = new StudentDAOImpl();
    }

    public void testAdd() {
        Student stu = new Student();
        
        dao.add(stu);
    }

    public void testDelete() {
        
        dao.delete("id");
    }

    public void testUpdate() {
        Student stu = new Student();
        
        dao.update(stu);
    }

    public void testLoadWithId() {
        
        Student stu = dao.load("xyz");
        
        assertNotNull(stu);
    }
    
    public void testLoadWithNullOrEmptyStr() {
        
        Student stu = dao.load("");
        assertNull(stu);
        
        stu = dao.load(null);
        assertNull(stu);
    }

}

 如果想一次执行几个方法，而不是所有方法改怎么办呢？

 TestSuite 该上场了。

为了方便比较，再来一个测试类 CourseDAOTest ，代码如下：

package org.young.junit.testsuite;

import org.junit.Before;
import org.junit.Test;
import org.young.junit.testsuite.dao.CourseDAO;
import org.young.junit.testsuite.dao.CourseDAOImpl;
import org.young.junit.testsuite.entity.Course;

import junit.framework.TestCase;
import junit.framework.TestSuite;

/**
 * Suite 的实现方式一
 * 
 *     public static Test suite(){} 的方式
 *
 * @author by Young.ZHU
 *        on 2013-9-30
 *
 * Package&FileName: org.young.junit.testsuite.CourseDAOTest
 */
public class CourseDAOTest extends TestCase {
    
    private CourseDAO dao;
    
    public CourseDAOTest() {
        super();
    }

    public CourseDAOTest(String name) {
        super(name);
    }

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        
        dao = new CourseDAOImpl();
    }
    
    /**
     * 注意：继承 TestCase 后，JUnit 4 里的 @Before 、@Test 等注解就没用了
     * 
     * @Before 的功能可由方法 setUp() 实现
     */
    @Before
    public void init() {
        System.out.println("fsdfsdf");
        dao = new CourseDAOImpl();
    }
    
    /**
     * 执行这个测试类的部分方法
     * 
     * 方法头必须是这样的 public static junit.framework.Test suite()
     * 即，静态（static） 的
     * 
     * @return
     */
    public static junit.framework.Test suite() {
        TestSuite suite = new TestSuite();
        
        /*
         * 字符串参数为想要执行的该测试类的方法
         */
        suite.addTest(new CourseDAOTest("testLoad"));
        suite.addTest(new CourseDAOTest("testAdd"));
        
        return suite;
    }

    @Test
    public void testAdd() {
        Course course = new Course();
        
        dao.add(course);
    }

    @Test
    public void testDelete() {
        fail("Not yet implemented");
    }

    @Test
    public void testUpdate() {
        fail("Not yet implemented");
    }

    @Test
    public void testLoad() {
        Course course = dao.load("course_id");
        
        assertNotNull(course);
    }

}

先运行一下，看下效果：

虽然这个测试类写了增（add）、删（delete）、改（update）、查（load），但实际执行的只有两个方法 —— testLoad 和 testAdd 。

秘密就在于代码第 60 行的 suite() 方法，这个方法决定了该测试类执行哪些方法。

有两点需要说明：

1、关于方法 suite() 的方法头

正如注释里写道的，这个方法的方法头是固定的 

public static junit.framework.Test suite() {
       // your code ...  
}

2、测试类的构造方法

测试类 CourseDAOTest  中第 30 行带参数的构造函数，在 66 行和 67 行用到了。

构造函数的参数即要执行的测试方法的名称。

最后，把两个集合起来看，测试类 AllTest ，代码如下：

package org.young.junit.testsuite;

import junit.framework.TestSuite;


public class AllTest {

    public static junit.framework.Test suite() {
        TestSuite suite = new TestSuite("All Test");
        
        /*
         * StudentDAOTest 类的全部测试方法
         */
        suite.addTest(new TestSuite(StudentDAOTest.class));
        /*
         * CourseDAOTest 类的部分方法
         */
        suite.addTest(CourseDAOTest.suite());
        
        return suite;
    }

}

运行后，效果如下：

详细代码可参考：

https://github.com/YoungZHU/CollectionCode4Java/tree/master/test/org/young/junit/testsuite