differentiate test messages from prod

ASFHyP3 · Apr 8, 2024 · 0fc5a3e · 0fc5a3e
1 parent 3893cf1
commit 0fc5a3e
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/status-messages/src/status_messages.py b/status-messages/src/status_messages.py
@@ -42,10 +42,16 @@ def lambda_handler(event: dict, context: dict) -> None:
     print(response)
 
     channel_info = mattermost.channels.get_channel_by_name_and_team_name('asf', CHANNEL)
-    dead_letter_queue_count = get_queue_count()
-    alert_emoji = ':alert:' if dead_letter_queue_count != '0' else ':large_green_circle:'
+    dead_letter_queue_count = int(get_queue_count())
+
+    queue_name = Path(QUEUE_URL).name
+    if 'test' in queue_name:
+        status_emoji = ':heavy_multiplication_x:' if dead_letter_queue_count != 0 else ':heavy_check_mark:'
+    else:
+        status_emoji = ':alert:' if dead_letter_queue_count != 0 else ':large_green_circle:'
+
     mattermost_message = (
-        f'{alert_emoji} Dead Letter Queue {Path(QUEUE_URL).name} Count for ITS_LIVE has '
+        f'{status_emoji} Dead Letter Queue {queue_name} Count for ITS_LIVE has '
         f'{dead_letter_queue_count} entries on {datetime.now(tz=timezone.utc).isoformat()}'
     )
     response = mattermost.posts.create_post(