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()