From: Swaraj Kumar <swaraj.kumar@samsung.com>
Date: Mon, 21 Apr 2025 08:05:19 +0000 (+0530)
Subject: Test cases for FG controller
X-Git-Tag: 4.0.0~5
X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=commitdiff_plain;h=refs%2Fchanges%2F29%2F14329%2F2;p=aiml-fw%2Fawmf%2Ftm.git

Test cases for FG controller

Change-Id: I4c71b3fe43c9cfcc373bd7c6b005224cdaac4a18
Signed-off-by: Swaraj Kumar <swaraj.kumar@samsung.com>
---

diff --git a/tests/test_featuregroupcontroller.py b/tests/test_featuregroupcontroller.py
index 25455ad..0f2ff6b 100644
--- a/tests/test_featuregroupcontroller.py
+++ b/tests/test_featuregroupcontroller.py
@@ -66,4 +66,13 @@ class TestCreateFeatureGroup:
         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
+        
+class TestGetFeatureGroup:
+
+    @patch('trainingmgr.service.featuregroup_service.get_all_featuregroups', side_effect=Exception("DB fail"))
+    def test_get_failure(self, mock_get, client):
+        response = client.get("/featureGroup")
+        assert response.status_code == 500
+        expected = ProblemDetails(500, "Internal Server Error", "Failed to get featuregroups").to_dict()
         assert response.get_json() == expected
\ No newline at end of file