def test_client_disconnection(self): """ Tests if client gets disconnected after TIMEOUT. """ timer = FakeTimer(TIMEOUT + 1) manager = ClientManager(timer) client_id = manager.register("rosetta")[1] self.assertIsNone(manager.get_client(client_id))
def test_add_clients_and_get_them(self): """ Tests if registered client's can be found in manager. """ sample_clients = ["curiosity", "venera", "spirit"] manager = ClientManager() client_with_id = [] for name in sample_clients: state = manager.register(name) self.assertTrue(state[0]) client_with_id.append((name, state[1])) for (name, client_id) in client_with_id: self.assertEqual(name, manager.get_client(client_id).get_name())