模拟类实例而不调用 __init__ 并模拟它们各自的属性

Question

我有一个类 MyClass 和一个复杂的 __init__ 函数。

这个类有一个我想测试的方法my_method(self)。

my_method 只需要类实例中的属性my_attribute。

有没有一种方法可以模拟类实例而不调用__init__，而是设置每个类实例的属性？

我有什么：

# my_class.py

from utils import do_something

class MyClass(object):

   def __init__(self, *args, **kwargs):
       # complicated function which I would like to bypass when initiating a mocked instance class
       pass

   def my_method(self):
      return do_something(self.my_attribute)

我尝试了什么

@mock.patch("my_class.MyClass")
def test_my_method(class_mock, attribute):
   
   instance = class_mock.return_value
   instance.my_attribute = attribute

   example_instance = my_class.MyClass()

   out_my_method = example_instance.my_method()
   # then perform some assertions on `out_my_method`

但是这仍然使用__init__，我希望我们可以绕过或模拟。

Answer 1

正如我在 cmets 中提到的，无需创建实例即可测试单个方法的一种方法是：

MyClass.my_method(any_object_with_my_attribute)

问题在于，与quamrana's answer 中的两个选项一样，我们现在已经扩展了任何未来更改的范围只是因为测试。如果对my_method 的更改需要访问附加属性，我们现在必须同时更改实现和其他内容（SuperClass、MockMyClass，或者在本例中为any_object_with_my_attribute_and_another_one )。

---

让我们举一个更具体的例子：

import json


class MyClass:

    def __init__(self, filename):
        with open(filename) as f:
            data = json.load(f)
        self.foo = data.foo
        self.bar = data.bar
        self.baz = data.baz

    def my_method(self):
        return self.foo ** 2

这里是任何需要MyClass 实例的测试。由于__init__ 中的文件访问而痛苦。一个更可测试的实现会将如何访问数据的细节和有效实例的初始化分开：

class MyClass:
    
    def __init__(self, foo, bar, baz):
        self.foo = foo
        self.bar = bar
        self.baz = baz

    def my_method(self):
        return self.foo ** 2

    @classmethod
    def from_json(cls, filename):
        with open(filename) as f:
            data = json.load(f)
        return cls(data.foo, data.bar, data.baz)

您必须将 MyClass("path/to/file") 重构为 MyClass.from_json("path/to/file")，但无论您已经拥有数据（例如在您的测试中），您都可以使用例如MyClass(1, 2, 3) 无需文件即可创建实例（您只需要在from_json 本身的测试中考虑文件）。这使得实例真正需要更加清晰，并且允许引入其他方式来构造实例而不改变接口。

Answer 2

我至少可以看到两个选项：

1. 提取一个超类：

   class SuperClass:
       def __init__(self, attribute):
           self.my_attribute = attribute
      def my_method(self):
         return do_something(self.my_attribute)
   
   class MyClass(SuperClass):
   
      def __init__(self, *args, **kwargs):
          super().__init__(attribute)   # I don't know where attribute comes from
          # complicated function which I would like to bypass when initiating a mocked instance class
   

   您的测试可以实例化SuperClass 并调用my_method()。

2. 照原样从MyClass继承并制作您自己的简单__init__()：

   class MockMyClass(MyClass):
       def __init__(self, attribute):
           self.my_attribute = attribute
   

   现在您的测试代码可以使用所需属性实例化MockMyClass 并调用my_method()

   例如，您可以编写如下测试

   def test_my_method(attribute):
   
      class MockMyClass(MyClass):
         def __init__(self, attribute):
            self.my_attribute = attribute
   
      out_my_method = MockMyClass(attribute).my_method()
      # perform assertions on out_my_method