From 0af7a330ccf20f5626c0ebae7f1a6e5287ea5e2e Mon Sep 17 00:00:00 2001 From: rajdeep11 Date: Tue, 8 Aug 2023 15:48:42 +0530 Subject: [PATCH] Adding additional parameter for GUI display Issue-Id: AIMLFW-50 Change-Id: Iea74f56663a49b72399f17543f145bf889b18f41 Signed-off-by: rajdeep11 --- tests/test_tm_apis.py | 4 ++-- trainingmgr/trainingmgr_main.py | 11 ++++++----- 2 files changed, 8 insertions(+), 7 deletions(-) diff --git a/tests/test_tm_apis.py b/tests/test_tm_apis.py index 88a8893..b7d565e 100644 --- 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 --- 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} -- 2.16.6