【转】玩转Google开源C++单元测试框架Google Test系列(gtest)之八 - 打造自己的单元测试框架

原文地址：http://www.cnblogs.com/coderzh/archive/2009/04/12/1434155.html

一、前言

上一篇我们分析了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贡献代码的话，加入他们吧。　　　

本Demo代码下载：/Files/coderzh/Code/nancytest.rar

本篇是该系列最后一篇，其实gtest还有更多东西值得我们去探索，本系列也不可能将gtest介绍完全，还是那句话，想了解更多gtest相关的内容的话：

访问官方主页：http://code.google.com/p/googletest/

下载gtest源码： http://code.google.com/p/googletest/downloads/list