本文介绍了Python unittest的设置函数不使用在类上声明的模拟的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
所以我在编写单元测试时遇到了设置函数方面的问题。在我看来，它应该只是在你的函数之前执行代码，所以我可以把任何重复的东西放在里面。然而，这个函数似乎并没有将我作为补丁修饰器创建的模拟应用到整个类上。这是我希望它看起来是什么样子的一小部分：
@patch('geomet_data_registry.layer.base.get_today_and_now', new=mocked_get_date) # noqa
@patch('geomet_data_registry.layer.base.load_plugin', new=mocked_load_plugin)
@patch('geomet_data_registry.layer.base.TILEINDEX_PROVIDER_DEF', new=mocked_tileindex) # noqa
@patch('geomet_data_registry.layer.base.STORE_PROVIDER_DEF', new=mocked_store)
class TestInitBase(unittest.TestCase):
def setUp(self):
""" Code that executes before every function. """
self.maxDiff = None
self.today_date =
datetime.now(timezone.utc).isoformat().replace('+00:00', 'Z')
mocked_get_date.return_value = self.today_date
self.base_layer = BaseLayer({'name': 'model_gem_global'})
def test_Init(self):
expected_values = {'items': [],...
base_layer_attr = self.base_layer.__dict__
self.assertDictEqual(expected_values, base_layer_attr, msg=None)
setUp中实例化类BaseLayer时，没有使用我的模拟，我得到：
- 'receive_datetime': '2021-11-10T12:56:07.371067Z',
? ^^^
+ 'receive_datetime': '2021-11-10T12:56:07.371131Z',
? ^^^
- 'store': <MagicMock name='mock()' id='140158154534472'>,
- 'tileindex': <MagicMock name='mock()' id='140158154534472'>,
+ 'store': <BaseStore> Redis,
+ 'tileindex': <ElasticsearchTileIndex> http://localhost:9200,
setUp命名为mySetUp，并在函数开始时调用它。
一切都是这样工作的，我已经进行了所有的测试，而根本没有使用setUp，因为我记得我对自己说，这个东西坏了，我不会冒险在我的测试中使用这个函数。
谢谢！
推荐答案
问题是mock.patch修饰符应用于每个测试函数，而在setUp期间尚未完成修补。
要对所有测试使用相同的模拟，您必须在setUp/tearDown中开始/停止模拟。这可能类似于：
class TestInitBase(unittest.TestCase):
def setUp(self):
self.data_patcher = patch('geomet_data_registry.layer.base.get_today_and_now',
new=mocked_get_date)
self.data_patcher.start()
self.plugin_patcher = patch('geomet_data_registry.layer.base.load_plugin',
new=mocked_load_plugin)
self.plugin_patcher.start()
...
self.today_date =
datetime.now(timezone.utc).isoformat().replace('+00:00', 'Z')
mocked_get_date.return_value = self.today_date
self.base_layer = BaseLayer({'name': 'model_gem_global'})
def tearDown(self):
self.data_patcher.stop()
self.plugin_patcher.stop()
...
诚然，这不如使用装饰符。您仍然可以将修饰符用于不需要在setUp中打补丁的函数(如果有)。
附注：有了pytest，这就不那么麻烦了，因为您可以将安装和拆卸部分放在同一个夹具中。
这篇关于Python unittest的设置函数不使用在类上声明的模拟的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！