Handle connection reset in selftest
authorJeremy Stanley <fungi@yuggoth.org>
Thu, 9 Nov 2017 09:58:38 +0000 (09:58 +0000)
committerJeremy Stanley <fungi@yuggoth.org>
Thu, 9 Nov 2017 09:58:38 +0000 (09:58 +0000)
Within the selftest framework, catch ConnectionResetError exceptions
and provide a clearer error.

mudpy/tests/selftest.py

index 6239599..02d98a5 100644 (file)
@@ -260,6 +260,10 @@ def main():
                 index, match, received = lusers[conversant].expect(
                     [re.compile(question.encode("utf-8"), flags=re.DOTALL)], 5)
                 captures[conversant] += received.decode("utf-8")
                 index, match, received = lusers[conversant].expect(
                     [re.compile(question.encode("utf-8"), flags=re.DOTALL)], 5)
                 captures[conversant] += received.decode("utf-8")
+            except ConnectionResetError:
+                print("ERROR: Unable to connect to server.")
+                success = False
+                break
             except EOFError:
                 print("ERROR: luser%s premature disconnection expecting:\n\n"
                       "%s\n\n"
             except EOFError:
                 print("ERROR: luser%s premature disconnection expecting:\n\n"
                       "%s\n\n"