Fixed telegram tests

jertel · Nov 27, 2023 · 1b4e7c1 · 1b4e7c1
1 parent af0f8a5
commit 1b4e7c1
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/tests/alerters/telegram_test.py b/tests/alerters/telegram_test.py
@@ -33,10 +33,10 @@ def test_telegram_thread_id(caplog):
         alert.alert([match])
     expected_data = {
         'chat_id': rule['telegram_room_id'],
-        'text': '⚠ Test Telegram Rule ⚠ \nTest Telegram Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n',
+        'text': '⚠ *Test Telegram Rule* ⚠ ```\nTest Telegram Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n ```',
         'parse_mode': 'markdown',
         'disable_web_page_preview': True,
-        'message_thread_id' : 2
+        'message_thread_id': 2
     }
 
     mock_post_request.assert_called_once_with(
@@ -74,7 +74,8 @@ def test_telegram_markdown(caplog):
         'chat_id': rule['telegram_room_id'],
         'text': '⚠ *Test Telegram Rule* ⚠ ```\nTest Telegram Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n ```',
         'parse_mode': 'markdown',
-        'disable_web_page_preview': True
+        'disable_web_page_preview': True,
+        'message_thread_id': None
     }
 
     mock_post_request.assert_called_once_with(
@@ -113,7 +114,8 @@ def test_telegram_html(caplog):
         'chat_id': rule['telegram_room_id'],
         'text': '⚠ Test Telegram Rule ⚠ \nTest Telegram Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n',
         'parse_mode': 'html',
-        'disable_web_page_preview': True
+        'disable_web_page_preview': True,
+        'message_thread_id': None
     }
 
     mock_post_request.assert_called_once_with(
@@ -153,7 +155,8 @@ def test_telegram_proxy():
         'chat_id': rule['telegram_room_id'],
         'text': '⚠ *Test Telegram Rule* ⚠ ```\nTest Telegram Rule\n\n@timestamp: 2021-01-01T00:00:00\nsomefield: foobarbaz\n ```',
         'parse_mode': 'markdown',
-        'disable_web_page_preview': True
+        'disable_web_page_preview': True,
+        'message_thread_id': None
     }
 
     mock_post_request.assert_called_once_with(
@@ -190,7 +193,8 @@ def test_telegram_text_maxlength():
         'text': '⚠ *Test Telegram Rule' + ('a' * 3979) +
                 '\n⚠ *message was cropped according to telegram limits!* ⚠ ```',
         'parse_mode': 'markdown',
-        'disable_web_page_preview': True
+        'disable_web_page_preview': True,
+        'message_thread_id': None
     }
 
     mock_post_request.assert_called_once_with(
@@ -315,7 +319,8 @@ def test_telegram_matchs():
                 '----------------------------------------\n' +
                 ' ```',
         'parse_mode': 'markdown',
-        'disable_web_page_preview': True
+        'disable_web_page_preview': True,
+        'message_thread_id': None
     }
 
     mock_post_request.assert_called_once_with(