def reprocess_image(request_format, image_id):
session = Client().session()
image = session.query(Image).filter(Image.image_id == image_id).one()
truck = ImageTruck.new_from_image(image)
contents = ImageContents(image_id=image.image_id, contents=truck.contents, content_type=truck.content_type)
session.add(contents)
session.flush()
delay(g.queued_tasks, process_image, contents.image_contents_id)
if request_format == "json":
return {"status": "ok"}
else:
return redirect(url_for("show_image", image_id=image.image_id))
def test_new_from_image(self, subprocess, MockClient):
subprocess.check_output.return_value = \
'space-centurion.png: PNG image data, 1280 x 800, ' \
'8-bit/color RGB, non-interlaced'
def get_contents_to_filename(filename):
with open(filename, 'w') as fh:
fh.write('brain, skull, etc')
key = Mock()
key.get_contents_to_filename.side_effect = get_contents_to_filename
bucket = Mock()
bucket.get_key.return_value = key
client = Mock()
client.bucket.return_value = bucket
MockClient.return_value = client
image = Image(filename='faceface')
truck = ImageTruck.new_from_image(image)
eq_(truck.contents, 'brain, skull, etc')
bucket.get_key.assert_called_with('faceface')
