推迟调用以及Lambda表达式

背景

GMock

我们项目中现在的模块测试框架使用了CATCH+GMock的方式实现回归测试和打桩。

GMock的介绍在官网上有，这里为了铺垫，大概地描述一下GMock能实现的效果。大约可以看成这样：

1. void A() {
2.     if(B()) {
3.         //...
4.     }
5.     Else{
6.         //...
7.     }
8. }

A是被测函数，B是桩函数。

在测试的，使用GMock的话，我们可以这样写测试代码：

1. TEST_CASE(Test As normal case) {
2.     EXPECT_CALL(mockobj, B).times(1).WillOnce(Return(true)); // MockBAtrue
3.     A(); // A
4.     // BFailedB should be called but not called
5. }

模块测试

所以，使用GMock以后我们可以很愉快地打桩了，但是有一个问题是，必须在调用被测函数 （A）之前给B函数打桩（描述B应该被调用几次，以及有什么样的行为）。这在UT中虽然是没有什么问题的（因为UT中函数只调用一次），但是要是用在模块的时序测试上，就会使人产生时序上的混乱感。

比如我们有一个时序：

Tester  ---Msg1-–> B
                         B call IF1
                         B call IF2

Tester  ---Msg2-–> B
                         B call IF3
                         B call IF4

我们如果正常地按时序思路写测试代码，那么希望是这样的（Program1）：

1. TEST_START()
2. SendMsg(toB, msg1);
3. IF1_isExpectedTobeCalled(Mock)
4. IF2_isExpectedTobeCalled(Mock)
5. SendMsg(toB, msg2);
6. IF3_isExpectedTobeCalled(Mock)
7. IF4_isExpectedTobeCalled(Mock)
8. TEST_END()

但是，由于GMock的使用方法决定，我们必须先写成这样：

1. TEST_START()
2. IF1_isExpectedTobeCalled(Mock)
3. IF2_isExpectedTobeCalled(Mock)
4. SendMsg(toB, msg1);
5. IF3_isExpectedTobeCalled(Mock)
6. IF4_isExpectedTobeCalled(Mock)
7. SendMsg(toB, msg2);
8. TEST_END()

在很长的时序和很多的桩的情况下这就显得很别扭了。编写和维护的时候都很容易出错。

问题

能不能提供一种办法（宏），使得我们可以像（Program1）那样的顺序写代码，

同时，代码又是以Program2这样的顺序来执行呢？（即，书写时按我们的正常思路写，执行时，按GMock需要的顺序执行）

比如：写代码时可以这样：

1. TEST_START()
2. TEST_STEP(SendMsg(toB, msg1))
3. IF1_isExpectedTobeCalled(Mock)
4. IF2_isExpectedTobeCalled(Mock)
5. TEST_STEP(SendMsg(toB, msg2))
6. IF3_isExpectedTobeCalled(Mock)
7. IF4_isExpectedTobeCalled(Mock)
8. TEST_END()

而实际的执行顺序是：

1. IF1_isExpectedTobeCalled(Mock)
2. IF2_isExpectedTobeCalled(Mock)
3. SendMsg(toB, msg1);
4. IF3_isExpectedTobeCalled(Mock)
5. IF4_isExpectedTobeCalled(Mock)
6. SendMsg(toB, msg2);

 

解法

中间我自己的折腾过程总不详细描述了，实际上我们就是要实现推调用的效果，而且，由于我们知道调用需要推迟到哪个点，那么非常容易想到“析构函数”，因为析构函数会在作用域结束时被调用。所以我们如果可以把函数调用存储在一个对象里，然后让这个对象在指定的点析构，析构时调用我们之前存储的函数，目的就达到了。问题是“函数”如何存储。答案就是C++11中提供的function库和lamabda表达式，实现方法如下：

1. class CallLater {
2. public:
3.     CallLater(function<void(void)> _fun): m_fun(_fun){
4.     }
5.     ~CallLater() {
6.         m_fun();
7.     }
8. private:
9.     function<void(void)> m_fun;
10. };
11. #define TEST_STEP(fun)  } { CallLater temp ([](){ fun; });
12. #define TEST_START()     {
13. #define TEST_END()       }

相当地简洁和舒服。这就是为什么我非常喜欢C++11中的那些“语法糖”。