fix assertDisconnected

This commit is contained in:
Shivaram Lingamneni
2018-12-30 19:05:13 -05:00
parent 8b1c484ac4
commit f4b65a453d

View File

@ -305,15 +305,9 @@ class BaseServerTestCase(_IrcTestCase):
def assertDisconnected(self, client):
try:
self.getLines(client)
self.sendLine(client, 'PING foo')
while True:
l = self.getLine(client)
self.assertNotEqual(line, '')
m = message_parser.parse_message(l)
self.assertNotEqual(m.command, 'PONG',
'Client not disconnected.')
except socket.error:
self.getMessages(client)
self.getMessages(client)
except (socket.error, ConnectionClosed):
del self.clients[client]
return
else: