Add support for simulating deployment alerts (robusta-dev#380)

pavangudiwada · Jun 15, 2022 · 4b54392 · 4b54392
1 parent 1d964e5
commit 4b54392
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/playbooks/robusta_playbooks/prometheus_simulation.py b/playbooks/robusta_playbooks/prometheus_simulation.py
@@ -15,6 +15,7 @@ class PrometheusAlertParams(ActionParams):
     alert_name: str
     pod_name: Optional[str] = None
     node_name: Optional[str] = None
+    deployment_name: Optional[str] = None
     namespace: str = "default"
     status: str = "firing"
     severity: str = "error"
@@ -41,6 +42,8 @@ def prometheus_alert(
         labels["pod"] = prometheus_event_data.pod_name
     if prometheus_event_data.node_name is not None:
         labels["node"] = prometheus_event_data.node_name
+    if prometheus_event_data.deployment_name is not None:
+        labels["deployment"] = prometheus_event_data.deployment_name
 
     prometheus_event = AlertManagerEvent(
         **{