如何测试在没有环境的情况下会出现的异常

Question

我有一些类似的代码：

if not settings.VAR_URL: 
    raise Exception("VAR_URL is not defined") 

当我尝试像这样测试它时：

def test_for_absent_env(self):
    del os.environ['VAR_URL']
    o = Object()
    with self.assertRaises(Exception) as error:
        o.some_function()

        self.assertEqual(error.exception.message, "VAR_URL is not defined")

但它提供KeyError 而不是通过测试。我应该更正什么？

Answer 1

如果引发异常，您不应该采用这种方式进行测试。有一个特定的断言叫做assertRaises，可以用作上下文管理器（helps to then get the error message to check as well）：

with self.assertRaises(Exception) as error: 
    o.some_cool_function()

    self.assertEqual(error.exception.message, "VAR_URL is not defined")

请注意，在 Python>=3.1 中可以使用 assertRaises 作为上下文管理器。