changes based on kubeflow upgradation to get run info

Issue-id: AIMLFW-102

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

diff --git a/kfadapter/kfadapter_main.py b/kfadapter/kfadapter_main.py
index cdfb387..48a04cb 100644 (file)
--- a/kfadapter/kfadapter_main.py
+++ b/kfadapter/kfadapter_main.py
@@ -502,9 +502,9 @@ def kf_run(run_id):
                    {'ext': 1})
 
         run_info = KFCONNECT_KF_OBJ.get_kf_run(run_id)
-        run_dict['run_id'] = run_info.id
-        run_dict['run_name'] = run_info.name
-        run_dict['run_status'] = run_info.status
+        run_dict['run_id'] = run_info.run_id
+        run_dict['run_name'] = run_info.display_name
+        run_dict['run_status'] = run_info.state
         LOGGER.debug(run_dict)
     except Exception as err:
         LOGGER.error("Exception from KubeFlow in run")

diff --git a/test/test_kfadapter_main.py b/test/test_kfadapter_main.py
index 76931e2..fe5e75e 100644 (file)
--- a/test/test_kfadapter_main.py
+++ b/test/test_kfadapter_main.py
@@ -158,21 +158,21 @@ class testKfadapterApi(TestCase):
     def test_get_run(self, mock_get_kf_run):
         # given
         run = ApiRun()
-        run.id = "run-id"
-        run.name = "run-name"
-        run.status = "Running"
+        run.run_id = "run-id"
+        run.display_name = "run-name"
+        run.state = "Running"
         mock_get_kf_run.return_value = run
         
         # when
-        response = self.client.get("/runs/{}".format(run.id))
+        response = self.client.get("/runs/{}".format(run.run_id))
 
         # then
         mock_get_kf_run.assert_called_once()
         self.assertEqual(response.content_type, "application/json")
         self.assertEqual(response.status_code, status.HTTP_200_OK)
-        self.assertEqual(response.get_json()["run_id"], run.id)
-        self.assertEqual(response.get_json()["run_name"], run.name)
-        self.assertEqual(response.get_json()["run_status"], run.status)
+        self.assertEqual(response.get_json()["run_id"], run.run_id)
+        self.assertEqual(response.get_json()["run_name"], run.display_name)
+        self.assertEqual(response.get_json()["run_status"], run.state)
 
 
     def test_delete_run(self):