From 85dc8a2636bebb876dac8fe42a7884a471afefb0 Mon Sep 17 00:00:00 2001 From: Shivaram Lingamneni Date: Sun, 17 Feb 2019 15:18:52 -0500 Subject: [PATCH] deflake registration tests by waiting for quit --- irctest/controllers/oragono.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/irctest/controllers/oragono.py b/irctest/controllers/oragono.py index 54e435b..e31d27a 100644 --- a/irctest/controllers/oragono.py +++ b/irctest/controllers/oragono.py @@ -148,13 +148,13 @@ class OragonoController(BaseServerController, DirectoryBasedController): case.sendLine(client, 'CAP END') while case.getRegistrationMessage(client).command != '001': pass - list(case.getMessages(client)) + case.getMessages(client) case.sendLine(client, 'ACC REGISTER {} * {}'.format( username, password)) msg = case.getMessage(client) assert msg.command == '920', msg - list(case.getMessages(client)) - case.removeClient(client) + case.sendLine(client, 'QUIT') + case.assertDisconnected(client) def get_irctest_controller_class(): return OragonoController