Modified base_test restart_workflow_api to accept and pass on the new delete_files argument when restarting a workflow

This commit is contained in:
mike cullerton 2021-04-15 11:28:35 -04:00
parent 91e80d01bc
commit 69396f9888
1 changed files with 3 additions and 2 deletions

View File

@ -367,11 +367,12 @@ class BaseTest(unittest.TestCase):
self.assertEqual(workflow.workflow_spec_id, workflow_api.workflow_spec_id)
return workflow_api
def restart_workflow_api(self, workflow, clear_data=False, user_uid="dhf8r"):
def restart_workflow_api(self, workflow, clear_data=False, delete_files=False, user_uid="dhf8r"):
user = session.query(UserModel).filter_by(uid=user_uid).first()
self.assertIsNotNone(user)
url = (f'/v1.0/workflow/{workflow.id}/restart'
f'?clear_data={str(clear_data)}')
f'?clear_data={str(clear_data)}'
f'&delete_files={str(delete_files)}')
workflow_api = self.get_workflow_common(url, user)
self.assertEqual(workflow.workflow_spec_id, workflow_api.workflow_spec_id)
return workflow_api