120 lines
4 KiB
Python
120 lines
4 KiB
Python
import datetime
|
|
import os
|
|
|
|
from django.conf import settings
|
|
from django.utils.text import slugify
|
|
|
|
try:
|
|
import factory
|
|
except ImportError:
|
|
raise ImportError(
|
|
"No module named factory. To run photologue's tests you need to install factory-boy.")
|
|
|
|
from ..models import Gallery, ImageModel, Photo, PhotoEffect, PhotoSize
|
|
|
|
RES_DIR = os.path.join(os.path.dirname(__file__), '../res')
|
|
LANDSCAPE_IMAGE_PATH = os.path.join(RES_DIR, 'test_photologue_landscape.jpg')
|
|
PORTRAIT_IMAGE_PATH = os.path.join(RES_DIR, 'test_photologue_portrait.jpg')
|
|
SQUARE_IMAGE_PATH = os.path.join(RES_DIR, 'test_photologue_square.jpg')
|
|
QUOTING_IMAGE_PATH = os.path.join(RES_DIR, 'test_photologue_"ing.jpg')
|
|
UNICODE_IMAGE_PATH = os.path.join(RES_DIR, 'test_unicode_®.jpg')
|
|
NONSENSE_IMAGE_PATH = os.path.join(RES_DIR, 'test_nonsense.jpg')
|
|
SAMPLE_ZIP_PATH = os.path.join(RES_DIR, 'zips/sample.zip')
|
|
SAMPLE_NOT_IMAGE_ZIP_PATH = os.path.join(RES_DIR, 'zips/not_image.zip')
|
|
IGNORED_FILES_ZIP_PATH = os.path.join(RES_DIR, 'zips/ignored_files.zip')
|
|
|
|
|
|
class GalleryFactory(factory.django.DjangoModelFactory):
|
|
|
|
class Meta:
|
|
model = Gallery
|
|
|
|
title = factory.Sequence(lambda n: f'gallery{n:0>3}')
|
|
slug = factory.LazyAttribute(lambda a: slugify(a.title))
|
|
|
|
@factory.sequence
|
|
def date_added(n):
|
|
# Have to cater projects being non-timezone aware.
|
|
if settings.USE_TZ:
|
|
sample_date = datetime.datetime(
|
|
year=2011, month=12, day=23, hour=17, minute=40, tzinfo=datetime.timezone.utc)
|
|
else:
|
|
sample_date = datetime.datetime(year=2011, month=12, day=23, hour=17, minute=40)
|
|
return sample_date + datetime.timedelta(minutes=n)
|
|
|
|
@factory.post_generation
|
|
def sites(self, create, extracted, **kwargs):
|
|
"""
|
|
Associates the object with the current site unless ``sites`` was passed,
|
|
in which case the each item in ``sites`` is associated with the object.
|
|
|
|
Note that if PHOTOLOGUE_MULTISITE is False, all Gallery/Photos are automatically
|
|
associated with the current site - bear this in mind when writing tests.
|
|
"""
|
|
if not create:
|
|
return
|
|
if extracted:
|
|
for site in extracted:
|
|
self.sites.add(site)
|
|
|
|
|
|
class ImageModelFactory(factory.django.DjangoModelFactory):
|
|
|
|
class Meta:
|
|
model = ImageModel
|
|
abstract = True
|
|
|
|
|
|
class PhotoFactory(ImageModelFactory):
|
|
|
|
"""Note: after creating Photo instances for tests, remember to manually
|
|
delete them.
|
|
"""
|
|
|
|
class Meta:
|
|
model = Photo
|
|
|
|
title = factory.Sequence(lambda n: f'photo{n:0>3}')
|
|
slug = factory.LazyAttribute(lambda a: slugify(a.title))
|
|
image = factory.django.ImageField(from_path=LANDSCAPE_IMAGE_PATH)
|
|
|
|
@factory.sequence
|
|
def date_added(n):
|
|
# Have to cater projects being non-timezone aware.
|
|
if settings.USE_TZ:
|
|
sample_date = datetime.datetime(
|
|
year=2011, month=12, day=23, hour=17, minute=40, tzinfo=datetime.timezone.utc)
|
|
else:
|
|
sample_date = datetime.datetime(year=2011, month=12, day=23, hour=17, minute=40)
|
|
return sample_date + datetime.timedelta(minutes=n)
|
|
|
|
@factory.post_generation
|
|
def sites(self, create, extracted, **kwargs):
|
|
"""
|
|
Associates the object with the current site unless ``sites`` was passed,
|
|
in which case the each item in ``sites`` is associated with the object.
|
|
|
|
Note that if PHOTOLOGUE_MULTISITE is False, all Gallery/Photos are automatically
|
|
associated with the current site - bear this in mind when writing tests.
|
|
"""
|
|
if not create:
|
|
return
|
|
if extracted:
|
|
for site in extracted:
|
|
self.sites.add(site)
|
|
|
|
|
|
class PhotoSizeFactory(factory.django.DjangoModelFactory):
|
|
|
|
class Meta:
|
|
model = PhotoSize
|
|
|
|
name = factory.Sequence(lambda n: f'name{n:0>3}')
|
|
|
|
|
|
class PhotoEffectFactory(factory.django.DjangoModelFactory):
|
|
|
|
class Meta:
|
|
model = PhotoEffect
|
|
|
|
name = factory.Sequence(lambda n: f'effect{n:0>3}')
|