mirror of
https://github.com/pinry/pinry.git
synced 2025-11-13 08:35:41 +01:00
7db42bb075
Too lax unique constraints for Thumbnail coped with "on demand" thumbnail generation may leave database in inconsistent state where two thumbnail for the same size are saved. We should be able to prevent that from happening by generating all thumbnails when we save the image. Should fix #24, but I can't figure out a way to actually test it.
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
from django.conf import settings
|
|
from django.contrib.auth.models import Permission
|
|
from django.core.files.images import ImageFile
|
|
from django.db.models.query import QuerySet
|
|
from django.test import TestCase
|
|
from django_images.models import Thumbnail
|
|
|
|
import factory
|
|
from taggit.models import Tag
|
|
|
|
from ..models import Pin, Image
|
|
from ...users.models import User
|
|
|
|
|
|
TEST_IMAGE_PATH = 'logo.png'
|
|
|
|
|
|
class UserFactory(factory.Factory):
|
|
username = factory.Sequence(lambda n: 'user_{}'.format(n))
|
|
email = factory.Sequence(lambda n: 'user_{}@example.com'.format(n))
|
|
|
|
@factory.post_generation(extract_prefix='password')
|
|
def set_password(self, create, extracted, **kwargs):
|
|
self.set_password(extracted)
|
|
self.save()
|
|
|
|
@factory.post_generation(extract_prefix='user_permissions')
|
|
def set_user_permissions(self, create, extracted, **kwargs):
|
|
self.user_permissions = Permission.objects.filter(codename__in=['add_pin', 'add_image'])
|
|
|
|
|
|
class TagFactory(factory.Factory):
|
|
name = factory.Sequence(lambda n: 'tag_{}'.format(n))
|
|
|
|
|
|
class ImageFactory(factory.Factory):
|
|
FACTORY_FOR = Image
|
|
|
|
image = factory.LazyAttribute(lambda a: ImageFile(open(TEST_IMAGE_PATH, 'rb')))
|
|
|
|
@factory.post_generation()
|
|
def create_thumbnails(self, create, extracted, **kwargs):
|
|
for size in settings.IMAGE_SIZES.keys():
|
|
Thumbnail.objects.get_or_create_at_size(self.pk, size)
|
|
|
|
|
|
class PinFactory(factory.Factory):
|
|
submitter = factory.SubFactory(UserFactory)
|
|
image = factory.SubFactory(ImageFactory)
|
|
|
|
@factory.post_generation(extract_prefix='tags')
|
|
def add_tags(self, create, extracted, **kwargs):
|
|
if isinstance(extracted, Tag):
|
|
self.tags.add(extracted)
|
|
elif isinstance(extracted, list):
|
|
self.tags.add(*extracted)
|
|
elif isinstance(extracted, QuerySet):
|
|
self.tags = extracted
|
|
else:
|
|
self.tags.add(TagFactory())
|
|
|
|
|
|
class PinFactoryTest(TestCase):
|
|
def test_default_tags(self):
|
|
self.assertTrue(PinFactory().tags.get(pk=1).name.startswith('tag_'))
|
|
|
|
def test_custom_tag(self):
|
|
custom = 'custom_tag'
|
|
self.assertEqual(PinFactory(tags=Tag.objects.create(name=custom)).tags.get(pk=1).name, custom)
|
|
|
|
def test_custom_tags_list(self):
|
|
tags = TagFactory.create_batch(2)
|
|
PinFactory(tags=tags)
|
|
self.assertEqual(Tag.objects.count(), 2)
|
|
|
|
def test_custom_tags_queryset(self):
|
|
TagFactory.create_batch(2)
|
|
tags = Tag.objects.all()
|
|
PinFactory(tags=tags)
|
|
self.assertEqual(Tag.objects.count(), 2)
|
|
|
|
def test_empty_tags(self):
|
|
PinFactory(tags=[])
|
|
self.assertEqual(Tag.objects.count(), 0)
|