| | """Integration tests for simulation API endpoints.""" |
| |
|
| | from __future__ import annotations |
| |
|
| | import pytest |
| | from rest_framework import status |
| | from rest_framework.test import APIClient |
| |
|
| | from apps.profiles.models import DailyProgress, Profile, WallSection |
| |
|
| |
|
| | @pytest.mark.django_db |
| | class TestSimulationAPI: |
| | """Test cases for simulation API endpoints.""" |
| |
|
| | def test_simulate_endpoint_creates_profiles_and_sections( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test that simulation endpoint creates profiles and sections.""" |
| | config_data = { |
| | "config": "21 25 28\n17\n17 22 17 19 17", |
| | "num_teams": 4, |
| | "start_date": "2025-10-20", |
| | } |
| |
|
| | response = api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | assert response.status_code == status.HTTP_201_CREATED |
| | assert Profile.objects.count() == 3 |
| | assert WallSection.objects.count() == 9 |
| |
|
| | def test_simulate_endpoint_runs_simulation( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test that simulation endpoint creates daily progress records.""" |
| | config_data = { |
| | "config": "28 29", |
| | "num_teams": 2, |
| | "start_date": "2025-10-20", |
| | } |
| |
|
| | response = api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | assert response.status_code == status.HTTP_201_CREATED |
| | assert DailyProgress.objects.count() > 0 |
| |
|
| | assert response.data["total_sections"] == 2 |
| | assert int(response.data["total_days"]) > 0 |
| |
|
| | def test_simulate_endpoint_validates_config( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test that simulation endpoint validates config.""" |
| | config_data = { |
| | "config": "21 abc 28", |
| | "num_teams": 2, |
| | } |
| |
|
| | response = api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | assert response.status_code == status.HTTP_400_BAD_REQUEST |
| | assert "config" in response.data |
| | assert "Invalid number format" in str(response.data["config"]) |
| |
|
| | def test_simulate_endpoint_requires_config( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test that simulation endpoint requires config.""" |
| | config_data = { |
| | "num_teams": 2, |
| | } |
| |
|
| | response = api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | assert response.status_code == status.HTTP_400_BAD_REQUEST |
| | assert "config" in response.data |
| | assert "config parameter is required" in str(response.data["config"]) |
| |
|
| | def test_simulate_endpoint_validates_num_teams( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test that simulation endpoint validates num_teams.""" |
| | config_data = { |
| | "config": "21 25 28", |
| | "num_teams": -1, |
| | } |
| |
|
| | response = api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | assert response.status_code == status.HTTP_400_BAD_REQUEST |
| | assert "num_teams" in response.data |
| | assert "must be at least" in str(response.data["num_teams"]) |
| |
|
| | def test_days_endpoint_returns_ice_usage( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test GET /profiles/<id>/days/<day>/ endpoint.""" |
| | config_data = { |
| | "config": "28 29", |
| | "num_teams": 2, |
| | "start_date": "2025-10-20", |
| | } |
| |
|
| | api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | profile = Profile.objects.first() |
| | assert profile is not None |
| |
|
| | response = api_client.get(f"/api/profiles/{profile.id}/days/1/") |
| |
|
| | assert response.status_code == status.HTTP_200_OK |
| | assert "day" in response.data |
| | assert "total_feet_built" in response.data |
| | assert "total_ice_cubic_yards" in response.data |
| | assert "sections" in response.data |
| | assert response.data["day"] == 1 |
| | assert isinstance(response.data["sections"], list) |
| | assert len(response.data["sections"]) == 2 |
| |
|
| | def test_days_endpoint_returns_404_for_no_data( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test days endpoint returns 404 when no simulation data.""" |
| | profile = Profile.objects.create(name="Test", team_lead="Lead") |
| |
|
| | response = api_client.get(f"/api/profiles/{profile.id}/days/1/") |
| |
|
| | assert response.status_code == status.HTTP_404_NOT_FOUND |
| |
|
| | def test_overview_by_day_endpoint( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test GET /profiles/<id>/overview/<day>/ endpoint.""" |
| | config_data = { |
| | "config": "28 29", |
| | "num_teams": 2, |
| | "start_date": "2025-10-20", |
| | } |
| |
|
| | api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | profile = Profile.objects.first() |
| | assert profile is not None |
| |
|
| | response = api_client.get(f"/api/profiles/{profile.id}/overview/1/") |
| |
|
| | assert response.status_code == status.HTTP_200_OK |
| | assert "day" in response.data |
| | assert "cost" in response.data |
| | assert response.data["day"] == "1" |
| |
|
| | def test_overview_all_by_day_endpoint( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test GET /profiles/overview/<day>/ endpoint.""" |
| | config_data = { |
| | "config": "28\n29", |
| | "num_teams": 2, |
| | "start_date": "2025-10-20", |
| | } |
| |
|
| | api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | response = api_client.get("/api/profiles/overview/1/") |
| |
|
| | assert response.status_code == status.HTTP_200_OK |
| | assert "day" in response.data |
| | assert "cost" in response.data |
| | assert response.data["day"] == "1" |
| |
|
| | def test_overview_total_endpoint( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test GET /profiles/overview/ endpoint returns total days. |
| | |
| | The endpoint now calculates and returns the actual number of |
| | construction days from DailyProgress records instead of None. |
| | """ |
| | config_data = { |
| | "config": "28 29", |
| | "num_teams": 2, |
| | "start_date": "2025-10-20", |
| | } |
| |
|
| | api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | response = api_client.get("/api/profiles/overview/") |
| |
|
| | assert response.status_code == status.HTTP_200_OK |
| | assert "day" in response.data |
| | assert "cost" in response.data |
| | |
| | assert response.data["day"] > 0 |
| | assert isinstance(response.data["day"], int) |
| |
|
| | def test_spec_example_simulation( |
| | self, |
| | api_client: APIClient, |
| | ) -> None: |
| | """Test simulation with spec example data.""" |
| | config_data = { |
| | "config": "21 25 28\n17\n17 22 17 19 17", |
| | "num_teams": 20, |
| | "start_date": "2025-10-20", |
| | } |
| |
|
| | response = api_client.post("/api/profiles/simulate/", config_data, format="json") |
| |
|
| | assert response.status_code == status.HTTP_201_CREATED |
| | assert response.data["total_sections"] == 9 |
| |
|
| | profile_1 = Profile.objects.get(name="Profile 1") |
| |
|
| | day_1_response = api_client.get(f"/api/profiles/{profile_1.id}/days/1/") |
| | assert day_1_response.status_code == status.HTTP_200_OK |
| | assert day_1_response.data["total_ice_cubic_yards"] == "585" |
| |
|
| | overview_response = api_client.get("/api/profiles/overview/") |
| | assert overview_response.status_code == status.HTTP_200_OK |
| |
|
