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())
