python 'with'语句及其在类中的使用

Question

我有一个 TestCase，它使用套接字进行测试。由于即使在出现某些错误的情况下也应该始终关闭套接字，因此我创建了一个抽象套接字的上下文管理器类。

测试夹具如下所示，其中 MyClassToTest 是要测试的实际类，它在内部使用套接字设备。

with SocketContextManager() as device:
    foo = MyClassToTest(device)
    # make tests with foo

我想避免在每个测试夹具中重复这两行，但始终将其放入 setUp 中。但是我应该怎么做呢？以下代码

def setUp(self):
    with SocketContextManager() as device:
        self.foo = MyClassToTest(device)

不起作用，因为设备将在 setUp 结束时关闭。有没有办法像这样处理上下文管理器的实例化，或者我是否必须在每个测试装置中重复它？

Answer 1

根据 the documentation对于拆解:

This is called even if the test method raised an exception

因此您可以在setUp 中打开套接字，并在tearDown 中关闭它。即使您的测试用例引发异常，套接字仍将关闭。