def test_get_dataset_import_login_required(): with requests_mock.mock() as m: m.get('https://databasin.org/api/v1/dataset_imports/a1b2c3/', status_code=401) c = Client() with pytest.raises(LoginRequiredError): c.get_import('a1b2c3')
def test_get_dataset_import_forbidden(): with requests_mock.mock() as m: m.get('https://databasin.org/api/v1/dataset_imports/a1b2c3/', status_code=401) c = Client() c.username = '******' with pytest.raises(ForbiddenError): c.get_import('a1b2c3')
def test_dataset_import_cancel_login_required(dataset_import_data): with requests_mock.mock() as m: m.get('https://databasin.org/api/v1/dataset_imports/a1b2c3/', text=json.dumps(dataset_import_data)) m.delete('https://databasin.org/api/v1/dataset_imports/a1b2c3/', status_code=401) c = Client() dataset_import = c.get_import('a1b2c3') with pytest.raises(LoginRequiredError): dataset_import.cancel()
def test_get_dataset_import(dataset_import_data): with requests_mock.mock() as m: m.get('https://databasin.org/api/v1/dataset_imports/a1b2c3/', text=json.dumps(dataset_import_data)) c = Client() dataset_import = c.get_import('a1b2c3') assert m.called assert dataset_import.id == 'a1b2c3'
def test_dataset_import_cancel_forbidden(dataset_import_data): with requests_mock.mock() as m: m.get('https://databasin.org/api/v1/dataset_imports/a1b2c3/', text=json.dumps(dataset_import_data)) m.delete('https://databasin.org/api/v1/dataset_imports/a1b2c3/', status_code=401) c = Client() c.username = '******' dataset_import = c.get_import('a1b2c3') with pytest.raises(ForbiddenError): dataset_import.cancel()
def test_dataset_import_cancel(dataset_import_data): with requests_mock.mock() as m: m.get('https://databasin.org/api/v1/dataset_imports/a1b2c3/', text=json.dumps(dataset_import_data)) m.delete('https://databasin.org/api/v1/dataset_imports/a1b2c3/', status_code=204) c = Client() dataset_import = c.get_import('a1b2c3') dataset_import.cancel() assert m.call_count == 2