extends route tests

Author: GabrielBarberini

tests/test_routes/test_environment_route.py

@@ -1,6 +1,7 @@
 from unittest.mock import patch
 import pytest
 from fastapi.testclient import TestClient
+from fastapi import HTTPException
 from lib.models.environment import Env
 from lib.controllers.environment import EnvController
 from lib.views.environment import (
@@ -15,12 +16,12 @@
 
 
 @pytest.fixture
-def mock_env():
+def stub_env():
     return Env(latitude=0, longitude=0)
 
 
 @pytest.fixture
-def mock_env_summary():
+def stub_env_summary():
     return EnvSummary()
 
 
@@ -38,18 +39,49 @@ def test_create_env():
         }
         mock_create_env.assert_called_once_with(Env(latitude=0, longitude=0))
 
+def test_create_env_invalid_input():
+    response = client.post("/environments/", json={"latitude": "foo", "longitude": "bar"})
+    assert response.status_code == 422
 
-def test_read_env(mock_env):
+def test_create_env_server_error():
     with patch.object(
-        EnvController, "get_env_by_id", return_value=mock_env
+        EnvController, "create_env", side_effect=Exception("error")
+    ) as mock_create_env:
+        with pytest.raises(Exception):
+            response = client.post(
+                "/environments/", json={"latitude": 0, "longitude": 0}
+            )
+            assert response.status_code == 500
+            assert response.json() == {"detail": "Failed to create environment: error"}
+
+def test_read_env(stub_env):
+    with patch.object(
+        EnvController, "get_env_by_id", return_value=stub_env
     ) as mock_read_env:
         response = client.get("/environments/123")
         assert response.status_code == 200
-        expected_content = mock_env.model_dump()
+        expected_content = stub_env.model_dump()
         expected_content["date"] = expected_content["date"].isoformat()
         assert response.json() == expected_content
         mock_read_env.assert_called_once_with("123")
 
+def test_read_env_not_found():
+    with patch.object(
+        EnvController, "get_env_by_id", side_effect=HTTPException(status_code=404, detail="Environment not found")
+    ) as mock_read_env:
+        response = client.get("/environments/123")
+        assert response.status_code == 404
+        assert response.json() == {"detail": "Environment not found"}
+        mock_read_env.assert_called_once_with("123")
+
+def test_read_env_server_error():
+    with patch.object(
+        EnvController, "get_env_by_id", side_effect=Exception("error")
+    ) as mock_read_env:
+        with pytest.raises(Exception):
+            response = client.get("/environments/123")
+            assert response.status_code == 500
+            assert response.json() == {"detail": "Failed to read environment: error"}
 
 def test_update_env():
     with patch.object(
@@ -69,6 +101,37 @@ def test_update_env():
             "123", Env(latitude=1, longitude=1)
         )
 
+def test_update_env_invalid_input():
+    response = client.put("/environments/123", json={"latitude": "foo", "longitude": "bar"})
+    assert response.status_code == 422
+
+def test_update_env_not_found():
+    with patch.object(
+        EnvController,
+        "update_env_by_id",
+        side_effect=HTTPException(status_code=404, detail="Environment not found"),
+    ) as mock_update_env:
+        response = client.put(
+            "/environments/123", json={"longitude": 1, "latitude": 1}
+        )
+        assert response.status_code == 404
+        assert response.json() == {"detail": "Environment not found"}
+        mock_update_env.assert_called_once_with(
+            "123", Env(latitude=1, longitude=1)
+        )
+
+def test_update_env_server_error():
+    with patch.object(
+        EnvController,
+        "update_env_by_id",
+        side_effect=Exception("error"),
+    ) as mock_update_env:
+        with pytest.raises(Exception):
+            response = client.put(
+                "/environments/123", json={"longitude": 1, "latitude": 1}
+            )
+            assert response.status_code == 500
+            assert response.json() == {"detail": "Failed to update environment: error"}
 
 def test_delete_env():
     with patch.object(
@@ -84,14 +147,38 @@ def test_delete_env():
         }
         mock_delete_env.assert_called_once_with("123")
 
+def test_delete_env_not_found():
+    with patch.object(
+        EnvController,
+        "delete_env_by_id",
+        return_value=EnvDeleted(env_id="123"),
+    ) as mock_delete_env:
+        response = client.delete("/environments/123")
+        assert response.status_code == 200
+        assert response.json() == {
+            "env_id": "123",
+            "message": "Environment successfully deleted",
+        }
+        mock_delete_env.assert_called_once_with("123")
 
-def test_simulate_env(mock_env_summary):
+def test_delete_env_server_error():
     with patch.object(
-        EnvController, "simulate_env", return_value=mock_env_summary
+        EnvController,
+        "delete_env_by_id",
+        side_effect=Exception("error"),
+    ) as mock_delete_env:
+        with pytest.raises(Exception):
+            response = client.delete("/environments/123")
+            assert response.status_code == 500
+            assert response.json() == {"detail": "Failed to delete environment: error"}
+
+def test_simulate_env(stub_env_summary):
+    with patch.object(
+        EnvController, "simulate_env", return_value=stub_env_summary
     ) as mock_simulate_env:
         response = client.get("/environments/123/summary")
         assert response.status_code == 200
-        expected_content = mock_env_summary.model_dump()
+        expected_content = stub_env_summary.model_dump()
         expected_content["date"] = expected_content["date"].isoformat()
         expected_content["local_date"] = expected_content[
             "local_date"
@@ -102,6 +189,27 @@ def test_simulate_env(mock_env_summary):
         assert response.json() == expected_content
         mock_simulate_env.assert_called_once_with("123")
 
+def test_simulate_env_not_found():
+    with patch.object(
+        EnvController,
+        "simulate_env",
+        side_effect=HTTPException(status_code=404, detail="Environment not found"),
+    ) as mock_simulate_env:
+        response = client.get("/environments/123/summary")
+        assert response.status_code == 404
+        assert response.json() == {"detail": "Environment not found"}
+        mock_simulate_env.assert_called_once_with("123")
+
+def test_simulate_env_server_error():
+    with patch.object(
+        EnvController,
+        "simulate_env",
+        side_effect=Exception("error"),
+    ) as mock_simulate_env:
+        with pytest.raises(Exception):
+            response = client.get("/environments/123/summary")
+            assert response.status_code == 500
+            assert response.json() == {"detail": "Failed to simulate environment: error"}
 
 def test_read_rocketpy_env():
     with patch.object(
@@ -112,3 +220,25 @@ def test_read_rocketpy_env():
         assert response.content == b'rocketpy'
         assert response.headers["content-type"] == "application/octet-stream"
         mock_read_rocketpy_env.assert_called_once_with("123")
+
+def test_read_rocketpy_env_not_found():
+    with patch.object(
+        EnvController,
+        "get_rocketpy_env_binary",
+        side_effect=HTTPException(status_code=404, detail="Environment not found"),
+    ) as mock_read_rocketpy_env:
+        response = client.get("/environments/123/rocketpy")
+        assert response.status_code == 404
+        assert response.json() == {"detail": "Environment not found"}
+        mock_read_rocketpy_env.assert_called_once_with("123")
+
+def test_read_rocketpy_env_server_error():
+    with patch.object(
+        EnvController,
+        "get_rocketpy_env_binary",
+        side_effect=Exception("error"),
+    ) as mock_read_rocketpy_env:
+        with pytest.raises(Exception):
+            response = client.get("/environments/123/rocketpy")
+            assert response.status_code == 500
+            assert response.json() == {"detail": "Failed to read rocketpy environment: error"}