예제 #1
0
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)
예제 #2
0
 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)