From 1b4e7c1c517dbd62c45c0df08b2ac2eca451f087 Mon Sep 17 00:00:00 2001 From: Polina Sheviakova Date: Mon, 27 Nov 2023 21:54:32 +0300 Subject: [PATCH] Fixed telegram tests --- tests/alerters/telegram_test.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/tests/alerters/telegram_test.py b/tests/alerters/telegram_test.py index 404116ff..748c72bb 100644 --- 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(