修补模拟类的方法

Question

我的代码具有以下结构：

我有一个继承自BaseClass的类MyClass （这是一个附带的点，而不是我的问题的根源）。 然后我有另一个类MyClassManager调用MyClass的方法。

我正在为MyClassManager的方法编写单元MyClassManager ，我想控制MyClass方法之一的返回值，同时自动指定其余方法。

在我的测试中，我通过使用autospec=True修补类为MyClass创建了一个 Mock。 然后我尝试修补方法MyClass.method_to_patch并将其替换为Substitute.substitute_method 。 到现在为止还挺好。

但是现在当我运行测试时，类管理器创建了一个MyClass的实例，它是一个完全自动指定的 Mock，但它没有修补我想要替换的方法。

有没有办法结合这两个patch装饰器来实现我想要的？

class Substitute:

    def substitute_method(self, arg1, arg2):
        print("Running substitute method")
        return (arg1 > 0 and arg2 > 0)


class BaseClass:

    def method_to_patch(self, arg1, arg2):
        return arg1 == arg2


class MyClass(BaseClass):

    def myclass_method(self):
        print("myclass method called")


class MyClassManager:

    def method_to_test(self):
        my_class = MyClass()
        my_class.myclass_method()
        my_class.method_to_patch(10, 100)


class TestMyClass(unittest.TestCase):

    @patch.object(MyClass, "method_to_patch", Substitute.substitute_method)
    @patch("__main__.MyClass", autospec=True)
    def test_method_to_test(self, mock_class):
        class_manager = MyClassManager()
        class_manager.method_to_test()
        print(mock_class.call_count)


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

Answer 1

我在下一页找到了有关模拟嵌套属性调用的答案的线索： https : //www.integralist.co.uk/posts/mocking-in-python/ 。 相同的逻辑适用于方法调用。

手动调整模拟对象是不够的 - 您必须调整return_value对象的return_value 。

所以这里是测试的样子：

class TestMyClass(unittest.TestCase):

    @patch("__main__.MyClass", autospec=True)
    def test_method_to_test(self, mock_class):
        mock_class.return_value.method_to_patch = Substitute.substitute_method
        class_manager = MyClassManager()
        class_manager.method_to_test()
        print(mock_class.call_count)

现在我有一个代替MyClass的模拟对象，所以MyClass.myclass_method也被模拟，但我可以根据需要用Substitute.substitute_method代替MyClass.method_to_patch 。

最后一点 - substitute_method方法实际上是一个staticmethod所以它应该是这样的：

class Substitute:

    @staticmethod
    def substitute_method(arg1, arg2):
        print("Running substitute method")
        return (arg1 > 0 and arg2 > 0)

Answer 2

要模拟类中的方法以返回特定值，请使用 @patch。 目的。 使用@patch 模拟类中的方法。 对象但每次调用时返回不同的值，请使用 side_effect。