python中的模拟字典

Question

我是一般修补/模拟的新手，尤其是 python。我正在尝试使用以下方法修补将在测试用例中访问的字典

obj = Foo(None)
with patch.dict(obj.bar.items_dict,{TEST_DIR + '/Transparent.gif', data}):
    output = obj.create_css_for_imgs()

但我收到以下错误。请理解Foo的__init__()函数默认使barNone

AttributeError: 'NoneType' 对象没有属性 'items_dict'

类如下：

class Foo(object):
    def __init__(self, barVal):
        self.bar = barVal

class Bar():
    def __init__(self, val)
        self.items_dict = {}
        self._load(val)

Answer 1

不确定您是否需要patch。 Patch 更常用于“模拟”模块的功能。您可以使用Mock 对象创建用于测试的结构（使用“模拟”方法和属性）。

首先，我认为您收到错误是因为您将None 发送到 构造函数。但是NoneType 没有items_dict 属性。

这里有一个可以帮助你的小例子：

from mock import Mock

class Foo(object):

    def __init__(self, bar_val):  # type: (Bar) -> None
        self.bar = bar_val

    def get_bar_items_dict(self):  # type: () -> dict
        # just example - return items_dict of Bar object
        return self.bar.items_dict

    def create_css_for_imgs(self):  # type: () -> str
        return 'example'


class Bar(object):

    def __init__(self):
        self.items_dict = {}
        # other properties...


class TestExample(TestCase):

    def test_create_css_for_imgs(self):
        # create Mock instead Bar object and say that items_dict isn't empty dict
        bar = Mock(items_dict={'custom': 'dict'})
        # foo.bar will be 'Mock' object
        foo = Foo(bar)
        # check result of 'create_css_for_imgs' Foo object
        self.assertEqual(
            foo.create_css_for_imgs(),
            'example'
        )

        # check result of foo.bar.items_dict
        self.assertDictEqual(
            foo.get_bar_items_dict(),
            {'custom': 'dict'}
        )

让我们总结一下。您收到错误是因为您尝试获取None 类型的items_dict（看起来像：a = None a.items_dict）。 您可以创建任何用于测试的对象（使用“模拟”方法和属性），而无需“修补”原始类。示例：

one = Mock(one_propery=1)
two = Mock(one=one)
three = Mock(two=two)
three.my_method.return_value = [1, 2, 3]
# check our object
print three.two.one.one_propery # 1
print three.my_method() # [1, 2, 3]

在某些情况下，我们需要“模拟”类的方法（例如用于组测试）。在这种情况下你可以使用patch:

    mocked_create = mock.patch(
        'your_pack.your_module.Foo.create_css_for_imgs',
        return_value='changed example'
    )
    # change 'example' to 'changed example'
    mocked_create.start()
    foo = Foo(Bar())
    print foo.create_css_for_imgs()  # 'changed example'
    mocked_create.stop()

希望这会有所帮助。