task status ui tests

This commit is contained in:
bohanjason 2020-03-30 01:07:39 -04:00 committed by Dana Van Aken
parent 516d02006a
commit 814561573b
2 changed files with 6 additions and 1 deletions

View File

@ -808,22 +808,27 @@ def integration_tests():
LOG.info("\n\nIntegration Tests: PASSED!!\n") LOG.info("\n\nIntegration Tests: PASSED!!\n")
# Test task status UI
task_status_ui_test()
@task @task
def task_status_ui_test(): def task_status_ui_test():
# Test GPR # Test GPR
upload_code = 'ottertuneTestTuningGPR' upload_code = 'ottertuneTestTuningGPR'
response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code) response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code)
LOG.info(get_content(response))
assert 'Success:' in get_content(response) assert 'Success:' in get_content(response)
# Test DNN: # Test DNN:
upload_code = 'ottertuneTestTuningDNN' upload_code = 'ottertuneTestTuningDNN'
response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code) response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code)
LOG.info(get_content(response))
assert 'Success:' in get_content(response) assert 'Success:' in get_content(response)
# Test DDPG: # Test DDPG:
upload_code = 'ottertuneTestTuningDDPG' upload_code = 'ottertuneTestTuningDDPG'
response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code) response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code)
LOG.info(get_content(response))
assert 'Success:' in get_content(response) assert 'Success:' in get_content(response)
LOG.info("\n\nTask Status UI Tests: PASSED!!\n") LOG.info("\n\nTask Status UI Tests: PASSED!!\n")

View File

@ -1681,7 +1681,7 @@ def tuner_status_test(request, upload_code): # pylint: disable=unused-argument,
context = _tuner_statue_helper(session.project.id, session.id, result.id) context = _tuner_statue_helper(session.project.id, session.id, result.id)
overall_status = context['overall_status'] overall_status = context['overall_status']
num_completed, num_total = context['num_completed'].replace(' ', '').split('/') num_completed, num_total = context['num_completed'].replace(' ', '').split('/')
task_info = context['task_info'] task_info = context['tasks']
num_tasks = len(task_info) num_tasks = len(task_info)
if overall_status.lower() != 'success': if overall_status.lower() != 'success':
return HttpResponse("Failure: overall status {} should be success".format( return HttpResponse("Failure: overall status {} should be success".format(