无法创建模拟 class/object 来替换我 class 中私有成员对象的功能，以便在 gtest 中进行单元测试

Question

我有一个关注者class

#include "B.h"
class A{
    public:
        A();
        void Store(char* buff, int32_t len){
            if (!b_read){
                b_write = new B
            }
            b_write->write(buff, len);
        }
        void Read(char* buff, int32_t& len){
            if (!b_read){
                b_read = new B;
            }
            b_read->read(buff, len);
        }

    private:
        B *b_write;
        B *b_read;
}

B 在内部写入文件（批处理模式）并从该文件中读取（一次一条记录）。它以基于时间的格式执行，即每 x 秒后说一次。

我想为 class A 编写一个单元测试用例，我认为最好的方法是为 B 创建一个模拟 class（因为我不希望它依赖于 B 并且做基于时间的操作）以及我不必创建文件的地方。

在我的测试中，我会将数据存储到缓冲区中并从该缓冲区中读取。但是，我不确定如何让我的 main class 使用和创建我的 mock class B 的对象而不是 B

的原始定义

这就是我所做的

//Original class
class A{
    public:
        A();
        // I've created another constructor for this class.
        A(B *ptr){
            b_write = ptr;
            b_read = ptr;
        };
        void Store(char* buff, int32_t len){
            if (!b_read){
                b_write = new B
            }
            b_write->write(buff, len);
        }
        void Read(char* buff, int32_t& len){
            if (!b_read){
                b_read = new B;
            }
            b_read->read(buff, len);
        }

    private:
        B *b_write;
        B *b_read;
}
//New mock class B
class B{
    public:
        B(){
            m_buff = new char[1024];
            m_len = m_read = 0;
        }
        void write(char* buff, int32_t len){
            if (m_len + len >= 1024){
                return 0;
            }
            memcpy(buff + m_len, buff, len);
        }
        void read(char* buff, int32_t& len){
            if (!buff){
                buff = new char[10];
            }
            memcpy(buff, m_buff, 10)
            /*
            * Rest of the code handles shifting of the data and all
            **/
        }
    private:
        char* m_buff;
        int32_t m_len;
        int32_t m_read;
}

我不确定如何让我的原始 class 在我的单元测试中使用这个模拟 class B 的对象而不是原始 class B（我正在使用 gtest，我这里没有包含 google 测试单元）。提前致谢。

Answer 1

您可以通过创建接口 IB 并使 class B 实现该接口来实现。然后，在classA中，使用指向这个接口的指针。这样您就可以更改测试中的实现并使用模拟 class.

还有一件事。由于您已经在使用 gtest，我建议您不要自己编写模拟 class，而是使用 C++ 模拟框架 gmock（与 gtest 完全兼容）。以下是您将如何在您的案例中使用它：

class IB
{
public:
    virtual IB() {}
    virtual void write(char* buff, int32_t len) = 0;
    virtual void read(char* buff, int32_t& len) = 0;
}

class B : public IB
{
public:
    void write(char* buff, int32_t len) override
    {
        // Your code that writes buff
    }
    void read(char* buff, int32_t& len) override
    {
        // Your code that reads buff
    }
}

然后，制作一个 mock class，它也使用 gmock 实现了这个接口：

class BMock : public IB
{
public:
    MOCK_METHOD2(write, void(char*, int32_t));
    MOCK_METHOD2(read, void(char*, int32_t&));
}

最后，在您的测试中，您可以使用 gtest 模拟调用此函数。这就是你在测试体中要做的（用伪代码编写）：

vector<string> dummyData = { string("str1"), string("str2") };
BMock mockObj;
EXPECT_CALL(mockObj, read(_, _)).WillOnce( ... );

一开始弄清楚如何使用 gmock 可能有点困难，但在漫长的运行中肯定会得到回报。

编辑：

如果您正在开发一个性能关键的系统并且想避免虚函数，也有一些方法可以做到这一点。请查看以下链接：

https://code.google.com/p/googlemock/wiki/CookBook#Mocking_Nonvirtual_Methods

C++ High performance unit testing with Google Mock?

无法创建模拟 class/object 来替换我 class 中私有成员对象的功能，以便在 gtest 中进行单元测试

Not able to create mock class/object to replace the functionality of a private member object in my class for unit testing in gtest

c++

unit-testing

googletest