modify api test

pyisong · Jun 11, 2020 · 99f2b15 · 99f2b15
1 parent 019fa3f
commit 99f2b15
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/fate_flow/tests/api_tests/job_operation_test.py b/fate_flow/tests/api_tests/job_operation_test.py
@@ -30,7 +30,7 @@ def test_job_operation(self):
         job_id = response.json()['jobId']
 
         # query
-        response = requests.post("/".join([self.server_url, 'job', 'query']), json={'job_id': job_id})
+        response = requests.post("/".join([self.server_url, 'job', 'query']), json={'job_id': job_id, 'role': 'guest'})
         self.assertTrue(int(response.json()['retcode']) == 0)
         job_info = response.json()['data'][0]
 
@@ -42,8 +42,8 @@ def test_job_operation(self):
         response = requests.post("/".join([self.server_url, 'job', 'config']), json={'job_id': job_id, 'role': job_info['f_role'], 'party_id': job_info['f_party_id']})
         self.assertTrue(int(response.json()['retcode']) == 0)
 
-        print('waiting 3s ...')
-        time.sleep(3)
+        print('waiting 10s ...')
+        time.sleep(10)
 
         # stop
         response = requests.post("/".join([self.server_url, 'job', 'stop']), json={'job_id': job_id})

diff --git a/fate_flow/tests/api_tests/tracking_test.py b/fate_flow/tests/api_tests/tracking_test.py
@@ -36,7 +36,7 @@ def test_tracking(self):
         job_id = response.json()['jobId']
         job_info = {'f_status': 'running'}
         for i in range(60):
-            response = requests.post("/".join([self.server_url, 'job', 'query']), json={'job_id': job_id})
+            response = requests.post("/".join([self.server_url, 'job', 'query']), json={'job_id': job_id, 'role': 'guest'})
             self.assertTrue(response.status_code in [200, 201])
             job_info = response.json()['data'][0]
             if job_info['f_status'] in ['success', 'failed', 'canceled']: