Update test_preview_images.py

This commit is contained in:
Joachim 2021-05-28 21:26:13 +02:00
parent ec159ebcb0
commit 2bf254bce7

View file

@ -20,6 +20,7 @@ from bookwyrm.preview_images import (
import logging import logging
@patch("bookwyrm.emailing.send_email.delay") @patch("bookwyrm.emailing.send_email.delay")
class PreviewImages(TestCase): class PreviewImages(TestCase):
"""every response to a get request, html or json""" """every response to a get request, html or json"""
@ -38,8 +39,10 @@ class PreviewImages(TestCase):
local=True, local=True,
localname="possum", localname="possum",
avatar=SimpleUploadedFile( avatar=SimpleUploadedFile(
avatar_file, open(avatar_file, "rb").read(), content_type="image/jpeg" avatar_file,
) open(avatar_file, "rb").read(),
content_type="image/jpeg",
),
) )
with patch("bookwyrm.preview_images.generate_edition_preview_image_task.delay"): with patch("bookwyrm.preview_images.generate_edition_preview_image_task.delay"):
self.work = models.Work.objects.create(title="Test Work") self.work = models.Work.objects.create(title="Test Work")
@ -63,7 +66,9 @@ class PreviewImages(TestCase):
result = generate_preview_image(texts=texts, picture=image_file, rating=5) result = generate_preview_image(texts=texts, picture=image_file, rating=5)
self.assertIsInstance(result, Image.Image) self.assertIsInstance(result, Image.Image)
self.assertEqual(result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT)) self.assertEqual(
result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT)
)
def test_site_preview(self, *args, **kwargs): def test_site_preview(self, *args, **kwargs):
"""generate site preview""" """generate site preview"""
@ -74,7 +79,9 @@ class PreviewImages(TestCase):
self.assertIsInstance(site_preview_image, ImageFieldFile) self.assertIsInstance(site_preview_image, ImageFieldFile)
result = Image.open(site_preview_image) result = Image.open(site_preview_image)
self.assertEqual(result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT)) self.assertEqual(
result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT)
)
def test_edition_preview(self, *args, **kwargs): def test_edition_preview(self, *args, **kwargs):
"""generate user preview""" """generate user preview"""
@ -85,7 +92,9 @@ class PreviewImages(TestCase):
self.assertIsInstance(book_preview_image, ImageFieldFile) self.assertIsInstance(book_preview_image, ImageFieldFile)
result = Image.open(book_preview_image) result = Image.open(book_preview_image)
self.assertEqual(result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT)) self.assertEqual(
result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT)
)
def test_user_preview(self, *args, **kwargs): def test_user_preview(self, *args, **kwargs):
"""generate user preview""" """generate user preview"""
@ -96,4 +105,6 @@ class PreviewImages(TestCase):
self.assertIsInstance(user_preview_image, ImageFieldFile) self.assertIsInstance(user_preview_image, ImageFieldFile)
result = Image.open(user_preview_image) result = Image.open(user_preview_image)
self.assertEqual(result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT)) self.assertEqual(
result.size, (settings.PREVIEW_IMG_WIDTH, settings.PREVIEW_IMG_HEIGHT)
)