python - 使用 python 包装的 c++ SWIG 模拟输入

Question

我是 swig 和 python 单元测试的新手。

这就是我想要做的。我有一个需要用户输入的 C++ 函数。使用 SWIG 将 C++ 代码包装到 Python 代码中。我正在尝试使用 pythons unittest 模块来模拟输入。我试过模拟 builtins.input 并用 c++ 编写我自己的基本函数，它只返回一个字符串并模拟它。

当我到达 c++ 代码中的 std::cin 时，模拟 builtins.input 仍然挂起。并模拟返回字符串的函数，不返回模拟的 return_value。

我的猜测是出于某种原因我无法模拟函数的返回值，因为它是真正的 c++ 代码而不是真正的 python。

这是我正在使用的一些示例代码:

c++ 我可以包含需要的头文件，但它真的很简单。

#include "MockTest.h"
#include <iostream>
#include <string>

MockTest::MockTest() {}

std::string MockTest::getInput()
{
    std::string name;
    std::cout << "Enter you name" << std::endl;
    name = this->mockInput("hi"); //std::cin >> name;
    std::string toReturn = "Hello " + name + " person";
    return toReturn;
}

std::string MockTest::mockInput(std::string input)
{ 
    return input;
}

swig接口(interface)文件:

%module MockTest
%include "std_string.i"
%include "std_vector.i"
%include "std_iostream.i"

%{
#include "MockTest.h"
%}
%include <MockTest.h>

python测试脚本

from unittest.mock import patch
from unittest import TestCase
import unittest

import MockTest


class Test(TestCase):

    @patch('builtins.input', return_value="Will")
    def test_Mocktest(self, input):
        self.assertEqual(MockTest.MockTest().getInput(), 'Hello Will person')

    @patch('MockTest.MockTest.mockInput', return_value="Will")
    def test_Mocktest2(self, mockInput):
        self.assertEqual(MockTest.MockTest().getInput(), 'Hello Will person')

if __name__ == '__main__':
    unittest.main()

Answer 1

在花费了一些时间之后~一个小时在一个愚蠢的错误上:

• 生成的模块不完整这一事实(链接器 在未向其提供MockTest 对象时并未提示 undefined symbol )
• 最后生成的是因为我忘记在 MockTest.h 的类成员之前添加 public: ... X:(((

又花了一些时间调查真正的问题，我终于得出结论:问题不是因为C++语言(至少不是直接的)，而是出在中介层由 Swig 创建。

详情

MockTest 类(其mockInput 方法返回值我们试图修补)在模块MockTest 中定义>(!!! 即 MockTest.py !!!)由 Swig 自动生成(我的命令是 swig -o MockTest_wrap.cpp -python -c++ MockTest.i).这是它的定义:

class MockTest(_object):
    __swig_setmethods__ = {}
    __setattr__ = lambda self, name, value: _swig_setattr(self, MockTest, name, value)
    __swig_getmethods__ = {}
    __getattr__ = lambda self, name: _swig_getattr(self, MockTest, name)
    __repr__ = _swig_repr

    def __init__(self):
        this = _MockTest.new_MockTest()
        try:
            self.this.append(this)
        except __builtin__.Exception:
            self.this = this
    __swig_destroy__ = _MockTest.delete_MockTest
    __del__ = lambda self: None

    def getInput(self):
        return _MockTest.MockTest_getInput(self)

    def mockInput(self, input):
        return _MockTest.MockTest_mockInput(self, input)

正如您可能猜到的那样，这个 mockInput 正在被补丁而不是来自 C++ 的那个(它的名称为 MockTest_mockInput，它由 native 模块导出:_MockTest - native 名称为 _wrap_MockTest_mockInput，定义在 MocTest_wrap.cpp).当然，由于 getInput 调用 MockTest_getInput，patching mockInput(Python 包装器) 没有任何效果(就像你修改它的主体让我们说:return "123")。

我准备了一些示例代码来更好地说明行为(正如我在评论中提到的，我使用的是 Python 3.5.4)。请忽略最后 3 行代码和输出，直到您阅读下一段:

from unittest.mock import patch
import MockTest


if __name__ == "__main__":
    return_value = "value overridden by `patch`"
    mock_input_arg = "argument passed to `MockTest.mockInput`"
    mock_test = MockTest.MockTest()
    with patch("MockTest.MockTest.mockInput", return_value=return_value):
        print("`mock_test.getInput()` returned: \"{}\"".format(mock_test.getInput()))
        print("`mock_test.mockInput(\"{}\")` returned: \"{}\"".format(mock_input_arg, mock_test.mockInput(mock_input_arg)))

    print("\nDon't mind the following output for now...\n")  # SAME THING about the following code

    with patch("MockTest._MockTest.MockTest_mockInput", return_value=return_value):
        print("`mock_test.getInput()` returned: \"{}\"".format(mock_test.getInput()))
        print("`mock_test.mockInput(\"{}\")` returned: \"{}\"".format(mock_input_arg, mock_test.mockInput(mock_input_arg)))

输出:

c:\Work\Dev\StackOverflow\q45934545>"c:\Install\x64\Python\Python\3.5\python.exe" dummy.py
Enter you name
`mock_test.getInput()` returned: "Hello hi person"
`mock_test.mockInput("argument passed to `MockTest.mockInput`")` returned: "value overridden by `patch`"

Don't mind the following output for now...

Enter you name:
`mock_test.getInput()` returned: "Hello hi person"
`mock_test.mockInput("argument passed to `MockTest.mockInput`")` returned: "value overridden by `patch`"

然后我进一步尝试补丁 MockTest._MockTest.MockTest_mockInput，但我得到了相同的输出，因为我没有补丁 MockTest::mockInput(来自MockTest.cpp)但_wrap_MockTest_mockInput(来自MocTest_wrap.cpp)。

下表是 C++ 代码和 Python 之间的所有层的表，用于 mockInput(对于 getInput 是完全一样):

当 getInput 调用 mockInput(layer 0)时，patch 应该发生，但不幸的是，这不适用于 Python。

我想到的第一个^st 解决方案是直接patching getInput(第 1 层)(如果它在许多地方使用) ).