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 (file)
--- 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