fix test_websocket.py

ThomasFarstrike · ThomasFarstrike · commit ba1452b38513 · 2025-11-12T00:54:12.000+01:00
diff --git a/micropython-nostr b/micropython-nostr
@@ -1 +1 @@
-Subproject commit a3a8759a16e7bf53f51b8f15503ebffa2a12f045
+Subproject commit 6d7281e86a3ccc2dd050a27994170c2952a470b3
diff --git a/tests/test_websocket.py b/tests/test_websocket.py
@@ -74,22 +74,22 @@ async def main(self) -> None:
         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
             await asyncio.sleep(1)
         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:
