def test_no_api_key(): with requests_mock.Mocker() as mock: mock.get(URL + "&docketID=" + DOCKET_ID, json='The test yields a bad api key', status_code=403) with pytest.raises(reggov_api_doc_error.IncorrectApiKeyException): get_docket('', DOCKET_ID)
def test_maxed_api_key(): with requests_mock.Mocker() as mock: mock.get(URL + API_KEY + "&docketID=" + DOCKET_ID, json='The test yields a overused api key', status_code=429) with pytest.raises(reggov_api_doc_error.ExceedCallLimitException): get_docket(API_KEY, DOCKET_ID)
def test_no_docket_id(): with requests_mock.Mocker() as mock: mock.get(URL + API_KEY + "&docketID=", json='The test yields a bad id', status_code=404) with pytest.raises(reggov_api_doc_error.BadDocIDException): get_docket(API_KEY, '')
def test_bad_docket_id_pattern(): with requests_mock.Mocker() as mock: bad_docket = 'b4d' + DOCKET_ID + 'b4d' mock.get(URL + API_KEY + "&docketID=" + bad_docket, json='The test yields a bad id pattern', status_code=400) with pytest.raises(reggov_api_doc_error.IncorrectIDPatternException): get_docket(API_KEY, bad_docket)
def test_bad_docket_id(): with requests_mock.Mocker() as mock: bad_docket = DOCKET_ID + '-0101' mock.get(URL + API_KEY + "&docketID=" + bad_docket, json='The test yields a bad id', status_code=404) with pytest.raises(reggov_api_doc_error.BadDocIDException): get_docket(API_KEY, bad_docket)
def test_get_docket(): with requests_mock.Mocker() as mock: mock.get(URL + API_KEY + "&docketID=" + DOCKET_ID, json={'test': 'The test is successful'}) response = get_docket(API_KEY, DOCKET_ID) assert (response == {'test': 'The test is successful'})
def find_docket_data(manager, job, job_id): print("Getting docket from regulations.gov...\n") data = get_docket(manager.api_key, job['docket_id']) LOGGER.info("Job#%s: Packaging docket..", str(job_id)) results = package_docket(data, manager.client_id, job_id) return results