VS 2022

Demo_1

  1. #include <iostream>
  2. #include <gtest/gtest.h>
  3. #include <gmock/gmock.h>
  5. using namespace testing;
  7. class Foo {
  8. public:
  9.     virtual int count() = 0;
  10. };
  12. class MockFoo : public Foo {
  13. public:
  14.     MOCK_METHOD0(count, int());
  15. };
  17. class Bar {
  18. public:
  19.     Bar(Foo* foo) : m_foo(foo) { }
  20.     int count() { return m_foo->count(); }
  22. private:
  23.     Foo* m_foo;
  24. };
  27. TEST(BarTest, mock_foo_in_bar) {
  28.     MockFoo* mockFoo = new MockFoo();
  29.     Bar* bar = new Bar(mockFoo);
  31.     EXPECT_CALL(*mockFoo, count())
  32.         .Times(3)
  33.         .WillOnce(Return(3))
  34.         .WillOnce(Return(6))
  35.         .WillOnce(Return(5));
  37.     EXPECT_EQ(bar->count(), 3);
  38.     EXPECT_EQ(bar->count(), 6);
  39.     EXPECT_EQ(bar->count(), 5);
  41.     delete bar;
  42.     delete mockFoo;
  43. }
  45. int main(int argc, char** argv)
  46. {
  47.     testing::InitGoogleTest(&argc, argv);
  48.     return RUN_ALL_TESTS();
  49. }
  1. [==========] Running 1 test from 1 test suite.
  2. [----------] Global test environment set-up.
  3. [----------] 1 test from BarTest
  4. [ RUN      ] BarTest.mock_foo_in_bar
  5. [       OK ] BarTest.mock_foo_in_bar (0 ms)
  6. [----------] 1 test from BarTest (1 ms total)
  8. [----------] Global test environment tear-down
  9. [==========] 1 test from 1 test suite ran. (2 ms total)
  10. [  PASSED  ] 1 test.

Demo_2: 桩掉C函数

Demo_3

  1. #include <gtest/gtest.h>
  2. #include <gmock/gmock.h>
  3. #include <iostream>
  5. using namespace std;
  6. using ::testing::AtLeast;
  7. using ::testing::Return;
  8. using ::testing::_;
  10. class DataBaseConnect
  11. {
  12. public:
  13.     virtual bool login(string username, string password) { return true; }
  14.     virtual bool logout(string username) { return true; }
  15.     virtual int fetchRecord() { return -1; }
  16. };
  18. class MockDB : public DataBaseConnect
  19. {
  20. public: 
  21.     MOCK_METHOD0(fetchRecord, int());
  22.     MOCK_METHOD1(logout, bool (string username));
  23.     MOCK_METHOD2(login, bool (string username, string password));
  24. };
  26. class MyDatabase
  27. {
  28.     DataBaseConnect& dbC;
  29. public:
  30.     MyDatabase(DataBaseConnect& _dbC) :dbC{_dbC}{}
  32.     int Init(string username, string password)
  33.     {
  34.         if (dbC.login(username, password) != true)
  35.         {
  36.             cout << "DB FAILTURE\n"; return -1;
  37.         }
  38.         else
  39.         {
  40.             cout << "DB SUCCESS\n"; return 1;
  41.         }
  42.     }
  43. };
  45. TEST(MyDBTest, LoginTest)
  46. {
  47.     // Arrange
  48.     MockDB mdb;
  49.     MyDatabase db{ mdb };
  50.     EXPECT_CALL(mdb, login("Terminator", "I'm Back"))
  51.         .Times(1)
  52.         .WillOnce(Return(true));
  55.     // Act
  56.     int retValue = db.Init("Terminator", "I'm Back");
  58.     // Assert
  59.     EXPECT_EQ(retValue, 1);
  60. }
  1. [==========] Running 1 test from 1 test suite.
  2. [----------] Global test environment set-up.
  3. [----------] 1 test from MyDBTest
  4. [ RUN      ] MyDBTest.LoginTest
  5. DB SUCCESS
  6. [       OK ] MyDBTest.LoginTest (0 ms)
  7. [----------] 1 test from MyDBTest (0 ms total)
  9. [----------] Global test environment tear-down
  10. [==========] 1 test from 1 test suite ran. (1 ms total)
  11. [  PASSED  ] 1 test.