def test_deserialize_on_creation():
serializer = MagicMock()
serializer.deserialize_state.return_value = [("py", None), ("sc", None)]
manager = ClientManager(serializer)
assert "py" in manager.get_sessions_list()
assert "sc" in manager.get_sessions_list()
serializer = MagicMock()
manager = ClientManager(serializer)
assert len(manager.get_sessions_list()) == 0
def test_get_session_id_for_client_not_there():
manager = ClientManager()
manager.get_sessions_list = MagicMock(return_value=[])
id = manager.get_session_id_for_client("name")
assert id is None
def test_client_names_returned():
client = MagicMock()
manager = ClientManager()
manager.add_client("name0", client)
manager.add_client("name1", client)
assert_equals({"name0", "name1"}, set(manager.get_sessions_list()))
def test_get_session_id_for_client():
manager = ClientManager()
manager.get_sessions_list = MagicMock(return_value=["name"])
manager._livy_clients["name"] = MagicMock()
id = manager.get_session_id_for_client("name")
assert id is not None