从os.environ模拟密钥

Question

我正在尝试测试一个函数，如果它在os.environ中找不到特定的键，则应该返回OSError，但是我不知道该如何模拟它。 我也完全可以模拟os.environ完全为空。

with mock.patch.dict('os.environ', dict()): print(os.environ)

要么

with mock.patch.dict(os.environ, dict()): print(os.environ)

无论如何都会打印我的整个环境。

Answer 1

我改用MonkeyPatching os.environ：

old_os_environ = os.environ try: os.environ = {} self.assertRaises(OSError, my_function) finally: os.environ = old_os_environ