testQuitErrors: Make it slightly less flaky with solanum?

2021-08-14 11:45:49 +02:00 · 2021-08-14 11:45:49 +02:00 · cc3d70c7d3
parent ff0d795485
commit cc3d70c7d3
1 changed files with 8 additions and 4 deletions
--- a/irctest/server_tests/connection_registration.py
+++ b/irctest/server_tests/connection_registration.py
@ -88,10 +88,14 @@ class ConnectionRegistrationTestCase(cases.BaseServerTestCase):
        self.connectClient("foo")
        self.getMessages(1)
        self.sendLine(1, "QUIT")
+        while True:
            try:
-            commands = {m.command for m in self.getMessages(1)}
+                new_messages = self.getMessages(1)
+                if not new_messages:
+                    break
+                commands = {m.command for m in new_messages}
            except ConnectionClosed:
-            assert False, "Connection closed without ERROR."
+                break
        self.assertIn(
            "ERROR", commands, fail_msg="Did not receive ERROR as a reply to QUIT."
        )