2

我正在使用模拟库来修补由两个文件导入的模块，这些文件定义了一个类及其子类。就像是：

# ---- file module_to_patch.py -----
def foo():
    return 1

def faa():
    return 2

# ---- file base_class.py -----
import module_to_patch

class base_class(object):
    def __init__(self):
        self.x = module_to_patch.foo()

# ---- file sub_class.py -----
import module_to_patch

class sub_class(object):
    def __init__(self):
        super(sub_class, self).__init__()
        self.y = module_to_patch.faa()

据我了解（见这里），在我的测试中，我应该同时修补base_class.module_to_patch和sub_class.module_to_patch，例如：

# file test_sub_class.py:
from unittest import TestCase
from mock import patch

class TestSubClass(TestCase):

    def setUp(self):
        self.patcher_1 = patch('base_class.module_to_patch', autospec=True)
        self.mock_1 = self.patcher_1.start()
        self.patcher_2 = patch('sub_class.module_to_patch', autospec=True)
        self.mock_2 = self.patcher_2.start()

    def tearDown(self):
        patch.stopall()

问题：

  • 有没有更简单的方法来实现这个功能？
  • 如果不是 - 两个补丁都修补同一个实体，我怎样才能让两个补丁返回同一个Mock()对象？我只是作为对象传递mock_1给吗？patcher_2new
  • 以上是处理这种情况的正确方法吗？
4

1 回答 1

0

实现我认为您正在尝试做的最简单的方法是简单地模拟导入对象中的模块。如果您只想为整个测试套件模拟模块：

# file test_sub_class.py:
from unittest import TestCase
from mock import Mock

import sub_class
import base_class
MODULE_MOCK = Mock()
sub_class.module_to_patch = MODULE_MOCK
base_class.module_to_patch = MODULE_MOCK

class TestSubClass(TestCase):

    def test_sub_class_init(self):
        self.base_class = base_class.base_class()
        self.assertTrue(MODULE_MOCK.foo.called)

        self.sub_class = sub_class.sub_class()
        self.assertTrue(MODULE_MOCK.faa.called)

如果你只想模拟一个测试，你可以使用patch.object作为装饰器，并给它你希望 patch.object 使用的 Mock 对象（MOCK_OBJECT）：

# file test_sub_class.py:
from unittest import TestCase
from mock import Mock, patch

import sub_class
import base_class

MOCK_OBJECT = Mock()

class TestSubClass(TestCase):

    @patch.object(sub_class, 'module_to_patch', MOCK_OBJECT)
    @patch.object(base_class, 'module_to_patch', MOCK_OBJECT)
    def test_sub_class_init(self):
        self.base_class = base_class.base_class()
        self.assertTrue(MOCK_OBJECT.foo.called)
        self.sub_class = sub_class.sub_class()
        self.assertTrue(MOCK_OBJECT.faa.called)
于 2013-08-23T01:00:45.160 回答