feat: add supervisor id to DAQJobAlertSlack

ENRG-tr · Nov 15, 2024 · e6d31b1 · e6d31b1
1 parent 7fe318b
commit e6d31b1
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/daq/alert/alert_slack.py b/src/daq/alert/alert_slack.py
@@ -43,6 +43,13 @@ def send_webhook(self, alert: DAQJobMessageAlert):
                     "author_name": alert.daq_job_info.daq_job_class_name,
                     "title": "Alert!",
                     "fields": [
+                        {
+                            "title": "Supervisor ID",
+                            "value": alert.daq_job_info.supervisor_config.supervisor_id
+                            if alert.daq_job_info.supervisor_config
+                            else "-",
+                            "short": True,
+                        },
                         {
                             "title": "Severity",
                             "value": alert.alert_info.severity,

diff --git a/src/tests/test_slack.py b/src/tests/test_slack.py
@@ -31,6 +31,7 @@ def test_send_alert(self):
             date=datetime(2023, 10, 1, 12, 0, 0),
         )
         self.daq_job.send_webhook(alert)
+        assert alert.daq_job_info is not None
         self.mock_slack.post.assert_called_once_with(
             attachments=[
                 {
@@ -39,6 +40,11 @@ def test_send_alert(self):
                     "author_name": alert.daq_job_info.daq_job_class_name,
                     "title": "Alert!",
                     "fields": [
+                        {
+                            "title": "Supervisor ID",
+                            "value": "mock",
+                            "short": True,
+                        },
                         {
                             "title": "Severity",
                             "value": DAQAlertSeverity.ERROR,