deflake registration tests by waiting for quit

This commit is contained in:
Shivaram Lingamneni 2019-02-17 15:18:52 -05:00
parent 17303fa7fe
commit 85dc8a2636

View File

@ -148,13 +148,13 @@ class OragonoController(BaseServerController, DirectoryBasedController):
case.sendLine(client, 'CAP END') case.sendLine(client, 'CAP END')
while case.getRegistrationMessage(client).command != '001': while case.getRegistrationMessage(client).command != '001':
pass pass
list(case.getMessages(client)) case.getMessages(client)
case.sendLine(client, 'ACC REGISTER {} * {}'.format( case.sendLine(client, 'ACC REGISTER {} * {}'.format(
username, password)) username, password))
msg = case.getMessage(client) msg = case.getMessage(client)
assert msg.command == '920', msg assert msg.command == '920', msg
list(case.getMessages(client)) case.sendLine(client, 'QUIT')
case.removeClient(client) case.assertDisconnected(client)
def get_irctest_controller_class(): def get_irctest_controller_class():
return OragonoController return OragonoController