def test_maxed_api_key():
with requests_mock.Mocker() as mock:
mock.get(URL + API_KEY + "&po=" + OFFSET + '&crd=' + START_DATE + '-' +
END_DATE,
json='The test yields a overused api key',
status_code=429)
with pytest.raises(reggov_api_doc_error.ExceedCallLimitException):
get_documents(API_KEY, OFFSET, START_DATE, END_DATE)
def test_no_api_key():
with requests_mock.Mocker() as mock:
mock.get(URL + '' + "&po=" + OFFSET + '&crd=' + START_DATE + '-' +
END_DATE,
json='The test yields a bad api key',
status_code=403)
with pytest.raises(reggov_api_doc_error.IncorrectApiKeyException):
get_documents('', OFFSET, START_DATE, END_DATE)
def test_bad_url():
with requests_mock.Mocker() as mock:
mock.get(URL + API_KEY + "&po=" + OFFSET + '&crd=' + START_DATE + '-' +
END_DATE,
json='The test yields a bad id',
status_code=404)
with pytest.raises(reggov_api_doc_error.BadDocIDException):
get_documents(API_KEY, OFFSET, START_DATE, END_DATE)
def find_documents_data(manager, job, job_id):
print("Getting documents from regulations.gov...\n")
data = get_documents(manager.api_key, job["page_offset"],
job["start_date"], job["end_date"])
LOGGER.info("Job#%s: Packaging documents...", str(job_id))
results = package_documents(data, manager.client_id, job_id)
return results
def test_get_documents():
with requests_mock.Mocker() as mock:
mock.get(URL + API_KEY + "&po=" + OFFSET + '&crd=' + START_DATE + '-' +
END_DATE,
json={'test': 'The test is successful'})
response = get_documents(API_KEY, OFFSET, START_DATE, END_DATE)
assert response == {'test': 'The test is successful'}