Adding additional parameter for GUI display

Issue-Id: AIMLFW-50

Change-Id: Iea74f56663a49b72399f17543f145bf889b18f41
Signed-off-by: rajdeep11 <rajdeep.sin@samsung.com>

diff --git a/tests/test_tm_apis.py b/tests/test_tm_apis.py
index 88a8893..b7d565e 100644 (file)
--- a/tests/test_tm_apis.py
+++ b/tests/test_tm_apis.py
@@ -1018,10 +1018,10 @@ class Test_get_feature_group_by_name:
         self.client = trainingmgr_main.APP.test_client(self)
         self.logger = trainingmgr_main.LOGGER
 
-    result=[('testing', '', 'InfluxSource', True, '21.0.0.21', '12345', '', '', '', '')]
+    result=[('testing', '', 'InfluxSource', True, '21.0.0.21', '12345', '', '', '', '','')]
     @patch('trainingmgr.trainingmgr_main.get_feature_group_by_name_db', return_value=result)
     def test_get_feature_group_by_name(self, mock1):
-        expected_data=b'{"featuregroup": [{"featuregroup_name": "testing", "features": [""], "datalake": "InfluxSource", "dme": true, "dme_host": "21.0.0.21", "dme_port": "12345", "bucket": "", "token": "", "source_name": "", "db_org": ""}]}'
+        expected_data=b'{"featuregroup": [{"featuregroup_name": "testing", "features": [""], "datalake": "InfluxSource", "dme": true, "dme_host": "21.0.0.21", "measured_obj_class": "12345", "dme_port": "", "bucket": "", "token": "", "source_name": "", "db_org": ""}]}'
         fg_name='testing'
         response=self.client.get('/featureGroup/{}'.format(fg_name))
         assert response.status_code == 200 , "status code is not equal"

diff --git a/trainingmgr/trainingmgr_main.py b/trainingmgr/trainingmgr_main.py
index 749819f..3c9d9b8 100644 (file)
--- a/trainingmgr/trainingmgr_main.py
+++ b/trainingmgr/trainingmgr_main.py
@@ -1460,11 +1460,12 @@ def get_feature_group_by_name(featuregroup_name):
                     "datalake": res[2],
                     "dme": res[3],
                     "dme_host": res[4],
-                    "dme_port": res[5],
-                    "bucket":res[6],
-                    "token":res[7],
-                    "source_name":res[8],
-                    "db_org":res[9]
+                    "measured_obj_class":res[5],
+                    "dme_port": res[6],
+                    "bucket":res[7],
+                    "token":res[8],
+                    "source_name":res[9],
+                    "db_org":res[10]
                 }
                 feature_group.append(dict_data)
             api_response={"featuregroup":feature_group}