mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-03 09:51:11 -05:00
notify-api-412 use black to enforce python style standards
This commit is contained in:
@@ -8,68 +8,99 @@ from app.clients.document_download import (
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture(scope='function')
|
||||
@pytest.fixture(scope="function")
|
||||
def document_download(client, mocker):
|
||||
client = DocumentDownloadClient()
|
||||
current_app = mocker.Mock(config={
|
||||
'DOCUMENT_DOWNLOAD_API_HOST': 'https://document-download',
|
||||
'DOCUMENT_DOWNLOAD_API_KEY': 'test-key'
|
||||
})
|
||||
current_app = mocker.Mock(
|
||||
config={
|
||||
"DOCUMENT_DOWNLOAD_API_HOST": "https://document-download",
|
||||
"DOCUMENT_DOWNLOAD_API_KEY": "test-key",
|
||||
}
|
||||
)
|
||||
client.init_app(current_app)
|
||||
return client
|
||||
|
||||
|
||||
def test_get_upload_url(document_download):
|
||||
assert document_download.get_upload_url('service-id') == 'https://document-download/services/service-id/documents'
|
||||
assert (
|
||||
document_download.get_upload_url("service-id")
|
||||
== "https://document-download/services/service-id/documents"
|
||||
)
|
||||
|
||||
|
||||
def test_upload_document(document_download):
|
||||
with requests_mock.Mocker() as request_mock:
|
||||
request_mock.post('https://document-download/services/service-id/documents', json={
|
||||
'document': {'url': 'https://document-download/services/service-id/documents/uploaded-url'}
|
||||
}, request_headers={
|
||||
'Authorization': 'Bearer test-key',
|
||||
}, status_code=201)
|
||||
request_mock.post(
|
||||
"https://document-download/services/service-id/documents",
|
||||
json={
|
||||
"document": {
|
||||
"url": "https://document-download/services/service-id/documents/uploaded-url"
|
||||
}
|
||||
},
|
||||
request_headers={
|
||||
"Authorization": "Bearer test-key",
|
||||
},
|
||||
status_code=201,
|
||||
)
|
||||
|
||||
resp = document_download.upload_document('service-id', 'abababab')
|
||||
resp = document_download.upload_document("service-id", "abababab")
|
||||
|
||||
assert resp == 'https://document-download/services/service-id/documents/uploaded-url'
|
||||
assert (
|
||||
resp == "https://document-download/services/service-id/documents/uploaded-url"
|
||||
)
|
||||
|
||||
|
||||
def test_should_raise_400s_as_DocumentDownloadErrors(document_download):
|
||||
with pytest.raises(DocumentDownloadError) as excinfo, requests_mock.Mocker() as request_mock:
|
||||
request_mock.post('https://document-download/services/service-id/documents', json={
|
||||
'error': 'Invalid mime type'
|
||||
}, status_code=400)
|
||||
with pytest.raises(
|
||||
DocumentDownloadError
|
||||
) as excinfo, requests_mock.Mocker() as request_mock:
|
||||
request_mock.post(
|
||||
"https://document-download/services/service-id/documents",
|
||||
json={"error": "Invalid mime type"},
|
||||
status_code=400,
|
||||
)
|
||||
|
||||
document_download.upload_document('service-id', 'abababab')
|
||||
document_download.upload_document("service-id", "abababab")
|
||||
|
||||
assert excinfo.value.message == 'Invalid mime type'
|
||||
assert excinfo.value.message == "Invalid mime type"
|
||||
assert excinfo.value.status_code == 400
|
||||
|
||||
|
||||
def test_should_raise_non_400_statuses_as_exceptions(document_download):
|
||||
with pytest.raises(expected_exception=Exception) as excinfo, requests_mock.Mocker() as request_mock:
|
||||
with pytest.raises(
|
||||
expected_exception=Exception
|
||||
) as excinfo, requests_mock.Mocker() as request_mock:
|
||||
request_mock.post(
|
||||
'https://document-download/services/service-id/documents',
|
||||
json={'error': 'Auth Error Of Some Kind'},
|
||||
status_code=403
|
||||
"https://document-download/services/service-id/documents",
|
||||
json={"error": "Auth Error Of Some Kind"},
|
||||
status_code=403,
|
||||
)
|
||||
|
||||
document_download.upload_document('service-id', 'abababab')
|
||||
document_download.upload_document("service-id", "abababab")
|
||||
|
||||
assert type(excinfo.value) == Exception # make sure it's a base exception, so will be handled as a 500 by v2 api
|
||||
assert str(excinfo.value) == 'Unhandled document download error: {"error": "Auth Error Of Some Kind"}'
|
||||
assert (
|
||||
type(excinfo.value) == Exception
|
||||
) # make sure it's a base exception, so will be handled as a 500 by v2 api
|
||||
assert (
|
||||
str(excinfo.value)
|
||||
== 'Unhandled document download error: {"error": "Auth Error Of Some Kind"}'
|
||||
)
|
||||
|
||||
|
||||
def test_should_raise_exceptions_without_http_response_bodies_as_exceptions(document_download):
|
||||
with pytest.raises(expected_exception=Exception) as excinfo, requests_mock.Mocker() as request_mock:
|
||||
def test_should_raise_exceptions_without_http_response_bodies_as_exceptions(
|
||||
document_download,
|
||||
):
|
||||
with pytest.raises(
|
||||
expected_exception=Exception
|
||||
) as excinfo, requests_mock.Mocker() as request_mock:
|
||||
request_mock.post(
|
||||
'https://document-download/services/service-id/documents',
|
||||
exc=requests.exceptions.ConnectTimeout
|
||||
"https://document-download/services/service-id/documents",
|
||||
exc=requests.exceptions.ConnectTimeout,
|
||||
)
|
||||
|
||||
document_download.upload_document('service-id', 'abababab')
|
||||
document_download.upload_document("service-id", "abababab")
|
||||
|
||||
assert type(excinfo.value) == Exception # make sure it's a base exception, so will be handled as a 500 by v2 api
|
||||
assert str(excinfo.value) == 'Unhandled document download error: ConnectTimeout()'
|
||||
assert (
|
||||
type(excinfo.value) == Exception
|
||||
) # make sure it's a base exception, so will be handled as a 500 by v2 api
|
||||
assert str(excinfo.value) == "Unhandled document download error: ConnectTimeout()"
|
||||
|
||||
Reference in New Issue
Block a user