diff --git a/fuelclient/tests/unit/v2/lib/test_deployment_graph.py b/fuelclient/tests/unit/v2/lib/test_deployment_graph.py index d886b76d..e045ad5d 100644 --- a/fuelclient/tests/unit/v2/lib/test_deployment_graph.py +++ b/fuelclient/tests/unit/v2/lib/test_deployment_graph.py @@ -192,7 +192,7 @@ class TestDeploymentGraphFacade(test_api.BaseLibTest): def test_graphs_download_cluster(self): matcher_get = self.m_request.get( '/api/v1/clusters/1/deployment_graphs/custom_graph', - json={'tasks': []} + json=[{'tasks': []}] ) self.client.download(env_id=1, level='cluster', graph_type='custom_graph') diff --git a/fuelclient/v1/graph.py b/fuelclient/v1/graph.py index fcb4cb7d..fe5526c2 100644 --- a/fuelclient/v1/graph.py +++ b/fuelclient/v1/graph.py @@ -143,7 +143,7 @@ class GraphClient(base_v1.BaseV1Client): 'cluster': lambda: self.get_graph_for_model( related_model='clusters', related_model_id=env_id, - graph_type=graph_type).get('tasks', []), + graph_type=graph_type)[0].get('tasks', []), 'plugins': lambda: self.get_merged_plugins_tasks( env_id=env_id,