检查套接字端口是否可用



情况:

我有一个应用程序,它有一个TCP服务器作为它的一部分(基于socketserver),我想对它进行功能测试。

我为自己创建了一个测试混合体,我在每个测试用例中都使用它:

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中检查一下,以确保测试不会失败,因为应用程序实例的端口有问题。

我做过的研究:

我已经浏览了以下谷歌搜索的第一页:

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

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

平台

这需要是多平台的(至少是Windows、Linux、MacOS)。它必须至少在p3.3+中工作。如果"有一个lib"就足够了;)

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

相关内容

最新更新