| """ |
| Unit and Integration tests for FloorGen FastAPI Production Server. |
| """ |
|
|
| import pytest |
| from fastapi.testclient import TestClient |
| from floorgen.api.server import app |
|
|
| client = TestClient(app) |
|
|
|
|
| def test_health_endpoint(): |
| response = client.get("/health") |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["status"] == "healthy" |
| assert data["model_loaded"] is True |
| assert "version" in data |
|
|
|
|
| def test_ready_endpoint(): |
| response = client.get("/ready") |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["ready"] is True |
| assert data["indexed_plans"] > 0 |
| assert data["checkpoint_epoch"] >= 80 |
|
|
|
|
| def test_rooms_endpoint(): |
| response = client.get("/api/v1/rooms") |
| assert response.status_code == 200 |
| data = response.json() |
| assert "living_room" in data["rooms"] |
| assert "master_bedroom" in data["rooms"] |
|
|
|
|
| def test_generate_endpoint_validation(): |
| |
| response = client.post("/api/v1/generate", json={ |
| "rooms": ["space_station_docking_bay"] |
| }) |
| assert response.status_code == 422 |
|
|
|
|
| def test_generate_endpoint_success(): |
| response = client.post("/api/v1/generate", json={ |
| "rooms": ["living_room", "master_bedroom", "bathroom", "kitchen"], |
| "steps": 10, |
| "batch": 1, |
| "solver": True |
| }) |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["realism_score"] > 80.0 |
| assert data["circulation"] == 1.0 |
| assert "<svg" in data["svg"] |
| assert "rooms" in data["json_spec"] |
| assert len(data["json_spec"]["rooms"]) == 4 |
|
|
|
|
| def test_metrics_endpoint(): |
| response = client.get("/metrics") |
| assert response.status_code == 200 |
| assert "floorgen_http_requests_total" in response.text |
| assert "floorgen_generation_requests_total" in response.text |
|
|
|
|
| def test_generate_with_ifc_and_compliance(): |
| response = client.post("/api/v1/generate", json={ |
| "rooms": ["living_room", "master_bedroom", "bathroom", "kitchen"], |
| "steps": 10, |
| "batch": 1, |
| "solver": True, |
| "export_ifc": True |
| }) |
| assert response.status_code == 200 |
| data = response.json() |
| assert data["ifc_base64"] is not None |
| assert data["compliance"] is not None |
| assert "compliance_score" in data["compliance"] |
| assert "walls" in data["json_spec"] |
| assert "furniture" in data["json_spec"] |
|
|
|
|
| def test_job_store_persistence(tmp_path): |
| from floorgen.api.job_store import PersistentJobStore |
|
|
| db_path = str(tmp_path / "test_jobs.db") |
| store = PersistentJobStore(db_path=db_path) |
| store.save_job("test_job_1", {"status": "running", "total_requests": 2}) |
| |
| |
| job = store.get_job("test_job_1") |
| assert job is not None |
| assert job["status"] == "running" |
| assert job["total_requests"] == 2 |
|
|
| |
| store2 = PersistentJobStore(db_path=db_path) |
| job2 = store2.get_job("test_job_1") |
| assert job2 is not None |
| assert job2["status"] == "running" |
| assert job2["total_requests"] == 2 |
|
|
|
|
|
|
