def clientConnectionLost(self, client): class ReasonMockUp: def __str__(self): return "you mock me" def check(self, foo): return False log_history.reset() client.transport.loseConnection() client.connectionLost(ReasonMockUp()) self.assertEquals(client._keepalive_timer, None) self.assertEquals(self.client_factory[0].protocol_instance, None) return True
def clientConnectionLost(self, client): class ReasonMockUp: def __str__(self): return "you mock me" def check(self, foo): return False silence_all_messages() clear_all_messages() client.connectionLost(ReasonMockUp()) self.assertEquals(get_messages(), [ 'connectionLost: reason = you mock me', 'UGAMEClient.connectionLost you mock me' ]) self.assertEquals(client._ping_timer, None) self.assertEquals(self.client_factory[0].protocol_instance, None) return True