From 865bb0235fcde0b98e905aa409aa4fef42a4c0d3 Mon Sep 17 00:00:00 2001 From: Swaraj Kumar Date: Mon, 21 Apr 2025 13:31:09 +0530 Subject: [PATCH] Test cases for FG controller Implement Error Handling in Responses in all the components of AIMLFW Issue Id: AIMLFW-181 Change-Id: I7dee5471b1b7fb84f29c079595a8eb9117f675aa Signed-off-by: Swaraj Kumar --- tests/test_featuregroupcontroller.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_featuregroupcontroller.py b/tests/test_featuregroupcontroller.py index 68804af..25455ad 100644 --- a/tests/test_featuregroupcontroller.py +++ b/tests/test_featuregroupcontroller.py @@ -59,4 +59,11 @@ class TestCreateFeatureGroup: response = client.post("/featureGroup", json={}) assert response.status_code == 400 expected = ProblemDetails(400, "Bad Request", "some DB error").to_dict() + assert response.get_json() == expected + + @patch('trainingmgr.schemas.featuregroup_schema.FeatureGroupSchema.load', side_effect=Exception("unexpected fail")) + def test_generic_exception(self, mock_load, client): + response = client.post("/featureGroup", json={}) + assert response.status_code == 500 + expected = ProblemDetails(500, "Internal Server Error", "unexpected fail").to_dict() assert response.get_json() == expected \ No newline at end of file -- 2.16.6