Merge pull request #178 from dmitry-lipetsk/D20250131_001--find_free_port

dmitry-lipetsk · web-flow · commit 45f7976743af · 2025-01-31T12:26:53.000+03:00
PortManager::find_free_port is updated
diff --git a/testgres/helpers/port_manager.py b/testgres/helpers/port_manager.py
@@ -26,10 +26,11 @@ def find_free_port(self, ports: Optional[Set[int]] = None, exclude_ports: Option
         if ports is None:
             ports = set(range(1024, 65535))
 
-        if exclude_ports is None:
-            exclude_ports = set()
+        assert type(ports) == set  # noqa: E721
 
-        ports.difference_update(set(exclude_ports))
+        if exclude_ports is not None:
+            assert isinstance(exclude_ports, Iterable)
+            ports.difference_update(exclude_ports)
 
         sampled_ports = random.sample(tuple(ports), min(len(ports), 100))
 
