JUNIT4 简单入门

JUnit 4是JUnit框架有史以来的最大改进，其主要目标便是利用Java 5的Annotation特性简化测试用例的编写。让我们看看如何使用JUnit 4来进行Unit测试。

请注意：本文主要介绍JUnit 4的最新特性和用法，并不会讲解Unit测试的基础。如果你对JUnit框架还不熟悉，请先参考“Eclipse快速上手指南 (2) 使用JUnit测试”一文，学习如何编写JUnit测试。

我们使用的开发环境是Eclipse 3.2，它已经自带了JUnit 4.1，你需要将JUnit 4 Library添加到项目用到的Library中。另外，必须使用JDK 5.0或更高版本。

要在Eclipse环境之外运行JUnit，需要下载JUnit 4.1，具体请访问JUnit.org。

我们先看一个简单的Math类：

 

package  com.javaeedev.junit4;
public class  Math {

    public int abs(int  value) {
        return value>=0 ? value : (- value);
    }

    public int div(int a, int  b) {
        return a /  b;
    }

    /**
     * BUG: if b less than 0!
      */
     public float exp(int a, int  b) {
        float r = 1 ;
        for(int i=0; i<b; i++ )
            r = r *  a;
        return  r;
    }
}

 

注意exp()方法是有Bug的，如果传入参数2, -1，则期待的返回值应为0.5F，但实际返回值为1.0F。

下面我们看看传统的JUnit的TestCase：

 

public class MathTest extends  TestCase {

    public void setUp() { super .setUp(); }
    public void tearDown() { super .tearDown(); }

    public void testAbs() { assertTrue(true ); }
    public void  testDiv() {...}
    public void  testExp() {...}

}

 

JUnit依赖反射来执行每个以test开头的方法。然而，在最新的JUnit 4中，由于有了Annotation的支持，我们的测试方法不需要再以testXxx标识了，而是写上一个@Test标注即可。例如：

 

@Test public void doAbs() {...}

 

甚至MathTest类也不必继承自TestCase。你也许会想到，不继承自TestCase就无法调用assertXxx方法了，正因为如此，所有的assertXxx方法全部以静态方法被放入了Assert类，使用Assert.assertXxx()调用。如果使用

 

import static org.junit.Assert.*;

 

则原有的代码不必改动。

setUp()和tearDown()方法也依赖@Before和@After标记，这样做的最大的好处是在继承体系内不必担心忘记了在setUp()方法中调用父类的super.setUp()方法，JUnit框架会自动处理父类的@Before和@After标记的方法。

并且，JUnit框架对@Before和@After的调用顺序类似于类的构造方法和析构方法，即@Before按照父类到子类的顺序调用，@After则相反，这样保证了资源的正确获取和释放。

当然，不再强迫必须使用setUp和tearDown作为方法名，可以使用更有意义的方法名，例如：initDatabase()和closeDatabase()，只要它们被标注了@Before和@After即可。

来看看使用Annotation的MathTest：

 

package  com.javaeedev.junit4;

import static org.junit.Assert.* ;

import org.junit.* ;

public class  MathTest {

    public  MathTest() {
        System.out.println("new MathTest instance." );
    }

    @Before
    public void setUp() throws  Exception {
        System.out.println("call @Before before a test method" );
    }

    @After
    public void tearDown() throws  Exception {
        System.out.println("call @After after a test method" );
    }

    @Test
    public void  doAbs() {
        Math math = new  Math();
        assertEquals(200, math.abs(200 ));
        assertEquals(100, math.abs(-100 ));
        assertEquals(0, math.abs(0 ));
    }

    @Test
    public void  doDiv() {
        Math math = new  Math();
        assertEquals(5, math.div(100, 20 ));
        assertEquals(4, math.div(100, 21 ));
    }

    @Test(expected=ArithmeticException.class )
    public void  doDiv0() {
        new Math().div(127, 0 );
    }

    @Test(timeout=1 )
    public void  doLongTimeTask() {
        double d = 0 ;
        for(int i=1; i<10000000; i++ )
            d+= i;
    }

    @Test
    public void  testExp() {
        Math math = new  Math();
        assertEquals(32f, math.exp(2, 5), 0.001f );
        assertEquals(1f, math.exp(2, 0), 0.001f );
        assertEquals(0.5f, math.exp(2, (-1)), 0.001f );
    }

}

 

对测试异常，JUnit 4可以用expected=Exception.class来期待一个预期的异常，而不必编写

 

try  {
  ...
  fail("No exception" );
}
catch (Exception e) {
  // OK!
}

 

来看看doDiv0测试，我们期待一个除数为0的ArithmeticException，因此编写如下测试方法：

 

@Test(expected=ArithmeticException.class )
public void  doDiv0() {
    new Math().div(127, 0 );
}

 

对于非常耗时的测试，@Test还有一个timeout来标识该方法最长执行时间，超过此时间即表示该测试方法失败：

 

@Test(timeout=1 )
public void  doLongTimeTask() {
    double d = 0 ;
    for(int i=1; i<10000000; i++ )
        d+= i;
}

 

以上方法若执行时间超过1ms则测试失败，由于依赖CPU的执行速度，在不同的机器上测试结果也不同。

JUnit 4另一个较大的变化是引入了@BeforeClass和@AfterClass，它们在一个Test类的所有测试方法执行前后各执行一次。这是为了能在@BeforeClass中初始化一些昂贵的资源，例如数据库连接，然后执行所有的测试方法，最后在@AfterClass中释放资源。

正如你能想到的，由于@BeforeClass和@AfterClass仅执行一次，因此它们只能标记静态方法，在所有测试方法中共享的资源也必须是静态引用：

 

private static  Object dbConnection;

@BeforeClass
public static void setUpBeforeClass() throws  Exception {
    System.out.println("call @BeforeClass and init database connection" );
    dbConnection = new  Object();
}

@AfterClass
public static void tearDownAfterClass() throws  Exception {
    System.out.println("call @AfterClass to release database connection" );
    dbConnection = null ;
}

 

最后执行测试用例，可以看到结果：

各个方法执行顺序如下：

call @BeforeClass and init database connection

new MathTest instance.
call @Before before a test method
call @After after a test method

new MathTest instance.
call @Before before a test method
call @After after a test method

...

call @AfterClass to release database connection

可以看到，@BeforeClass是在实例化MathTest之前调用的，因此不能在构造方法中初始化共享资源。

最后需要注意的是由于Java 5的自动Box/Unbox特性，在调用assertEquals()时要特别注意，如果你传入：

assertEquals(100F, 100);

则按照自动Box变为：

assertEquals(new Float(100F), new Integer(100));

测试失败，因为Float类和Integer类不是同一类型。

因此要特别注意float和double的测试。事实上对float和double应使用

assertEquals(float, float, float delta);
assertEquals(double, double, double delta);

delta指定了两个作比较的浮点数的相差范围，在此范围内的两个浮点数将认为相等。可以传入一个很小的数例如0.0001F。 

JUnit 4非常适合使用Java 5的开发人员，但是无法在Java 1.4中获得这些好处，并且，也不与以前的版本兼容。因此，如果你正在使用Java 5，就可以考虑使用JUnit 4来编写测试。