From 634ed500440c219e0736f3839830a66711f674f7 Mon Sep 17 00:00:00 2001 From: Mouse Reeve Date: Thu, 7 Jul 2022 10:24:19 -0700 Subject: [PATCH] Tests for image generator file This isn't what I need to be working on right now but it bugs me immensely that the test coverage is at 89% and not 90% and this was the lowest hanging fruit to get more coverage. Truly a me problem. --- bookwyrm/tests/models/test_book_model.py | 30 +++++++++++++++++++++++- 1 file changed, 29 insertions(+), 1 deletion(-) diff --git a/bookwyrm/tests/models/test_book_model.py b/bookwyrm/tests/models/test_book_model.py index d4f59fa58..a5ddc2504 100644 --- a/bookwyrm/tests/models/test_book_model.py +++ b/bookwyrm/tests/models/test_book_model.py @@ -1,6 +1,10 @@ """ testing models """ -from dateutil.parser import parse +from io import BytesIO +import pathlib +from dateutil.parser import parse +from PIL import Image +from django.core.files.base import ContentFile from django.test import TestCase from django.utils import timezone @@ -96,3 +100,27 @@ class Book(TestCase): self.first_edition.description = "hi" self.first_edition.save() self.assertEqual(self.first_edition.edition_rank, 1) + + def test_thumbnail_fields(self): + """Just hit them""" + settings.ENABLE_THUMBNAIL_GENERATION = True + image_file = pathlib.Path(__file__).parent.joinpath( + "../../static/images/default_avi.jpg" + ) + image = Image.open(image_file) + output = BytesIO() + image.save(output, format=image.format) + book = models.Edition.objects.create(title="hello") + book.cover.save("test.jpg", ContentFile(output.getvalue())) + self.assertIsNotNone(book.cover_bw_book_xsmall_webp.url) + self.assertIsNotNone(book.cover_bw_book_xsmall_jpg.url) + self.assertIsNotNone(book.cover_bw_book_small_webp.url) + self.assertIsNotNone(book.cover_bw_book_small_jpg.url) + self.assertIsNotNone(book.cover_bw_book_medium_webp.url) + self.assertIsNotNone(book.cover_bw_book_medium_jpg.url) + self.assertIsNotNone(book.cover_bw_book_large_webp.url) + self.assertIsNotNone(book.cover_bw_book_large_jpg.url) + self.assertIsNotNone(book.cover_bw_book_xlarge_webp.url) + self.assertIsNotNone(book.cover_bw_book_xlarge_jpg.url) + self.assertIsNotNone(book.cover_bw_book_xxlarge_webp.url) + self.assertIsNotNone(book.cover_bw_book_xxlarge_jpg.url)