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