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