模拟离开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 the os.environ：

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