一、前言
上一篇我們分析了gtest的一些內部實現,總的來說整體的流程并不復雜。本篇我們就嘗試編寫一個精簡版本的C++單元測試框架:nancytest ,通過編寫這個簡單的測試框架,將有助于我們理解gtest。
二、整體設計
使用最精簡的設計,我們就用兩個類,夠簡單吧:
1. TestCase類包含單個測試案例的信息。
2. UnitTest類
負責所有測試案例的執行,管理。
三、TestCase類
TestCase類包含一個測試案例的基本信息,包括:測試案例名稱,測試案例執行結果,同時還提供了測試案例執行的方法。我們編寫的測試案例都繼承自TestCase類。
class TestCase{public: TestCase(const char* case_name) : testcase_name(case_name){} // 執行測試案例的方法 virtual void Run() = 0; int nTestResult; // 測試案例的執行結果 const char* testcase_name; // 測試案例名稱};
四、UnitTest類
我們的UnitTest類和gtest的一樣,是一個單件。我們的UnitTest類的邏輯非常簡單:
1. 整個進程空間保存一個UnitTest 的單例。
2. 通過RegisterTestCase()將測試案例添加到測試案例集合testcases_中。
3. 執行測試案例時,調用UnitTest::Run(),遍歷測試案例集合testcases_,調用案例的Run()方法
class UnitTest{public: // 獲取單例 static UnitTest* GetInstance(); // 注冊測試案例 TestCase* RegisterTestCase(TestCase* testcase); // 執行單元測試 int Run(); TestCase* CurrentTestCase; // 記錄當前執行的測試案例 int nTestResult; // 總的執行結果 int nPassed; // 通過案例數 int nFailed; // 失敗案例數PRotected: std::vector<TestCase*> testcases_; // 案例集合};
下面是UnitTest類的實現:
UnitTest* UnitTest::GetInstance(){ static UnitTest instance; return &instance;}TestCase* UnitTest::RegisterTestCase(TestCase* testcase){ testcases_.push_back(testcase); return testcase;}int UnitTest::Run(){ nTestResult = 1; for (std::vector<TestCase*>::iterator it = testcases_.begin(); it != testcases_.end(); ++it) { TestCase* testcase = *it; CurrentTestCase = testcase; std::cout << green << "======================================" << std::endl; std::cout << green << "Run TestCase:" << testcase->testcase_name << std::endl; testcase->Run(); std::cout << green << "End TestCase:" << testcase->testcase_name << std::endl; if (testcase->nTestResult) { nPassed++; } else { nFailed++; nTestResult = 0; } } std::cout << green << "======================================" << std::endl; std::cout << green << "Total TestCase : " << nPassed + nFailed << std::endl; std::cout << green << "Passed : " << nPassed << std::endl; std::cout << red << "Failed : " << nFailed << std::endl; return nTestResult;}
五、NTEST宏
接下來定一個宏NTEST,方便我們寫我們的測試案例的類。
#define TESTCASE_NAME(testcase_name) / testcase_name##_TEST#define NANCY_TEST_(testcase_name) /class TESTCASE_NAME(testcase_name) : public TestCase /{ /public: / TESTCASE_NAME(testcase_name)(const char* case_name) : TestCase(case_name){}; / virtual void Run(); /private: / static TestCase* const testcase_; /}; //TestCase* const TESTCASE_NAME(testcase_name) / ::testcase_ = UnitTest::GetInstance()->RegisterTestCase( / new TESTCASE_NAME(testcase_name)(#testcase_name)); /void TESTCASE_NAME(testcase_name)::Run()#define NTEST(testcase_name) / NANCY_TEST_(testcase_name)
六、RUN_ALL_TEST宏
然后是執行所有測試案例的一個宏:
#define RUN_ALL_TESTS() / UnitTest::GetInstance()->Run();七、斷言的宏EXPECT_EQ
這里,我只寫一個簡單的EXPECT_EQ :
#define EXPECT_EQ(m, n) / if (m != n) / { / UnitTest::GetInstance()->CurrentTestCase->nTestResult = 0; / std::cout << red << "Failed" << std::endl; / std::cout << red << "Expect:" << m << std::endl; / std::cout << red << "Actual:" << n << std::endl; / }
八、案例Demo
夠簡單吧,再來看看案例怎么寫:
#include "nancytest.h"int Foo(int a, int b){ return a + b;}NTEST(FooTest_PassDemo){ EXPECT_EQ(3, Foo(1, 2)); EXPECT_EQ(2, Foo(1, 1));}NTEST(FooTest_FailDemo){ EXPECT_EQ(4, Foo(1, 2)); EXPECT_EQ(2, Foo(1, 2));}int _tmain(int argc, _TCHAR* argv[]){ return RUN_ALL_TESTS();}
整個一山寨版gtest,呵。執行一下,看看結果怎么樣:

九、總結
本篇介紹性的文字比較少,主要是我們在上一篇深入解析gtest時已經將整個流程弄清楚了,而現在編寫的nancytest又是其非常的精簡版本,所有直接看代碼就可以完全理解。希望通過這個Demo,能夠讓大家對gtest有更加直觀的了解?;氐介_篇時所說的,我們沒有必要每個人都造一個輪子,因為gtest已經非常出色的為我們做好了這一切。如果我們每個人都寫一個自己的框架的話,一方面我們要付出大量的維護成本,一方面,這個框架也許只能對你有用,無法讓大家從中受益。gtest正是這么一個優秀C++單元測試框架,它完全開源,允許我們一起為其貢獻力量,并能讓更多人從中受益。如果你在使用gtest過程中發現gtest不能滿足你的需求時(或發現BUG),gtest的開發人員非常急切的想知道他們哪來沒做好,或者是gtest其實有這個功能,但是很多用戶都不知道。所以你可以直接聯系gtest的開發人員,或者你直接在這里回帖,我會將您的意見轉告給gtest的主要開發人員。如果你是gtest的超級粉絲,原意為gtest貢獻代碼的話,加入他們吧?! ?/p>
本Demo代碼下載:/Files/coderzh/Code/nancytest.rar
本篇是該系列最后一篇,其實gtest還有更多東西值得我們去探索,本系列也不可能將gtest介紹完全,還是那句話,想了解更多gtest相關的內容的話:
訪問官方主頁:http://code.google.com/p/googletest/
下載gtest源碼: http://code.google.com/p/googletest/downloads/list