Question

假设我们有一个仅在生产阶段存在的模块系统。在测试时，这些模块不存在。但我仍然想为使用这些模块的代码编写测试。我们还假设我知道如何从这些模块中模拟所有必需的对象。问题是：我如何方便地将模块存根添加到当前层次结构中？

这是一个小例子。我想测试的功能放在一个名为actual.py的文件中：

actual.py:


def coolfunc():
  from level1.level2.level3_1 import thing1
  from level1.level2.level3_2 import thing2
  do_something(thing1)
  do_something_else(thing2)

在我的测试套件中，我已经拥有了我需要的一切：我有thing1_mock和thing2_mock。我也有测试功能。我需要的是将level1.level2...添加到当前模块系统中。像这样：

tests.py

import sys
import actual

class SomeTestCase(TestCase):
  thing1_mock = mock1()
  thing2_mock = mock2()

  def setUp(self):
    sys.modules['level1'] = what should I do here?

  @patch('level1.level2.level3_1.thing1', thing1_mock)
  @patch('level1.level2.level3_1.thing1', thing2_mock)
  def test_some_case(self):
    actual.coolfunc()

我知道我可以用包含另一个对象的对象替换sys.modules['level1']，依此类推。但对我来说似乎有很多代码。我认为必须有更简单，更漂亮的解决方案。我找不到它。

Answer 1

所以，没有人帮我解决问题，我决定自己解决。 Here是一个名为surrogate的微型库，允许用户为不存在的模块创建存根。

Lib可以与mock一起使用，如下所示：

from surrogate import surrogate
from mock import patch

@surrogate('this.module.doesnt.exist')
@patch('this.module.doesnt.exist', whatever)
def test_something():
    from this.module.doesnt import exist
    do_something()

首先@surrogate装饰器为不存在的模块创建存根，然后@patch装饰器可以改变它们。正如@patch，@surrogate装饰器可以“复数”使用，因此可以存储多个模块路径。所有存根仅在修饰函数的生命周期内存在。

如果有人使用这个lib，那就太好了：）

Answer 2

您可以在patch()方法中使用“创建”参数，如果该属性不存在，它将强制创建该属性。

如何模拟不存在的模块的层次结构？

2 个答案: