2 - Google Mock 进阶篇 - 如何mock多重载函数

同样对于重载函数的方法执行Mock也是一样的，不需要什么特别的方式。

 FooInterface.h
#ifndef FOOINTERFACE_H_
#define FOOINTERFACE_H_

#include<string>
 
namespace seamless {

	class FooInterface {
	public:
		virtual ~FooInterface() {};
		virtual std::string getArbitraryString() = 0;
		virtual int getPosition() = 0;
		 virtual bool Transform(int* g) = 0;
//多重载函数 Add()
		virtual int Add(int x)=0;
		virtual int Add(int times, int x)=0;


	protected:
		virtual void Resume()=0;

	private:
		virtual int GetTimeOut()=0;
	};


//MockFoo.h
#ifndef MOCKFOO_H_
#define MOCKFOO_H_

#include "FooInterface.h"
#include "gmock/gmock.h"
#include <string>
using namespace seamless;
using namespace std;

class MockFoo : public FooInterface {
public:
	MOCK_METHOD0(getArbitraryString, string());
	MOCK_METHOD0(getPosition, int());

	
	MOCK_METHOD1(Transform, bool(int* g));

	//mock私有、保护方法
	MOCK_METHOD0(Resume, void());
	MOCK_METHOD0(GetTimeOut, int());

	//mock 重载函数
	MOCK_METHOD1(Add, int(int x));
	MOCK_METHOD2(Add, int(int time, int x));

};

#endif // MOCKFOO_H_
//TestFooInterface.cpp
#include "pch.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "MockFoo.h"


using namespace seamless;
using namespace std;
using namespace testing;

using ::testing::Return;
class TestFooInterface :public testing::Test
{
protected:
	virtual void SetUp()
	{
	 //TEST_F（）测试初始化一些共用的数据 
	}
	virtual void TearDown()
	{ 
	}
	MockFoo m_foo;
};

//多重载函数 mock ,一个参数Add()方法测试
TEST_F(TestFooInterface,OverLoadTest){
	EXPECT_CALL(m_foo, Add(2))
		.WillOnce(Return(2));
	int returnValue;
	returnValue = m_foo.Add(2);
};

//多重载函数 mock,2个参数Add()方法测试
TEST_F(TestFooInterface, OverLoadTest2) {
	EXPECT_CALL(m_foo, Add(2,2))
		.WillOnce(Return(2));
	int returnValue;
	returnValue = m_foo.Add(2, 2);

};

int main(int argc, char** argv) {
	testing::InitGoogleMock(&argc, argv);
	printf("--》start");
	//如果不加下面这句话：执行所有的test， 那么TestTurtle的测试将不执行
	  RUN_ALL_TESTS();
	return EXIT_SUCCESS;
}

可以看出FooInterface.h的Add方法有2个，同名，参数不同。

  virtual int Add(Element x);

  virtual int Add(int times, Element x);

在模拟Mock类中， 模拟的方法跟普通的方法一样的，只要重复将这两个add方法都全部Mock一下就可以了。

  MOCK_METHOD1(Add, int(Element x));

  MOCK_METHOD2(Add, int(int times, Element x));

最后的单元测试执行也是一样的，把这两个重载的方法都执行一下就可以了。就是那么简单。代码已经全部贴上。

注意：如果你没有把所有的重载函数都Mock模拟的话，编译器会警告你基类中的一些函数被隐藏了。修正的方法是用using将它们引入域中：

 

class MockFoo :public Foo {

 using Foo::Add;

 MOCK_METHOD1(Add, int(Element x));

 // We don't want to mock int Add(int times, Element x);

 ...

};