Changeset View
Changeset View
Standalone View
Standalone View
common/tests/factories/films.py
| import factory | import factory | ||||
| from factory import fuzzy | from factory import fuzzy | ||||
| from factory.django import DjangoModelFactory | from factory.django import DjangoModelFactory | ||||
| from common.tests.factories.helpers import generate_image_path | from common.tests.factories.helpers import generate_image_path | ||||
| from common.tests.factories.static_assets import StaticAssetFactory, StorageLocationFactory | from common.tests.factories.static_assets import StaticAssetFactory | ||||
| from common.tests.factories.users import UserFactory | from common.tests.factories.users import UserFactory | ||||
| from films.models import ( | from films.models import ( | ||||
| Film, | Film, | ||||
| Collection, | Collection, | ||||
| Asset, | Asset, | ||||
| ProductionLog, | ProductionLog, | ||||
| ProductionLogEntry, | ProductionLogEntry, | ||||
| ProductionLogEntryAsset, | ProductionLogEntryAsset, | ||||
| Show All 17 Lines | class FilmFactory(DjangoModelFactory): | ||||
| release_date = factory.Faker('date') | release_date = factory.Faker('date') | ||||
| is_published = True | is_published = True | ||||
| logo = factory.LazyFunction(generate_image_path) | logo = factory.LazyFunction(generate_image_path) | ||||
| poster = factory.LazyFunction(generate_image_path) | poster = factory.LazyFunction(generate_image_path) | ||||
| picture_header = factory.LazyFunction(generate_image_path) | picture_header = factory.LazyFunction(generate_image_path) | ||||
| thumbnail = factory.LazyFunction(generate_image_path) | thumbnail = factory.LazyFunction(generate_image_path) | ||||
| storage_location = factory.SubFactory( | |||||
| StorageLocationFactory, name=factory.SelfAttribute('..title') | |||||
| ) | |||||
| @factory.django.mute_signals(search_signals.post_save) | @factory.django.mute_signals(search_signals.post_save) | ||||
| class CollectionFactory(DjangoModelFactory): | class CollectionFactory(DjangoModelFactory): | ||||
| class Meta: | class Meta: | ||||
| model = Collection | model = Collection | ||||
| film = factory.SubFactory(FilmFactory) | film = factory.SubFactory(FilmFactory) | ||||
| name = factory.Faker('text', max_nb_chars=30) | name = factory.Faker('text', max_nb_chars=30) | ||||
| slug = factory.Faker('slug') | slug = factory.Faker('slug') | ||||
| text = factory.Faker('paragraph') | text = factory.Faker('paragraph') | ||||
| storage_location = factory.SelfAttribute('film.storage_location') | |||||
| thumbnail = factory.LazyFunction(generate_image_path) | thumbnail = factory.LazyFunction(generate_image_path) | ||||
| @factory.django.mute_signals(search_signals.post_save) | @factory.django.mute_signals(search_signals.post_save) | ||||
| class AssetFactory(DjangoModelFactory): | class AssetFactory(DjangoModelFactory): | ||||
| class Meta: | class Meta: | ||||
| model = Asset | model = Asset | ||||
| film = factory.SubFactory(FilmFactory) | film = factory.SubFactory(FilmFactory) | ||||
| collection = factory.SubFactory(CollectionFactory, film=factory.SelfAttribute('..film')) | collection = factory.SubFactory(CollectionFactory, film=factory.SelfAttribute('..film')) | ||||
| static_asset = factory.SubFactory( | static_asset = factory.SubFactory(StaticAssetFactory) | ||||
| StaticAssetFactory, storage_location=factory.SelfAttribute('..film.storage_location') | |||||
| ) | |||||
| name = factory.Faker('text', max_nb_chars=30) | name = factory.Faker('text', max_nb_chars=30) | ||||
| slug = factory.Faker('slug') | slug = factory.Faker('slug') | ||||
| description = factory.Faker('paragraph') | description = factory.Faker('paragraph') | ||||
| category = fuzzy.FuzzyChoice(AssetCategory.choices, getter=lambda c: c[0]) | category = fuzzy.FuzzyChoice(AssetCategory.choices, getter=lambda c: c[0]) | ||||
| is_published = True | is_published = True | ||||
| @factory.django.mute_signals(search_signals.post_save) | @factory.django.mute_signals(search_signals.post_save) | ||||
| class ProductionLogFactory(DjangoModelFactory): | class ProductionLogFactory(DjangoModelFactory): | ||||
| class Meta: | class Meta: | ||||
| model = ProductionLog | model = ProductionLog | ||||
| film = factory.SubFactory(FilmFactory) | film = factory.SubFactory(FilmFactory) | ||||
| summary = factory.Faker('text') | summary = factory.Faker('text') | ||||
| user = factory.SubFactory(UserFactory) | user = factory.SubFactory(UserFactory) | ||||
| storage_location = factory.SelfAttribute('film.storage_location') | |||||
| thumbnail = factory.LazyFunction(generate_image_path) | thumbnail = factory.LazyFunction(generate_image_path) | ||||
| class ProductionLogEntryFactory(DjangoModelFactory): | class ProductionLogEntryFactory(DjangoModelFactory): | ||||
| class Meta: | class Meta: | ||||
| model = ProductionLogEntry | model = ProductionLogEntry | ||||
| production_log = factory.SubFactory(ProductionLogFactory) | production_log = factory.SubFactory(ProductionLogFactory) | ||||
| Show All 19 Lines | |||||