83 lines
2.1 KiB
Python
83 lines
2.1 KiB
Python
import io
|
|
from wsgiref.validate import InputWrapper
|
|
|
|
from unittest.mock import call, MagicMock, mock_open
|
|
|
|
import falcon
|
|
from falcon import testing
|
|
import msgpack
|
|
import pytest
|
|
|
|
from app.app import create_app
|
|
from app.images import ImageStore
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_store():
|
|
return MagicMock()
|
|
|
|
|
|
@pytest.fixture()
|
|
def client(mock_store):
|
|
app = create_app(mock_store)
|
|
return testing.TestClient(app)
|
|
|
|
|
|
def test_list_images(client):
|
|
doc = {
|
|
'images': [
|
|
{
|
|
'href': '/images/1eaf6ef1-7f2d-4ecc-a8d5-6e8adba7cc0e.png',
|
|
}
|
|
]
|
|
}
|
|
|
|
response = client.simulate_get('/images')
|
|
result_doc = msgpack.unpackb(response.content, raw=False)
|
|
|
|
assert result_doc == doc
|
|
assert response.status == falcon.HTTP_OK
|
|
|
|
|
|
def test_posted_image(client, mock_store):
|
|
file_name = 'fake-image-name.xyz'
|
|
mock_store.save.return_value = file_name
|
|
image_content_type = 'image/xyz'
|
|
|
|
response = client.simulate_post(
|
|
'/images',
|
|
body=b'some-fake-bytes',
|
|
headers={'content-type': image_content_type}
|
|
)
|
|
|
|
assert response.status == falcon.HTTP_CREATED
|
|
assert response.headers['location'] == '/images/{}'.format(file_name)
|
|
saver_call = mock_store.save.call_args
|
|
|
|
assert isinstance(saver_call[0][0], InputWrapper)
|
|
assert saver_call[0][1] == image_content_type
|
|
|
|
|
|
def test_saving_image(monkeypatch):
|
|
# This still has some mocks, but they are more localized and do not
|
|
# have to be monkey-patched into standard library modules (always a
|
|
# risky business).
|
|
mock_file_open = mock_open()
|
|
|
|
fake_uuid = '123e4567-e89b-12d3-a456-426655440000'
|
|
|
|
def mock_uuidgen():
|
|
return fake_uuid
|
|
|
|
fake_image_bytes = b'fake-image-bytes'
|
|
fake_request_stream = io.BytesIO(fake_image_bytes)
|
|
storage_path = 'fake-storage-path'
|
|
store = ImageStore(
|
|
storage_path,
|
|
uuidgen=mock_uuidgen,
|
|
fopen=mock_file_open
|
|
)
|
|
|
|
assert store.save(fake_request_stream, 'image/png') == fake_uuid + '.png'
|
|
assert call().write(fake_image_bytes) in mock_file_open.mock_calls
|