Question

我一直在寻找一种方法来模拟同一个类的多个实例。尝试了How to get many instances of one Mock object的解决方案，这确实有效。但是，我还想在我的测试用例中为mocked类的每个新实例指定属性和行为。还没有找到答案。

示例：

class A:
    def __init__(self, name):
        self.name = name

    def get_name(self):
        print("hi " + self.name)
        return self.name


class B:
    def __init__(self):
        self.a_map = {}

    def get_all_a_values(self, names):
        for name in names:
            a = A(name)
            self.a_map[a] = a.get_name()


with mock.patch('__main__.A') as mockA:
    b = B()
    mockA.side_effect = mock.Mock
    names = ['person0', 'person1', 'person2', 'person3', 'person4']
    b.get_all_a_values(names)
    print(b.a_map)
    assert len(b.a_map) == 5

如何更改此代码，以便在创建mockA实例时，还设置实例的name属性？例如我需要这段代码的输出类似于：

{<Mock id='13609360'>: 'person0', <Mock id='13608720'>: 'person1', <Mock id='13610128'>: 'person2', <Mock id='13609744'>: 'person3', <Mock id='13608976'>: 'person4'}

设置mockA.get_name.return_value不会起作用，因为它将为每个实例使用相同的值。

Answer 1

请注意，该名称由A的get_name方法返回。因此，您可以使用side_effect覆盖该方法，该side_effect返回调用该方法的名称。看起来你真正想要的不是多个模拟实例，而是模拟中的一个方法，它根据传递给它的内容返回不同的值。

以下代码适用于我：

import mock

class A:
    def __init__(self, name):
        self.name = name

    def get_name(self):
        print("hi " + self.name)
        return self.name


class B:
    def __init__(self):
        self.a_map = {}

    def get_all_a_values(self, names):
        for name in names:
            a = A(name)
            self.a_map[a] = a.get_name()


def side_effect(name):
    mm = mock.MagicMock()
    mm.get_name.return_value = name
    return mm

with mock.patch('__main__.A') as mockA:
    b = B()
    mockA.side_effect = side_effect
    names = ['person0', 'person1', 'person2', 'person3', 'person4']
    b.get_all_a_values(names)
    print(b.a_map)
    assert len(b.a_map) == 5

当我运行它时，我的输出是：

{<MagicMock id='4491688656'>: 'person0', <MagicMock id='4491693264'>: 'person1', <MagicMock id='4491757456'>: 'person4', <MagicMock id='4491722960'>: 'person2', <MagicMock id='4491742224'>: 'person3'}

我发现模拟文档相对令人困惑，但side_effects部分值得混淆，因为它们是模拟中最有用的功能之一。

模拟对象的多个实例

1 个答案: