GMOCK - 返回类型为void时如何修改方法参数

时间:2017-09-07 09:28:18

标签: c++ gmock

我有一个接受指向另一个类的指针的类,并且有一个方法read():

class B:
{
public:
......
void read(char * str);
......
};

class A
{
public:
A(B *bobj):b(bobj);
B* b;
void read (char * str);
..........
};

我调用下面的对象

A * aobj = new A(new B());

现在我应该能够访问以下两个类的读取方法:

char text[100];
b->read(text)
aobj->read(text)

A和B类的读取方法被编码为将一些值复制到输入数组中。

如何编写函数的MOCK方法将参数修改为特定值?

ON_CALL(*b, read(::testing::_)).WillByDefault(::testing::Return(outBuffer));

给我一​​个编译问题,因为read方法不能按定义返回值?

1 个答案:

答案 0 :(得分:3)

我使用了Invoke函数来实现这一点。这是一个示例代码。

如果你使用> = c ++ 11,那么你可以简单地给一个lambda赋予Invoke函数。

#include <gtest/gtest.h>                                                                                                                                                                                             
#include <gmock/gmock.h>

using ::testing::Mock;
using ::testing::_;
using ::testing::Invoke;

class actual_reader
{
   public:
   virtual void read(char* data)
   {   
      std::cout << "This should not be called" << std::endl;
   }   
};
void mock_read_function(char* str)
{
   strcpy(str, "Mocked text");
}

class class_under_test
{
   public:
      void read(actual_reader* obj, char *str)
      {   
         obj->read(str);
      }   
};

class mock_reader : public actual_reader
{
   public:
   MOCK_METHOD1(read, void(char* str));
};


TEST(test_class_under_test, test_read)
{
   char text[100];

   mock_reader mock_obj;
   class_under_test cut;

   EXPECT_CALL(mock_obj, read(_)).WillOnce(Invoke(mock_read_function));
   cut.read(&mock_obj, text);
   std::cout << "Output : " << text << std::endl;

   // Lambda example
   EXPECT_CALL(mock_obj, read(_)).WillOnce(Invoke(
            [=](char* str){
               strcpy(str, "Mocked text(lambda)");
            }));
   cut.read(&mock_obj, text);
   std::cout << "Output : " << text << std::endl;



}