From ee2e77b1e6089925ca037f919bb6f7398c1d0bc1 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 b08a46f..9f3b564 100644 --- a/irctest/controllers/oragono.py +++ b/irctest/controllers/oragono.py @@ -132,13 +132,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