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