def remove_downloads(job_manager):
requested_job = job_manager.request_job(User(100))
job = job_to_json(requested_job)
while json.loads(job)['job_type'] == 'download':
job_manager.report_failure(User(100))
requested_job = job_manager.request_job(User(100))
job = job_to_json(requested_job)
return job
def test_none_job_to_json():
job_example = NoneJob("2aae0fca-bf7d-4444-ab9b-0120414aa0b5")
job_json = job_translator.job_to_json(job_example)
job_json_expected = \
{"job_id": "2aae0fca-bf7d-4444-ab9b-0120414aa0b5",
"job_type": "none"}
job_json_expected = json.dumps(job_json_expected)
assert job_json == job_json_expected
def test_docket_job_to_json():
job_example = DocketJob("172dc679-c2d7-4a84-b766-8cd5ee138057",
"CMS-2005-0001")
job_json = job_translator.job_to_json(job_example)
job_json_expected = {"job_id": "172dc679-c2d7-4a84-b766-8cd5ee138057",
"docket_id": "CMS-2005-0001",
"job_type": "docket"}
job_json_expected = json.dumps(job_json_expected)
assert job_json == job_json_expected
def test_document_job_to_json():
job_example = DocumentJob("bfcadfea-db21-44aa-a234-9833291cbbc9",
"CMS-2005-0001-0001")
job_json = job_translator.job_to_json(job_example)
job_json_expected = {"job_id": "bfcadfea-db21-44aa-a234-9833291cbbc9",
"document_id": "CMS-2005-0001-0001",
"job_type": "document"}
job_json_expected = json.dumps(job_json_expected)
assert job_json == job_json_expected
def test_documents_job_to_json():
job_example = DocumentsJob("d022c3f5-6d69-4cbe-9330-b2a5cc8e6ff5",
"1000",
"2005-08-04T00:00:00-04:00",
"2005-10-03T23:59:59-04:00")
job_json = job_translator.job_to_json(job_example)
job_json_expected = {"job_id": "d022c3f5-6d69-4cbe-9330-b2a5cc8e6ff5",
"page_offset": "1000",
"start_date": "2005-08-04T00:00:00-04:00",
"end_date": "2005-10-03T23:59:59-04:00",
"job_type": "documents"}
job_json_expected = json.dumps(job_json_expected)
assert job_json == job_json_expected
def test_download_job_to_json():
url = "https://api.data.gov/regulations/v3/download \
?documentId=CMS-2005-0001-0001&contentType=pdf"
job_example = \
DownloadJob("2aae0fca-bf7d-4444-ab9b-0120414aa0b5",
"CMS/CMS-2005/",
"title_of_file",
"pdf",
url)
job_json = job_translator.job_to_json(job_example)
job_json_expected = \
{"job_id": "2aae0fca-bf7d-4444-ab9b-0120414aa0b5",
"folder_name": "CMS/CMS-2005/",
"file_name": "title_of_file",
"file_type": "pdf",
"url": url,
"job_type": "download"}
job_json_expected = json.dumps(job_json_expected)
assert job_json == job_json_expected