From ba39dad06c49131a4ba622d070f123d8e56d702b Mon Sep 17 00:00:00 2001 From: Joachim Date: Sat, 29 May 2021 15:32:04 +0200 Subject: [PATCH] Add save_and_cleanup test --- bookwyrm/tests/test_preview_images.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/bookwyrm/tests/test_preview_images.py b/bookwyrm/tests/test_preview_images.py index a48c60803..41d11d439 100644 --- a/bookwyrm/tests/test_preview_images.py +++ b/bookwyrm/tests/test_preview_images.py @@ -16,6 +16,7 @@ from bookwyrm.preview_images import ( generate_edition_preview_image_task, generate_user_preview_image_task, generate_preview_image, + save_and_cleanup, ) import logging @@ -70,6 +71,19 @@ class PreviewImages(TestCase): result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT) ) + def test_store_preview_image(self, *args, **kwargs): + image = Image.new("RGB", (200, 200), color="#F00") + + result = save_and_cleanup(image, instance=self.local_user) + self.assertTrue(result) + + self.local_user.refresh_from_db() + self.assertIsInstance(self.local_user.preview_image, ImageFieldFile) + self.assertIsNotNone(self.local_user.preview_image) + self.assertEqual(self.local_user.preview_image.width, 200) + self.assertEqual(self.local_user.preview_image.height, 200) + + def test_site_preview(self, *args, **kwargs): """generate site preview""" generate_site_preview_image_task()