mirror of
https://github.com/pinry/pinry.git
synced 2025-11-13 00:25:41 +01:00
93 lines
2.8 KiB
Python
93 lines
2.8 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):
|
|
FACTORY_FOR = User
|
|
|
|
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):
|
|
FACTORY_FOR = Tag
|
|
|
|
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):
|
|
FACTORY_FOR = Pin
|
|
|
|
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):
|
|
tags = PinFactory.create().tags.all()
|
|
self.assertTrue(all([tag.name.startswith('tag_') for tag in tags]))
|
|
self.assertEqual(tags.count(), 1)
|
|
|
|
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)
|