如何正确地进行单元测试来清理套接字

Question

我最近一直在使用一些套接字，在使用侦听套接字编写一些单元测试用例时，我反复得到error: [Errno 98] Address already in use。

这是一些显示错误的示例代码。

import unittest
import socket

class TestUnit(unittest.TestCase):
    def setUp(self):
        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.socket.bind((socket.gethostname(), 10000))
        self.socket.listen(10)
        self.addCleanup(self.clean)

    def test_nothing(self):
        self.assertEqual(False, False)

    def test_something(self):
        self.assertEqual(True, True)

    def clean(self):
        self.socket.close()

它似乎发生在其中一个测试抛出异常的时候。没有任何异常，它可以按预期工作。但这在某种程度上让测试变得无用，因为在第一个抛出异常的测试之后，所有的测试都会抛出异常。

Answer 1

socket.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1) 

应该会有帮助

基本上，关闭的套接字不会立即被堆栈释放。因此，如果您尝试立即重用它(即使在只有一个绑定套接字，但关闭并重新启动应用程序的情况下也是如此)，您将看到相同的错误。REUSEADDR允许再次绑定同一套接字。

但是，如果您的套接字处于定时等待状态，并且您尝试相同的目标，则它将失败。

您还应该阅读此套接字选项的手册页，以了解它的限制。

SO_REUSEADDR on SO