minor changes to cluster status tests (#891)

Co-authored-by: Alexandra Belousov <sashabelousovrh@Alexandras-MacBook-Pro.local>
run-house · Jun 13, 2024 · 64609cb · 64609cb
1 parent 42d9cb5
commit 64609cb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_resources/test_clusters/test_cluster.py b/tests/test_resources/test_clusters/test_cluster.py
@@ -663,7 +663,7 @@ def test_send_status_to_db(self, cluster):
             headers=headers,
         )
         assert get_status_data_resp.status_code == 200
-        get_status_data = get_status_data_resp.json()["data"]
+        get_status_data = get_status_data_resp.json()["data"][0]
         assert get_status_data["resource_type"] == status.get("cluster_config").get(
             "resource_type"
         )
@@ -681,7 +681,7 @@ def test_send_status_to_db(self, cluster):
             f"{api_server_url}/resource/{cluster_uri}/cluster/status",
             headers=headers,
         )
-        assert get_status_data_resp.json()["data"]["status"] == "terminated"
+        assert get_status_data_resp.json()["data"][0]["status"] == "terminated"
 
     @pytest.mark.level("minimal")
     def test_status_scheduler_basic_flow(self, cluster):
@@ -715,4 +715,4 @@ def test_status_scheduler_basic_flow(self, cluster):
         )
 
         assert get_status_data_resp.status_code == 200
-        assert get_status_data_resp.json()["data"]["status"] == "running"
+        assert get_status_data_resp.json()["data"][0]["status"] == "running"