Support missing performance data

Its normal for some tests (eg paused, failed etc) to not have any performance data
chelnak · Mar 4, 2024 · 36eb3bb · 36eb3bb
1 parent b3f95e6
commit 36eb3bb
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/status_cake_exporter/_test_collector.py b/status_cake_exporter/_test_collector.py
@@ -78,8 +78,7 @@ def transform(
 
     for i in tests:
         logger.debug(f"Transforming test id: {i['id']}")
-        t.append(
-            {
+        test = {
                 "test_id": str(i["id"]),
                 "test_type": str(
                     i["test_type"]
@@ -88,12 +87,14 @@ def transform(
                 "test_url": i["website_url"],
                 "test_status_int": get_uptime_status(i["status"]),
                 "test_uptime_percent": str(i["uptime"]),
-                "test_performance": str(i["performance"]),
                 "maintenance_status_int": get_test_maintenance_status(
                     i["id"], tests_in_maintenance
                 ),
             }
-        )
+
+        if hasattr(i, 'performance'):
+            test["test_performance"]= str(i["performance"])
+        t.append(test)
         logger.debug(f"Transformed test id: {i['id']}")
 
     logger.debug(f"Test transformation complete. Returning {len(t)} metrics")
@@ -179,7 +180,8 @@ def collect(self):
             )
 
             for i in metrics:
-                performance_gauge.add_metric([i["test_id"]], (i["test_performance"]))
+                if 'test_performance' in i.keys():
+                    performance_gauge.add_metric([i["test_id"]], (i["test_performance"]))
 
             yield performance_gauge