mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 22:13:26 +00:00
69730a78cc
Automatically generated by the following script, based on the output
of lint with flake8-comma:
import re
import sys
last_filename = None
last_row = None
lines = []
for msg in sys.stdin:
m = re.match(
r"\x1b\[35mflake8 \|\x1b\[0m \x1b\[1;31m(.+):(\d+):(\d+): (\w+)", msg
)
if m:
filename, row_str, col_str, err = m.groups()
row, col = int(row_str), int(col_str)
if filename == last_filename:
assert last_row != row
else:
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
with open(filename) as f:
lines = f.readlines()
last_filename = filename
last_row = row
line = lines[row - 1]
if err in ["C812", "C815"]:
lines[row - 1] = line[: col - 1] + "," + line[col - 1 :]
elif err in ["C819"]:
assert line[col - 2] == ","
lines[row - 1] = line[: col - 2] + line[col - 1 :].lstrip(" ")
if last_filename is not None:
with open(last_filename, "w") as f:
f.writelines(lines)
Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
40 lines
1.7 KiB
Python
40 lines
1.7 KiB
Python
from zerver.lib.send_email import FromAddress
|
|
from zerver.lib.test_classes import WebhookTestCase
|
|
from zerver.models import Recipient
|
|
from zerver.webhooks.zabbix.view import MISCONFIGURED_PAYLOAD_ERROR_MESSAGE
|
|
|
|
|
|
class ZabbixHookTests(WebhookTestCase):
|
|
STREAM_NAME = 'zabbix'
|
|
URL_TEMPLATE = "/api/v1/external/zabbix?api_key={api_key}&stream={stream}"
|
|
FIXTURE_DIR_NAME = 'zabbix'
|
|
|
|
def test_zabbix_alert_message(self) -> None:
|
|
"""
|
|
Tests if zabbix alert is handled correctly
|
|
"""
|
|
expected_topic = "www.example.com"
|
|
expected_message = "PROBLEM (Average) alert on [www.example.com](https://zabbix.example.com/tr_events.php?triggerid=14032&eventid=10528):\n* Zabbix agent on www.example.com is unreachable for 5 minutes\n* Agent ping is Up (1)"
|
|
self.send_and_test_stream_message('zabbix_alert', expected_topic, expected_message)
|
|
|
|
def test_zabbix_invalid_payload_with_missing_data(self) -> None:
|
|
"""
|
|
Tests if invalid Zabbix payloads are handled correctly
|
|
"""
|
|
self.url = self.build_webhook_url()
|
|
payload = self.get_body('zabbix_invalid_payload_with_missing_data')
|
|
result = self.client_post(self.url, payload, content_type='application/json')
|
|
self.assert_json_error(result, "Invalid payload")
|
|
|
|
expected_message = MISCONFIGURED_PAYLOAD_ERROR_MESSAGE.format(
|
|
bot_name=self.test_user.full_name,
|
|
support_email=FromAddress.SUPPORT,
|
|
).strip()
|
|
|
|
msg = self.get_last_message()
|
|
self.assertEqual(msg.content, expected_message)
|
|
self.assertEqual(msg.recipient.type, Recipient.PERSONAL)
|
|
|
|
def get_body(self, fixture_name: str) -> str:
|
|
return self.webhook_fixture_data("zabbix", fixture_name, file_type="json")
|