def test_serve_from_string(self, mimetype):
content = b'image_content'
self.mock_decode.return_value = content
resp = ImageServer.serve_from_string('image_string', mimetype)
self.mock_decode.assert_called_with('image_string')
self.assertEqual(resp.content_type, mimetype)
self.mock_response.assert_called_with(content)
def get_pending_message(self, username):
message = self.accessor.get_last_message(username)
if not message:
return message, 404
message = Message.load_from_document(message)
path = message.current
resp = ImageServer().serve_thumbnail_from_path(path, 300)
return resp
class Images(BaseResource):
def __init__(self, api=None):
super().__init__(api=api)
self.image_server = ImageServer()
@base_ns.produces(['image/*'])
def get(self, file_id):
return self.image_server.serve_from_mongodb(file_id)
def test_serve_from_s3(self, bucket, exp_bucket):
mock_img = self.mock_s3.get.return_value
content = b'image_content'
mock_img.read.return_value = content
mock_img.mimetype = 'image/type'
resp = ImageServer.serve_from_s3('file_id', bucket)
self.mock_s3.get.assert_called_with('file_id', bucket_name=exp_bucket)
self.assertEqual(resp.content_type, 'image/type')
self.mock_response.assert_called_with(content)
def test_serve_from_mongodb(self):
mock_img = self.mock_grid_fs.get.return_value
content = b'image_content'
mock_img.read.return_value = content
mock_img.mimetype = 'image/type'
resp = ImageServer().serve_from_mongodb('file_id')
self.mock_grid_fs.get.assert_called_with('file_id')
self.assertEqual(resp.content_type, 'image/type')
self.mock_response.assert_called_with(content)
def __init__(self, api=None):
super().__init__(api=api)
self.image_server = ImageServer()