fix test_websocket.py

This commit is contained in:
Thomas Farstrike
2025-11-12 00:54:12 +01:00
parent 720fc2f1cb
commit ba1452b385
2 changed files with 6 additions and 6 deletions
+5 -5
View File
@@ -74,7 +74,7 @@ class TestMutlipleWebsocketsAsyncio(unittest.TestCase):
await self.closeall()
for _ in range(10):
print("Waiting for on_open to be called...")
print(f"self.on_open_called: {self.on_open_called} so waiting for on_open to be called...")
if self.on_open_called == min(len(self.relays),self.max_allowed_connections):
print("yes, it was called!")
break
@@ -82,14 +82,14 @@ class TestMutlipleWebsocketsAsyncio(unittest.TestCase):
self.assertTrue(self.on_open_called == min(len(self.relays),self.max_allowed_connections))
for _ in range(10):
print("Waiting for on_close to be called...")
if self.on_close_called == min(len(self.relays),self.max_allowed_connections):
print(f"self.on_close_called: {self.on_close_called} so waiting for on_close to be called...")
if self.on_close_called >= min(len(self.relays),self.max_allowed_connections):
print("yes, it was called!")
break
await asyncio.sleep(1)
self.assertTrue(self.on_close_called == min(len(self.relays),self.max_allowed_connections))
self.assertGreaterEqual(self.on_close_called, min(len(self.relays),self.max_allowed_connections), "on_close was called for less than allowed connections")
self.assertTrue(self.on_error_called == min(len(self.relays),self.max_allowed_connections))
self.assertEqual(self.on_error_called, len(self.relays) - self.max_allowed_connections, "expecting one error per failed connection")
# Wait for *all* of them to finish (or be cancelled)
# If this hangs, it's also a failure: