检查socket端口是否可用

Question

情况：

我有一个应用程序，其中包含一个 TCP 服务器（基于 socketserver），我想对其进行功能测试。

我为自己创建了一个测试 mixin，我在每个测试用例中都使用它：

class AppProcessManagingMixin:
    _used_tcp_ports = set()

    def start_my_app(self):
        port = self._take_tcp_port()
        prepare_config_with_given_tcp_port(port)
        # use prepared config by subprocess below
        Popen([python_executable_path, my_app_name, ...).start()

    def _take_tcp_port(self):
        available_found = False
        while not available_found:
            port = random.randint(9000, 9999) # these ints are arbitrary here
            available_found = port not in self._used_tcp_ports
        self._used_tcp_ports.add(port)
        return port

    def stop_my_app(self):
        so_something_that_Im_sure_will_stop_my_app_after_a_few_seconds()

每个测试用例setUp() 调用self.start_my_app()，每个tearDown() - self.stop_my_app()，我可以依靠停止我的应用程序。我需要我的应用程序的许多实例（并且无法在类范围的设置/拆卸中启动/停止应用程序），因为我的一些测试正在检查应用程序是否在从空工作目录开始的场景中运行良好（我只需要“空”应用程序实例用于这些测试）。

我对第二种方法有问题，_take_tcp_port。在介绍它之前，我经常遇到已获取地址的问题（因为所有测试都尝试在同一端口上运行 TCP 服务器）。现在它不再经常发生，但仍然 - 它发生了。

我知道它依赖于系统，几秒钟后端口将再次可用，但这会使自动化测试变得更加困难。

问题：

如何检查 TCP 服务器是否有可用的地址（主机和端口）？我想在_take_tcp_port 中检查它，以确保测试不会因为应用实例的占用端口问题而失败。

我所做的研究：

我浏览了以下 google 搜索的第一页：

• 检查python端口是否打开
• 检查python端口是否可用
• 发现可用端口

以及我发现的所有方法都基于尝试获取套接字。问题是，如果我成功获得套接字，我可以确定这个端口是打开的，但是因为我这样做了，我不能确定我的应用程序能够获得它。我需要没有副作用的东西。

平台

这需要是多平台的（至少是 Windows、Linux、MacOS）。它必须至少在 p3.3+ 中工作。如果“有一个库”就足够了；）

Answer 1

您无需手动扫描可用端口 - 操作系统会为您完成。只需在调用 bind() 时指定端口 0，操作系统就会选择一个。这通常对测试很有用——您可以使用端口 0 启动一个端点，让它询问系统它实际获得的端口 (getsockaddr)，然后使用该端口号启动第二个端点。