from integration_tests.test_files.validate_results import delete_result_files_from_s3
from integration_tests.test_files.validate_results import post_livy_jobs
from integration_tests.test_files.validate_results import wait_for_batch_to_succeed

import sys

if sys.argv[1] == 'delete_result_files_from_s3':
    delete_result_files_from_s3()
elif sys.argv[1] == 'post_livy_jobs':
    post_livy_jobs(sys.argv[2], sys.argv[3], sys.argv[4])
elif sys.argv[1] == 'wait_for_batch_to_succeed':
    wait_for_batch_to_succeed()