morluto/rea
1
0
Fork 0
mirror of https://github.com/morluto/rea.git synced 2026-10-11 21:39:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(test): wait for Hopper reconnect listener readiness (#1885)

Browse Source 
The socket path exists after bind but before listen. Wait up to two seconds
for the initial connection on ConnectionRefusedError, closing each failed
socket. Propagate other errors and keep reconnect attempts immediate.
Preserve authentication, lost replies, expiry, cleanup and thread assertions.

Co-authored-by: factory-droid[bot] <138933559+factory-droid[bot]@users.noreply.github.com>
Co-authored-by: Kaoru0822 <[email protected]>
This commit is contained in:
Ysopwnable
2026-10-12 00:02:01 +08:00
committed by GitHub
co-authored by factory-droid[bot] Kaoru0822
parent 29d2624b34
commit d3e83b0899
1 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/fixtures/hopperBridgeReconnectProbe.py
Vendored
+17 -6
View File
@@ -45,11 +45,21 @@ def probe(path, socket_path, mode):
assert time.monotonic() < deadline, failures
time.sleep(0.005)
def connect():
client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
client.settimeout(2)
client.connect(socket_path)
return client, client.makefile("rwb")
def connect(wait_until_listening=False):
deadline = time.monotonic() + 2 if wait_until_listening else None
while True:
client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
client.settimeout(2)
try:
client.connect(socket_path)
except OSError as error:
client.close()
if (not isinstance(error, ConnectionRefusedError)
or deadline is None or time.monotonic() >= deadline):
raise
time.sleep(0.005)
else:
return client, client.makefile("rwb")
def send(file, method, token="owned-token", params=None):
file.write((json.dumps({"id": 1, "method": method, "token": token, "params": params or {}}) + "\n").encode())
@@ -63,7 +73,8 @@ def probe(path, socket_path, mode):
if "event" not in value:
return value
client, file = connect()
# bind() creates the socket path before listen() makes it connectable.
client, file = connect(wait_until_listening=True)
send(file, "health", "foreign-token" if mode == "unauthorized" else "owned-token")
if mode == "lost-reply":
assert entered.wait(2)