56 lines
2.1 KiB
Python
56 lines
2.1 KiB
Python
|
import unittest
|
||
|
|
from uptime_kuma_test_case import UptimeKumaTestCase
|
||
|
|
from uptime_kuma_api import UptimeKumaException
|
||
|
|
|
||
|
|
|
||
|
|
class TestNotification(UptimeKumaTestCase):
|
||
|
|
def test_notification(self):
|
||
|
|
expected_notification = {
|
||
|
|
"name": "notification 1",
|
||
|
|
"default": True,
|
||
|
|
"apply_existing": True,
|
||
|
|
"type_": "push_by_techulus",
|
||
|
|
"push_by_techulus_apikey": "123456789"
|
||
|
|
}
|
||
|
|
|
||
|
|
# test notification
|
||
|
|
with self.assertRaisesRegex(UptimeKumaException, r'Invalid API key'):
|
||
|
|
self.api.test_notification(**expected_notification)
|
||
|
|
|
||
|
|
# add notification
|
||
|
|
r = self.api.add_notification(**expected_notification)
|
||
|
|
self.assertEqual(r["msg"], "Saved")
|
||
|
|
notification_id = r["id"]
|
||
|
|
|
||
|
|
# get notification
|
||
|
|
notification = self.api.get_notification(notification_id)
|
||
|
|
self.compare(notification, expected_notification)
|
||
|
|
|
||
|
|
# get notifications
|
||
|
|
notifications = self.api.get_notifications()
|
||
|
|
notification = self.find_by_id(notifications, notification_id)
|
||
|
|
self.assertIsNotNone(notification)
|
||
|
|
self.compare(notification, expected_notification)
|
||
|
|
|
||
|
|
# edit notification
|
||
|
|
expected_notification["name"] = "notification 1 new"
|
||
|
|
expected_notification["default"] = False
|
||
|
|
expected_notification["apply_existing"] = False
|
||
|
|
expected_notification["type_"] = "push_deer"
|
||
|
|
expected_notification["push_deer_deer_key"] = "987654321"
|
||
|
|
del expected_notification["push_by_techulus_apikey"]
|
||
|
|
r = self.api.edit_notification(notification_id, **expected_notification)
|
||
|
|
self.assertEqual(r["msg"], "Saved")
|
||
|
|
notification = self.api.get_notification(notification_id)
|
||
|
|
self.compare(notification, expected_notification)
|
||
|
|
|
||
|
|
# delete notification
|
||
|
|
r = self.api.delete_notification(notification_id)
|
||
|
|
self.assertEqual(r["msg"], "Deleted")
|
||
|
|
with self.assertRaises(UptimeKumaException):
|
||
|
|
self.api.delete_notification(notification_id)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|