| import pytest |
| from fastapi.testclient import TestClient |
| from backend.api_server import app |
| import backend.api_server as api_server |
| import unittest.mock as mock |
|
|
| |
| client = TestClient(app) |
|
|
| |
| api_server.predictor = mock.Mock() |
|
|
| def test_health_endpoint(): |
| """Verify that the health check returns expected status and timestamp.""" |
| response = client.get("/health") |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["status"] in ["healthy", "degraded"] |
| assert "components" in data |
|
|
| def test_predict_endpoint_no_data(): |
| """Verify 404 behavior when a match_id does not exist in Redis.""" |
| |
| api_server.predictor.predict_live_match.return_value = {"error": "No live data available"} |
| |
| response = client.get("/predict/invalid_match_id") |
| assert response.status_code == 404 |
| assert response.json()["detail"] == "No live data available" |
|
|
| def test_list_matches_empty(): |
| """Verify match listing works even when Redis is empty.""" |
| with mock.patch("redis.Redis.keys") as mock_keys: |
| mock_keys.return_value = [] |
| response = client.get("/matches") |
| assert response.status_code == 200 |
| assert response.json() == [] |
|
|
| @pytest.mark.asyncio |
| async def test_predict_success(): |
| """Verify 200 OK and JSON structure for a successful prediction flow.""" |
| mock_result = { |
| "match_id": "test_match", |
| "win_probability": 0.65, |
| "confidence": 0.85, |
| "balls_analyzed": 10 |
| } |
| |
| api_server.predictor.predict_live_match.return_value = mock_result |
| |
| response = client.get("/predict/test_match") |
| assert response.status_code == 200 |
| assert response.json()["win_probability"] == 0.65 |
|
|
|
|
| def test_upcoming_schedule(): |
| """Verify upcoming matches endpoint returns schedule data from the CSV.""" |
| response = client.get("/upcoming/2026") |
| assert response.status_code == 200 |
| data = response.json() |
| assert "matchschedule" in data |
| |
| assert isinstance(data["matchschedule"], list) |
| if len(data["matchschedule"]) > 0: |
| match = data["matchschedule"][0] |
| assert "teama" in match |
| assert "teamb" in match |
| assert "venue" in match |
| assert "matchdate" in match |
|
|
|
|
| def test_points_table(): |
| """Verify points table endpoint returns all 10 IPL teams.""" |
| response = client.get("/points/2026") |
| assert response.status_code == 200 |
| data = response.json() |
| assert "points" in data |
| assert len(data["points"]) == 10 |
| team = data["points"][0] |
| assert "name" in team |
| assert "teamshortname" in team |
| assert "matchesplayed" in team |
| assert "points" in team |
|
|
