46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""
|
|
Test module for grafana formatter.
|
|
|
|
ref https://grafana.com/docs/grafana/latest/alerting/old-alerting/notifications/#webhook
|
|
"""
|
|
|
|
import unittest
|
|
|
|
import httpx
|
|
import nio
|
|
|
|
from .start import BOT_URL, FULL_ID, KEY, MATRIX_ID, MATRIX_PW, MATRIX_URL
|
|
|
|
|
|
class GrafanaFormatterTest(unittest.IsolatedAsyncioTestCase):
|
|
"""Grafana formatter test class."""
|
|
|
|
async def test_grafana_body(self):
|
|
"""Send a markdown message, and check the result."""
|
|
messages = []
|
|
client = nio.AsyncClient(MATRIX_URL, MATRIX_ID)
|
|
|
|
await client.login(MATRIX_PW)
|
|
room = await client.room_create()
|
|
|
|
with open("tests/example_grafana.json") as f:
|
|
example_grafana_request = f.read()
|
|
self.assertEqual(
|
|
httpx.post(
|
|
f"{BOT_URL}/{room.room_id}",
|
|
params={"formatter": "grafana", "key": KEY},
|
|
content=example_grafana_request,
|
|
).json(),
|
|
{"status": 200, "ret": "OK"},
|
|
)
|
|
|
|
sync = await client.sync()
|
|
messages = await client.room_messages(room.room_id, sync.next_batch)
|
|
await client.close()
|
|
|
|
message = messages.chunk[0]
|
|
self.assertEqual(message.sender, FULL_ID)
|
|
self.assertEqual(
|
|
message.body,
|
|
"#### [Alerting] Panel Title alert\nNotification Message\n\n* Count: 1\n",
|
|
)
|