c++ - 无法创建模拟类/对象来替换我类中私有(private)成员对象的功能，以便在 gtest 中进行单元测试

Question

我有一个跟类

#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 秒后说一次。

我想为 A 类编写一个单元测试用例，我认为最好的方法是为 B 创建一个模拟类(因为我不希望它依赖于 B 并执行基于时间的操作)，而我不想这样做必须创建一个文件。

在我的测试中，我会将数据存储到缓冲区中并从该缓冲区中读取。但是，我不确定如何让我的主类使用和创建我的模拟类 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;
}

我不确定如何让我的原始类在我的单元测试中使用这个模拟类 B 的对象而不是原始类 B(我使用的是 gtest，我没有在此处包含 google 测试单元)。提前致谢。

Answer 1

你可以通过创建一个接口(interface)来做到这一点IB并上课B实现该接口(interface)。然后，在类里面A , 使用指向此接口(interface)的指针。这样您就可以更改测试中的实现并使用模拟类。

还有一件事。由于您使用的是 gtest已经，我建议你不要自己编写模拟类，而是使用 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
    }
}

然后，创建一个模拟类，它也使用 gmock 实现了这个接口(interface):

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?