如何模拟使用补丁相对路径？

Question

我在python测试文件中有类似的内容：

from mock import patch,
from ..monkey import ook
[...]
@patch('monkey.ook', Mock(return_value=None))
def test_run_ook (self, mock_ook):
    self.assertIsNone(ook())
    mock_ook.run.assert_called_once_with('')

当我运行这个测试时，我得到一个ImportError: No module named monkey。显然，我正在修补的道路是不对的。但是，我不确定如何在不干扰sys.path或PYTHONPATH的情况下使它正确。

有什么线索吗？

Answer 1

据我所知，使用mock，您需要在修补时提供一个虚线名称。幸运的是，每个模块都可以访问一个特殊的模块级变量__name__，该变量包含模块的名称。使用此选项，如果要修补模块的本地变量，则应该能够执行以下操作：

import mock
import unittest

ook = lambda: "the ook"


class OokTest(unittest.TestCase):

    def test_ook(self):
        with mock.patch(__name__ + '.ook', return_value=None):
            self.assertIsNone(ook())
        self.assertEquals(ook(), "the ook")

    # the patch decorator should work the same way, I just tend to use the
    # context manager out of personal preference
    @mock.patch(__name__ + '.ook', return_value=None)
    def test_ook_2(self, mock_ook):
        self.assertIsNone(ook())

假设您已将该文件保存为quicktest.py，那么单元测试将给出以下结果：

$ python -m unittest quicktest
..
----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK

当然，from a.b import c在包中给了您一个简单的变量，所以同样的机制也应该起作用。