python - 在 python 中使用 'autospec=True' 时模拟类属性

Question

我想模拟一个具有以下要求的类:

• 该类具有公共(public)读/写属性，在其 __init__() 方法中定义
• 该类具有在创建对象时自动递增的公共(public)属性
• 我希望使用 autospec=True，因此类的 API 将严格检查调用

一个简化的类示例:

class MyClass():
    id = 0

    def __init__(self, x=0.0, y=1.0):
        self.x = x
        self.y = y
        self.id = MyClass._id
        MyClass.id +=1

    def calc_x_times_y(self):
        return self.x*self.y

    def calc_x_div_y(self, raise_if_y_not_zero=True):
        try:
            return self.x/self.y
        except ZeroDivisionError:
            if raise_if_y_not_zero:
                raise ZeroDivisionError
            else:
                return float('nan')

就属性而言，我需要模拟对象的行为与原始对象相同:

• 它应该自动递增分配给每个新创建的模拟对象的 id
• 它应该允许访问它的 x,y 属性但是 mock 方法调用应该被 mock 拦截，并验证其调用签名

继续进行此操作的最佳方式是什么？

编辑

我已经尝试了几种方法，包括子类化 Mock 类，使用 attach_mock() 和 mock_add_spec()，但总是遇到了死胡同。

我使用的是标准 mock图书馆。

Answer 1

由于没有收到任何答案，我将发布对我有用的方法(不一定是最好的方法，但这里有):

我创建了一个模拟工厂，它创建了一个 Mock() 对象，使用描述的语法设置它的 id 属性 here ，并返回对象:

 class MyClassMockFactory():
     _id = 0

     def get_mock_object(self, *args,**kwargs):
        mock = Mock(MyClass, autospec = True)
        self._attach_mock_property(mock , 'x', kwargs['x'])
        self._attach_mock_property(mock , 'y', kwargs['y'])
        self._attach_mock_property(mock , 'id', MyClassMockFactory._id)
        MyClassMockFactory._id += 1
        return mock

     def _attach_mock_property(self, mock_object, name, value):
         p = PropertyMock(return_value=value)
         setattr(type(mock_object), name, p)

现在，我可以为我的测试修补 MyClass() 构造函数:

class TestMyClass(TestCase):
     mock_factory = MyClassMockFactory()

     @patch('MyClass',side_effect=mock_factory.get_mock_object)
     test_my_class(self,*args):
         obj0 = MyClass()
         obj1 = MyClass(1.0,2.2)
         obj0.calc_x_times_y()
         # Assertions
         obj0.calc_x_times_y.assert_called_once_with()
         self.assertEqaul(obj0.id, 0)
         self.assertEqaul(obj1.id, 1)