Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Attempt to prevent network port reuse in tests

pull/3019/head
Kieran Prasch 2022-11-16 15:58:19 +00:00 committed by derekpierre
parent 08a436930c
commit baac3db5fa
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/utils/ursula.py
View File

@ -11,6 +11,8 @@ from nucypher.characters.lawful import Ursula
from nucypher.config.characters import UrsulaConfiguration
from tests.constants import NUMBER_OF_URSULAS_IN_DEVELOPMENT_NETWORK
__ACTIVE_PORTS = set()
def select_test_port() -> int:
"""
@ -24,10 +26,14 @@ def select_test_port() -> int:
with contextlib.closing(closed_socket) as open_socket:
open_socket.bind(('localhost', 0))
port = open_socket.getsockname()[1]
if port == UrsulaConfiguration.DEFAULT_REST_PORT or port > 64000:
rules = (
(port in __ACTIVE_PORTS),
(port == UrsulaConfiguration.DEFAULT_REST_PORT),
(port > 64000)
)
if any(rules):
return select_test_port()
__ACTIVE_PORTS.add(port)
open_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
return port
@ -135,4 +141,4 @@ def start_pytest_ursula_services(ursula: Ursula) -> Certificate:
MOCK_KNOWN_URSULAS_CACHE = dict()
MOCK_URSULA_STARTING_PORT = 51000 # select_test_port()
MOCK_URSULA_STARTING_PORT = select_test_port()