diff --git a/src/controllers/web_controller.py b/src/controllers/web_controller.py index f452509..3a087c0 100644 --- a/src/controllers/web_controller.py +++ b/src/controllers/web_controller.py @@ -17,7 +17,7 @@ from ..models.scoring import calculate_completeness_score from ..utils.analytics import log_sbom_generation, get_sbom_count from ..utils.formatter import export_aibom -from ..config import TEMPLATES_DIR, OUTPUT_DIR +from ..config import AIBOM_GEN_VERSION, TEMPLATES_DIR, OUTPUT_DIR logger = logging.getLogger(__name__) @@ -56,7 +56,11 @@ async def root(request: Request): @router.get("/status") async def get_status(): - return {"status": "operational", "version": "1.0.0", "generator_version": "2.0.0"} + return { + "status": "operational", + "version": AIBOM_GEN_VERSION, + "generator_version": AIBOM_GEN_VERSION, + } @router.post("/generate", response_class=HTMLResponse) async def generate_form( diff --git a/tests/test_web_controller.py b/tests/test_web_controller.py index 93f5b04..e3c0c75 100644 --- a/tests/test_web_controller.py +++ b/tests/test_web_controller.py @@ -113,6 +113,22 @@ def get_enhancement_report(self): class WebControllerTests(unittest.TestCase): + def test_status_uses_project_version(self): + app = FastAPI() + app.include_router(web_controller.router) + + response = TestClient(app).get("/status") + + self.assertEqual(response.status_code, 200) + self.assertEqual( + response.json(), + { + "status": "operational", + "version": web_controller.AIBOM_GEN_VERSION, + "generator_version": web_controller.AIBOM_GEN_VERSION, + }, + ) + def test_generate_route_writes_utf8_json_to_patched_output_dir(self): app = FastAPI() app.include_router(web_controller.router)