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