feat(restruct tests):

tests/integration/routes/test_report.py

@@ -0,0 +1,73 @@
+from pathlib import Path
+from unittest.mock import patch
+
+import pytest
+from fastapi import FastAPI
+from fastapi.testclient import TestClient
+
+from agentic_security.routes.report import router
+
+app = FastAPI()
+app.include_router(router)
+client = TestClient(app)
+
+
+@pytest.fixture
+def mock_csv_exists():
+    with patch.object(Path, "exists") as mock:
+        mock.return_value = True
+        yield mock
+
+
+@pytest.fixture
+def mock_csv_not_exists():
+    with patch.object(Path, "exists") as mock:
+        mock.return_value = False
+        yield mock
+
+
+def test_failures_csv_exists(mock_csv_exists):
+    """Test /failures endpoint when CSV file exists"""
+    with patch("agentic_security.routes.report.FileResponse") as mock_response:
+        mock_response.return_value = "mocked_response"
+        response = client.get("/failures")
+        assert response.status_code == 200
+        mock_response.assert_called_once_with("failures.csv")
+
+
+def test_failures_csv_not_exists(mock_csv_not_exists):
+    """Test /failures endpoint when CSV file doesn't exist"""
+    response = client.get("/failures")
+    assert response.status_code == 200
+    assert response.json() == {"error": "No failures found"}
+
+
+@pytest.mark.skip
+def test_get_plot():
+    """Test /plot.jpeg endpoint"""
+    # Mock data matching expected plot_security_report format
+    table_data = [
+        {
+            "module": "SQL Injection",
+            "tokens": 1000,
+            "failureRate": 75.5,
+        },
+        {
+            "module": "XSS Attack",
+            "tokens": 800,
+            "failureRate": 45.2,
+        },
+        {
+            "module": "CSRF Attack",
+            "tokens": 600,
+            "failureRate": 30.8,
+        },
+    ]
+
+    # Mock plot_security_report function
+
+    response = client.post("/plot.jpeg", json={"table": table_data})
+
+    # Verify response
+    assert response.status_code == 200
+    assert response.headers["content-type"] == "image/jpeg"