python - 如何在使用 inlineCallbacks 的试用测试用例中使用 assertRaises

Question

我正在尝试弄清楚如何编写断言引发异常的试验测试用例。

目前我有两种简单的方法来测试(成功和失败)。每个方法都返回一个已经被回调或错误返回的延迟。测试成功方法工作正常。在测试失败方法时，我希望能够断言引发了异常(使用 assertRaises)。

但是测试用例失败了，我得到:

twisted.trial.unittest.FailTest: ConnectionRefusedError not raised (<Deferred at 0x920e28c current result: <twisted.python.failure.Failure <class 'twisted.internet.error.ConnectionRefusedError'>>> returned)

代码如下:

from twisted.trial.unittest import TestCase
from twisted.internet.defer import inlineCallbacks, succeed, fail
from twisted.internet.error import ConnectionRefusedError

class MyObject:
    def success(self):
        return succeed(True)

    def failure(self):
        return fail(ConnectionRefusedError())


class TestErrBack(TestCase):
    def setUp(self):
        self.o = MyObject()

    @inlineCallbacks
    def test_success(self):
        result = yield self.o.success()
        self.assertTrue(result)

    @inlineCallbacks
    def test_failure(self):
        # this test case is failing !
        yield self.assertRaises(ConnectionRefusedError, self.o.failure)

我在 test_failure 中使用了正确的方法吗？ 我可以使用 try...catch 绕过对 self.o.failure 的调用，但我认为这种方法不如使用 assertRaises 好。

Answer 1

改用TestCase.assertFailure:

yield self.assertFailure(self.o.failure(), ConnectionRefusedError)

从 Twisted 12.3 开始，还有一个 TestCase.failureResultOf 助手:

self.failureResultOf(self.o.failure()).trap(ConnectionRefusedError)

从 13.1 开始，此 API 接受一个额外的参数并为您执行类型检查:

self.failureResultOf(self.o.failure(), ConnectionRefusedError)

这对于您知道 Deferred 已经触发并产生结果的测试很有用。如果 Deferred 在调用时没有失败结果，failureResultOf 会引发测试失败异常而不是返回失败。

这适用于您的示例代码并且应该适用于大多数单元测试。如果您正在使用试用来编写功能测试或集成测试，其中有实际的异步工作正在进行并且您不知道 Deferred 何时触发，那么您需要坚持使用第一个 API，断言失败。