def file_type_test(self, bucket_method, test_file, content_type, resized_size): bucket = Mock() bucket_method.return_value = bucket new_key = Mock() bucket.new_key.return_value = new_key image_handler = ImageHandler(filename=test_file) truck = ImageTruck.new_from_file(test_file) session = Client().session() image = ImageTable(filename='badcafe') session.add(image) session.flush() after_upload = Mock() ResizeImage._resize_image(image, image_handler, truck, 'thumbnail', after_upload) new_key.set_metadata.assert_called_with('Content-Type', content_type) resized_contents = new_key.set_contents_from_string.call_args[0][0] image_handler = ImageHandler(blob=resized_contents) eq_(image_handler.size, resized_size) after_upload.assert_called_once_with('thumbnail')
def file_type_test(self, bucket_method, test_file_name, content_type, resized_size): bucket = Mock() bucket_method.return_value = bucket new_key = Mock() bucket.new_key.return_value = new_key test_file = os.path.join(os.path.dirname(__file__), test_file_name) image_handler = ImageHandler.open(test_file) with open(test_file, 'r') as fh: truck = ImageTruck.new_from_stream(fh, content_type) session = Client().session() image = ImageTable(filename='badcafe') session.add(image) session.flush() ResizeImage._resize_image(image, image_handler, truck, 'thumbnail') new_key.set_metadata.assert_called_with('Content-Type', content_type) resized_contents = new_key.set_contents_from_string.call_args[0][0] image_handler = ImageHandler.open(StringIO.StringIO(resized_contents)) eq_(image_handler.size, resized_size)
def test_resize_a_portrait_image(self, MockImage): session = Client().session() image_handler = Mock() image_handler.size = (427, 640) image_handler.format = 'JPEG' image = ImageTable(filename='badcafe') session.add(image) session.flush() ResizeImage._resize_image(image, image_handler, Mock(), 'medium') resizes = session.query(ImageResize).all() eq_(len(resizes), 1) eq_(resizes[0].height, 500) eq_(resizes[0].width, 333)
def test_resize_an_image(self): test_file = os.path.join(os.path.dirname(__file__), 'test_image_640x427.jpg') image_handler = ImageHandler.open(test_file) truck = Mock() session = Client().session() image = ImageTable(filename='badcafe') session.add(image) session.flush() ResizeImage._resize_image(image, image_handler, truck, 'thumbnail') resizes = session.query(ImageResize).all() eq_(len(resizes), 1) eq_(resizes[0].width, 100) eq_(resizes[0].height, 66) eq_(resizes[0].suffix, 'thumbnail') (args, kwargs) = truck.upload_resize.call_args eq_(args[1], 'thumbnail')
def test_resize_an_image(self): image_handler = ImageHandler(filename=SMALL_JPG) truck = Mock() session = Client().session() image = ImageTable(filename='badcafe') session.add(image) session.flush() after_upload = Mock() ResizeImage._resize_image(image, image_handler, truck, 'thumbnail', after_upload) resizes = session.query(ImageResize).all() eq_(len(resizes), 1) eq_(resizes[0].width, 100) eq_(resizes[0].height, 66) eq_(resizes[0].suffix, 'thumbnail') (args, kwargs) = truck.upload_resize.call_args eq_(args[1], 'thumbnail') after_upload.assert_called_once_with('thumbnail')