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')