在Eclipse中使用JUnit4進(jìn)行單元測(cè)試(3)

作者：佚名 2009-06-08 20:04:06

本篇文章講述在Eclipse中使用JUnit4進(jìn)行單元測(cè)試(3)

上一篇文章中我們介紹了兩個(gè)Fixture標(biāo)注，分別是@Before和@After，我們來(lái)看看他們是否適合完成如下功能：有一個(gè)類是負(fù)責(zé)對(duì)大文件（超過(guò)500兆）進(jìn)行讀寫，他的每一個(gè)方法都是對(duì)文件進(jìn)行操作。換句話說(shuō)，在調(diào)用每一個(gè)方法之前，我們都要打開一個(gè)大文件并讀入文件內(nèi)容，這絕對(duì)是一個(gè)非常耗費(fèi)時(shí)間的操作。如果我們使用@Before和@After，那么每次測(cè)試都要讀取一次文件，效率及其低下。這里我們所希望的是在所有測(cè)試一開始讀一次文件，所有測(cè)試結(jié)束之后釋放文件，而不是每次測(cè)試都讀文件。JUnit的作者顯然也考慮到了這個(gè)問(wèn)題，它給出了@BeforeClass 和 @AfterClass兩個(gè)Fixture來(lái)幫我們實(shí)現(xiàn)這個(gè)功能。從名字上就可以看出，用這兩個(gè)Fixture標(biāo)注的函數(shù)，只在測(cè)試用例初始化時(shí)執(zhí)行@BeforeClass方法，當(dāng)所有測(cè)試執(zhí)行完畢之后，執(zhí)行@AfterClass進(jìn)行收尾工作。在這里要注意一下，每個(gè)測(cè)試類只能有一個(gè)方法被標(biāo)注為@BeforeClass 或 @AfterClass，并且該方法必須是Public和Static的。

二、限時(shí)測(cè)試。

還記得我在初級(jí)篇中給出的例子嗎，那個(gè)求平方根的函數(shù)有Bug，是個(gè)死循環(huán)：

public void squareRoot(int n) ...{

for (; ;) ; //Bug : 死循環(huán)

}

如果測(cè)試的時(shí)候遇到死循環(huán)，你的臉上絕對(duì)不會(huì)露出笑容。因此，對(duì)于那些邏輯很復(fù)雜，循環(huán)嵌套比較深的程序，很有可能出現(xiàn)死循環(huán)，因此一定要采取一些預(yù)防措施。限時(shí)測(cè)試是一個(gè)很好的解決方案。我們給這些測(cè)試函數(shù)設(shè)定一個(gè)執(zhí)行時(shí)間，超過(guò)了這個(gè)時(shí)間，他們就會(huì)被系統(tǒng)強(qiáng)行終止，并且系統(tǒng)還會(huì)向你匯報(bào)該函數(shù)結(jié)束的原因是因?yàn)槌瑫r(shí)，這樣你就可以發(fā)現(xiàn)這些Bug了。要實(shí)現(xiàn)這一功能，只需要給@Test標(biāo)注加一個(gè)參數(shù)即可，代碼如下：

    @Test(timeout = 1000)    public void squareRoot() ...{
        calculator.squareRoot(4);
        assertEquals(2, calculator.getResult());
    }

Timeout參數(shù)表明了你要設(shè)定的時(shí)間，單位為毫秒，因此1000就代表1秒。

三、測(cè)試異常

JAVA中的異常處理也是一個(gè)重點(diǎn)，因此你經(jīng)常會(huì)編寫一些需要拋出異常的函數(shù)。那么，如果你覺得一個(gè)函數(shù)應(yīng)該拋出異常，但是它沒拋出，這算不算Bug呢？這當(dāng)然是Bug，并JUnit也考慮到了這一點(diǎn)，來(lái)幫助我們找到這種Bug。例如，我們寫的計(jì)算器類有除法功能，如果除數(shù)是一個(gè)0，那么必然要拋出“除0異常”。因此，我們很有必要對(duì)這些進(jìn)行測(cè)試。代碼如下：

@Test(expected = ArithmeticException.class)

public void divideByZero() ...{

calculator.divide(0);

}

如上述代碼所示，我們需要使用

@Test標(biāo)注的expected屬性，將我們要檢驗(yàn)的異常傳遞給他，這樣JUnit框架就能自動(dòng)幫我們檢測(cè)是否拋出了我們指定的異常。

四、 Runner (運(yùn)行器)

大家有沒有想過(guò)這個(gè)問(wèn)題，當(dāng)你把測(cè)試代碼提交給JUnit框架后，框架如何來(lái)運(yùn)行你的代碼呢？答案就是——Runner。在JUnit中有很多個(gè)Runner，他們負(fù)責(zé)調(diào)用你的測(cè)試代碼，每一個(gè)Runner都有各自的特殊功能，你要根據(jù)需要選擇不同的Runner來(lái)運(yùn)行你的測(cè)試代碼。可能你會(huì)覺得奇怪，前面我們寫了那么多測(cè)試，并沒有明確指定一個(gè)Runner??？這是因?yàn)镴Unit中有一個(gè)默認(rèn)Runner，如果你沒有指定，那么系統(tǒng)自動(dòng)使用默認(rèn)Runner來(lái)運(yùn)行你的代碼。換句話說(shuō)，下面兩段代碼含義是完全一樣的：

import org.junit.internal.runners.TestClassRunner;

import org.junit.runner.RunWith;

//使用了系統(tǒng)默認(rèn)的TestClassRunner，與下面代碼完全一樣

public class CalculatorTest ...{

...

}

@RunWith(TestClassRunner.class)

public class CalculatorTest ...{

...

}

從上述例子可以看出，要想指定一個(gè)Runner，需要使用@RunWith標(biāo)注，并且把你所指定的Runner作為參數(shù)傳遞給它。另外一個(gè)要注意的是，@RunWith是用來(lái)修飾類的，而不是用來(lái)修飾函數(shù)的。只要對(duì)一個(gè)類指定了Runner，那么這個(gè)類中的所有函數(shù)都被這個(gè)Runner來(lái)調(diào)用。最后，不要忘了包含相應(yīng)的Package哦，上面的例子對(duì)這一點(diǎn)寫的很清楚了。接下來(lái)，我會(huì)向你們展示其他Runner的特有功能。

五、參數(shù)化測(cè)試。

你可能遇到過(guò)這樣的函數(shù)，它的參數(shù)有許多特殊值，或者說(shuō)他的參數(shù)分為很多個(gè)區(qū)域。比如，一個(gè)對(duì)考試分?jǐn)?shù)進(jìn)行評(píng)價(jià)的函數(shù)，返回值分別為“優(yōu)秀，良好，一般，及格，不及格”，因此你在編寫測(cè)試的時(shí)候，至少要寫5個(gè)測(cè)試，把這5中情況都包含了，這確實(shí)是一件很麻煩的事情。我們還使用我們先前的例子，測(cè)試一下“計(jì)算一個(gè)數(shù)的平方”這個(gè)函數(shù)，暫且分三類：正數(shù)、0、負(fù)數(shù)。測(cè)試代碼如下：

import org.junit.AfterClass;

import org.junit.Before;

import org.junit.BeforeClass;

import org.junit.Test;

import static org.junit.Assert.*;

public class AdvancedTest ...{

private static Calculator calculator ＝ new Calculator();

@Before

public void clearCalculator() ...{

calculator.clear();

}

@Test

public void square1() ...{

calculator.square(2);

assertEquals(4, calculator.getResult());

}

@Test

public void square2() ...{

calculator.square(0);

assertEquals(0, calculator.getResult());

}

@Test

public void square3() ...{

calculator.square(-3);

assertEquals(9, calculator.getResult());

}

為了簡(jiǎn)化類似的測(cè)試，JUnit4提出了“參數(shù)化測(cè)試”的概念，只寫一個(gè)測(cè)試函數(shù)，把這若干種情況作為參數(shù)傳遞進(jìn)去，一次性的完成測(cè)試。代碼如下：

import static org.junit.Assert.assertEquals;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.junit.runners.Parameterized;

import org.junit.runners.Parameterized.Parameters;

import java.util.Arrays;

import java.util.Collection;

@RunWith(Parameterized.class)

public class SquareTest ...{

private static Calculator calculator = new Calculator();

private int param;

private int result;

@Parameters

public static Collection data() ...{

return Arrays.asList(new Object[][]...{

...{2, 4},

...{0, 0},

...{－3, 9},

});

}

//構(gòu)造函數(shù)，對(duì)變量進(jìn)行初始化

public SquareTest(int param, int result) ...{

this.param = param;

this.result = result;

}

@Test

public void square() ...{

calculator.square(param);

assertEquals(result, calculator.getResult());

}

下面我們對(duì)上述代碼進(jìn)行分析。首先，你要為這種測(cè)試專門生成一個(gè)新的類，而不能與其他測(cè)試共用同一個(gè)類，此例中我們定義了一個(gè)SquareTest類。然后，你要為這個(gè)類指定一個(gè)Runner，而不能使用默認(rèn)的Runner了，因?yàn)樘厥獾墓δ芤锰厥獾腞unner嘛。@RunWith(Parameterized.class)這條語(yǔ)句就是為這個(gè)類指定了一個(gè)ParameterizedRunner。第二步，定義一個(gè)待測(cè)試的類，并且定義兩個(gè)變量，一個(gè)用于存放參數(shù)，一個(gè)用于存放期待的結(jié)果。接下來(lái)，定義測(cè)試數(shù)據(jù)的集合，也就是上述的data()方法，該方法可以任意命名，但是必須使用@Parameters標(biāo)注進(jìn)行修飾。這個(gè)方法的框架就不予解釋了，大家只需要注意其中的數(shù)據(jù)，是一個(gè)二維數(shù)組，數(shù)據(jù)兩兩一組，每組中的這兩個(gè)數(shù)據(jù)，一個(gè)是參數(shù)，一個(gè)是你預(yù)期的結(jié)果。比如我們的第一組{2, 4}，2就是參數(shù)，4就是預(yù)期的結(jié)果。這兩個(gè)數(shù)據(jù)的順序無(wú)所謂，誰(shuí)前誰(shuí)后都可以。之后是構(gòu)造函數(shù)，其功能就是對(duì)先前定義的兩個(gè)參數(shù)進(jìn)行初始化。在這里你可要注意一下參數(shù)的順序了，要和上面的數(shù)據(jù)集合的順序保持一致。如果前面的順序是{參數(shù)，期待的結(jié)果}，那么你構(gòu)造函數(shù)的順序也要是“構(gòu)造函數(shù)(參數(shù)，期待的結(jié)果)”，反之亦然。最后就是寫一個(gè)簡(jiǎn)單的測(cè)試?yán)耍颓懊娼榻B過(guò)的寫法完全一樣，在此就不多說(shuō)。

六、打包測(cè)試。

通過(guò)前面的介紹我們可以感覺到，在一個(gè)項(xiàng)目中，只寫一個(gè)測(cè)試類是不可能的，我們會(huì)寫出很多很多個(gè)測(cè)試類。可是這些測(cè)試類必須一個(gè)一個(gè)的執(zhí)行，也是比較麻煩的事情。鑒于此，JUnit為我們提供了打包測(cè)試的功能，將所有需要運(yùn)行的測(cè)試類集中起來(lái)，一次性的運(yùn)行完畢，大大的方便了我們的測(cè)試工作。具體代碼如下：

import org.junit.runner.RunWith;

import org.junit.runners.Suite;

@RunWith(Suite.class)

@Suite.SuiteClasses(...{

CalculatorTest.class,

SquareTest.class

})

public class AllCalculatorTests ...{

}

大家可以看到，這個(gè)功能也需要使用一個(gè)特殊的Runner，因此我們需要向@RunWith標(biāo)注傳遞一個(gè)參數(shù)Suite.class。同時(shí)，我們還需要另外一個(gè)標(biāo)注@Suite.SuiteClasses，來(lái)表明這個(gè)類是一個(gè)打包測(cè)試類。我們把需要打包的類作為參數(shù)傳遞給該標(biāo)注就可以了。有了這兩個(gè)標(biāo)注之后，就已經(jīng)完整的表達(dá)了所有的含義，因此下面的類已經(jīng)無(wú)關(guān)緊要，隨便起一個(gè)類名，內(nèi)容全部為空既可。

至此，本系列文章全部結(jié)束，希望能夠?qū)Υ蠹沂褂肑Unit4有所幫助。

【編輯推薦】