Add tests for notifications and relay warnings

This commit is contained in:
thePR0M3TH3AN
2025-07-14 11:51:42 -04:00
parent a502b9f6b4
commit 78fbe5e88f
3 changed files with 72 additions and 0 deletions

View File

@@ -16,3 +16,15 @@ def test_notifications_endpoint(client):
{"level": "WARNING", "message": "m2"},
]
assert api._pm.notifications.empty()
def test_notifications_endpoint_clears_queue(client):
cl, token = client
api._pm.notifications = queue.Queue()
api._pm.notifications.put(SimpleNamespace(message="hi", level="INFO"))
res = cl.get("/api/v1/notifications", headers={"Authorization": f"Bearer {token}"})
assert res.status_code == 200
assert res.json() == [{"level": "INFO", "message": "hi"}]
assert api._pm.notifications.empty()
res = cl.get("/api/v1/notifications", headers={"Authorization": f"Bearer {token}"})
assert res.json() == []