def setUp(self): self.bucket = storage.Bucket( 'https://s3.eu-central-1.amazonaws.com/testimagestorageforpippackage/' ) resource_broker.register( 's3', 'testimagestorage', aws_access_key_id='dsadasd', aws_secret_access_key='dsasda', region_name='eu-central-1' )
def test(self, mocked_task, patched_wheezy, *args): mocked_task.return_value = mock.MagicMock() storage_ = storage.S3ImageStorage(101, 'jpg') storage_.use_bucket(self.bucket) memcache_mocked = mock.MagicMock() resource_broker.register('cache_service', memcache_mocked) resource_broker.register('webengine', 'wheezy') image = Image.open(os.path.abspath(os.path.join(os.path.dirname(__file__), 'test.jpg'))) with mock.patch.object(storage_, '_get_image_from_url') as mocked_object: memcache_mocked.get.return_value = False mocked_object.return_value = image storage_.store_origin_from_file( image, '4000x4000' ) mock_to_compare = mock.MagicMock() patched_wheezy.HTTPResponse.return_value = mock_to_compare storage_.get_requested_image('https://microsoft.com/?size=200x100') patched_wheezy.HTTPResponse.assert_called_with(content_type='image/jpeg') mock_to_compare.write_bytes.assert_called_with(image.tobytes()) storage_.get_requested_image((200, 100))