python - 如何对使用环境变量的装饰器进行单元测试？

Question

我创建了一个装饰器以允许函数仅在特定环境中运行:

def accepted_environments(*envs):
    """
    The decorated function can be executed only in specified envs
    """
    def my_decorator(func_to_be_decorated):
        def wrapper():
            if settings.ENV_NAME not in envs:
                raise EnvironmentException
            return func_to_be_decorated()
        return wrapper
    return my_decorator

# Usage example
@accepted_environments('local', 'prod')
def hello():
    print("hello")

这似乎可行，但我想对其进行单元测试。问题是:我的测试可能在每个环境(本地、临时、产品)中运行。另外，我认为我的测试能够更改环境变量并不安全。

我应该“ mock ”这种行为吗？你会怎么做？谢谢!

Answer 1

使用 mock 覆盖 settings.ENV_NAME 的值进行测试。

def test_not_in_dev(self):
    with mock.patch.dict(settings.__dict__, ENV_NAME="dev"):
        self.assertRaises(EnvironmentException, hello)