最近使用gtest進行單元測試,採用打樁的形式。關於gtest的詳細說明就不多說了,網上的資料一大堆。主要講解使用時的參數如何配置以及遇到的問題。下面的例子模擬是加、減、乘、除四則運算,前提是不知道加、減、乘、除四則運算是如何實現的。
編譯命令自己根據實際情況進行搭建
#include <iostream>
#include <vector>
#include "gtest/gtest.h"
#include "gmock/gmock.h"
#include <string>
using namespace testing;
using namespace std;
//using ::testing::Return;
//using ::testing::ByRef;
class Operator //要測試的接口分別是add() sub()
{
public:
virtual int add(int x, int y) = 0;//加法
virtual int sub(int x, int y) = 0; //減法
virtual int multi(int x, int y) = 0; //乘法
virtual int divide(int x, int y) = 0; //除法
};
class MockOperator : public Operator {
public:
MOCK_METHOD2(add, int(int x, int y)); //2代表add()有2個參數
int Operation::Add(Operator *op, int a, int b) {
return op->add(a, b);
}
int Operation::Sub(Operator *op, int a, int b) {
return op->sub(a, b);
}
int Operation::Multi(Operator *op, int a, int b) {
return op->multi(a, b);
}
int Operation::Divide(Operator *op, int a, int b) {
return op->divide(a, b);
}
TEST(OperatorTest, add) {
MockOperator mock;
int t = 13;
EXPECT_CALL(mock, add(6, 7)).WillRepeatedly(Return(t));
Operation op;
EXPECT_EQ(13, op.Add(&mock, 6, 7)); //13 與 Add的結果進行比較
}
TEST(OperatorTest, sub) {
MockOperator mock;
int t = -1;
EXPECT_CALL(mock, sub(6, 7)).WillRepeatedly(Return(t)); //設定返回值是t
Operation op;
EXPECT_EQ(-1, op.Sub(&mock, 6, 7)); //-1 與 Sub的結果進行比較
}
TEST(OperatorTest, multi) {
MockOperator mockoperator;
int t = 18;
EXPECT_CALL(mockoperator, multi(2, 9)).WillRepeatedly(Return(t)); //設定返回值爲t
Operation op;
EXPECT_EQ(18, op.Multi(&mockoperator, 2, 9)); //18 與 Multi的結果進行比較
}
TEST(OperatorTest, divide) {
MockOperator mock;
int t = 3;
EXPECT_CALL(mock, divide(15, 5)).WillRepeatedly(Return(t)); //設定返回值爲t
Operation op;
EXPECT_EQ(3, op.Divide(&mock, 15, 5)); //3 與 Divide()的結果進行比較
}
int main(int argc, char **argv) {
std::cout << "start gtest" << endl;
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
結果如下: