Python 3如何为try编写单元测试，模块中的外部函数除外

Question

我想知道如何为try exceptblocks 编写 Python 3 单元测试 在 Python 模块中的函数定义之外定义的。

想象一下，在package/module.py 我有一段代码，例如：

导入操作系统

try:
  CONSTANT = os.environ['NOT_EXISTING_KEY']
except KeyError:
  CONSTANT = False
finally:
  del os

（请不要介意实际代码，我知道在这种特定情况下我可以使用os.getenv('NOT_EXISTING_KEY', False)，我感兴趣的是真正测试模块中的try-except块（函数之外）行为符合预期。

如何编写单元测试来检查 package.module.CONSTANT 是否设置为预期值？

在 unittest 文件中（我使用 pytest）我有类似的东西：

from package.module import CONSTANT

def test_constant_true():
  assert CONSTANT == 'expected_value'

测试如果 try 块正确执行，那么 CONSTANT 是否符合预期。

但是，我不知道如何模拟导入机制，以便 try 块中的 os.environ 引发异常，我可以测试 CONSTANT 是否设置为 False。

我该怎么做？

Answer 1

您可以使用monkeypatch设置环境变量，但您必须重新加载模块才能使更改生效：

from importlib import reload

from package import module


def test_constant_true(monkeypatch):
    monkeypatch.setenv('MY_KEY', '42')
    reload(module)
    assert module.CONSTANT == '42'


def test_constant_false():
    reload(module)
    assert not module.CONSTANT

鉴于package/module.py的这个内容：

import os

try:
    CONSTANT = os.environ['MY_KEY']
except KeyError:
    CONSTANT = False

Answer 2

您可以使用mock.patch.dict 模拟环境并将值导入您的单元测试方法中。像这样：

from unittest import TestCase, mock

class YourTest(TestCase):
    @mock.patch.dict('os.environ', {'NOT_EXISTING_KEY': 'value'})
    def test_constant_key_defined(self, mocked):
        """ Tests when the key is defined """
        from package.module import CONSTANT
        self.assertEqual(CONSTANT, 'value')

    def test_constant_key_not_defined(self):
        """ Tests when the key is not defined """
        from package.module import CONSTANT
        self.assertEqual(CONSTANT, 'value')

您可以使用importlib.reload，就像@mrbean-bremen 的答案，我不熟悉。