update

tests/test_api.py

@@ -0,0 +1,99 @@
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from fastapi.testclient import TestClient
+from unittest.mock import MagicMock, patch, AsyncMock
+import pytest
+from main import app
+
+# We need to mock the bot before importing or using the app fully, 
+# although patching it contextually is better.
+# Because 'bot' is a global instance in main.py, we should patch it where it is used.
+
+@pytest.fixture
+def client():
+    # Mock the bot instance methods including login/close for lifespan
+    with patch("main.bot") as mock_bot:
+        # Use AsyncMock for async methods
+        
+        mock_bot.login = AsyncMock()
+        mock_bot.close = AsyncMock()
+        mock_bot.send_message = AsyncMock()
+        
+        with TestClient(app) as c:
+            yield c
+
+def test_health_check(client):
+    response = client.get("/health")
+    assert response.status_code == 200
+    assert response.json() == {"status": "ok"}
+
+def test_notify_success(client):
+    # We need to verify that bot.send_message was called.
+    
+    with patch("main.bot.send_message", new_callable=AsyncMock) as mock_send:
+        # Updated payload
+        payload = {
+            "service_name": "TestService",
+            "content": "System is down"
+        }
+        response = client.post("/notify", json=payload)
+        assert response.status_code == 200
+        assert response.json() == {"status": "queued"}
+        
+        # Check formatted message
+        mock_send.assert_called_once()
+        args, kwargs = mock_send.call_args
+        # args[0] is plain_message, args[1] is html_message, args[2] is room_id
+        assert args[0] == "[TestService]\nSystem is down"
+        assert args[1] == "<b>[TestService]</b><br>System is down"
+
+def test_jellyfin_movie(client):
+    with patch("main.bot.send_message", new_callable=AsyncMock) as mock_send:
+        payload = {
+            "notification_type": "ItemAdded",
+            "item_type": "Movie",
+            "name": "The Matrix",
+            "year": 1999,
+            "overview": "A computer hacker..."
+        }
+        response = client.post("/jellyfin", json=payload)
+        assert response.status_code == 200
+        
+        mock_send.assert_called_once()
+        args, _ = mock_send.call_args
+        assert "[Jellyfin]\nNew Movie: The Matrix (1999)\nA computer hacker..." in args[0]
+        assert "<b>[Jellyfin]</b><br>New Movie: The Matrix (1999)<br>A computer hacker..." in args[1]
+
+def test_jellyfin_episode(client):
+    with patch("main.bot.send_message", new_callable=AsyncMock) as mock_send:
+        payload = {
+            "notification_type": "ItemAdded",
+            "item_type": "Episode",
+            "name": "Pilot",
+            "series_name": "Lost",
+            "season": 1,
+            "episode": 1
+        }
+        response = client.post("/jellyfin", json=payload)
+        assert response.status_code == 200
+        
+        mock_send.assert_called_once()
+        args, _ = mock_send.call_args
+        assert "[Jellyfin]\nNew Episode: Lost - S01E01 - Pilot" in args[0]
+
+def test_notify_custom_room(client):
+    with patch("main.bot.send_message", new_callable=AsyncMock) as mock_send:
+        payload = {
+            "service_name": "TestService",
+            "content": "msg",
+            "room_id": "!custom:room"
+        }
+        response = client.post("/notify", json=payload)
+        assert response.status_code == 200
+        mock_send.assert_called_once_with("[TestService]\nmsg", "<b>[TestService]</b><br>msg", "!custom:room")
+
+def test_notify_validation_error(client):
+    response = client.post("/notify", json={"service_name": "JustServiceNoContent"}) 
+    assert response.status_code == 422