Changeset View
Changeset View
Standalone View
Standalone View
common/tests/factories/training.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, generate_file_path | from common.tests.factories.helpers import generate_image_path, generate_file_path | ||||
| from common.tests.factories.static_assets import StorageLocationFactory | |||||
| from search import signals as search_signals | from search import signals as search_signals | ||||
| from training.models import ( | from training.models import ( | ||||
| Training, | Training, | ||||
| TrainingStatus, | TrainingStatus, | ||||
| TrainingType, | TrainingType, | ||||
| TrainingDifficulty, | TrainingDifficulty, | ||||
| Chapter, | Chapter, | ||||
| Section, | Section, | ||||
| Show All 11 Lines | class TrainingFactory(DjangoModelFactory): | ||||
| slug = factory.Faker('slug') | slug = factory.Faker('slug') | ||||
| description = factory.Faker('sentence') | description = factory.Faker('sentence') | ||||
| summary = factory.Faker('paragraph') | summary = factory.Faker('paragraph') | ||||
| status = fuzzy.FuzzyChoice(TrainingStatus.choices, getter=lambda c: c[0]) | status = fuzzy.FuzzyChoice(TrainingStatus.choices, getter=lambda c: c[0]) | ||||
| type = fuzzy.FuzzyChoice(TrainingType.choices, getter=lambda c: c[0]) | type = fuzzy.FuzzyChoice(TrainingType.choices, getter=lambda c: c[0]) | ||||
| difficulty = fuzzy.FuzzyChoice(TrainingDifficulty.choices, getter=lambda c: c[0]) | difficulty = fuzzy.FuzzyChoice(TrainingDifficulty.choices, getter=lambda c: c[0]) | ||||
| 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('..name') | |||||
| ) | |||||
| class ChapterFactory(DjangoModelFactory): | class ChapterFactory(DjangoModelFactory): | ||||
| class Meta: | class Meta: | ||||
| model = Chapter | model = Chapter | ||||
| training = factory.SubFactory(TrainingFactory) | training = factory.SubFactory(TrainingFactory) | ||||
| index = factory.Sequence(lambda n: n) | index = factory.Sequence(lambda n: n) | ||||
| Show All 15 Lines | |||||
| class AssetFactory(DjangoModelFactory): | class AssetFactory(DjangoModelFactory): | ||||
| class Meta: | class Meta: | ||||
| model = Asset | model = Asset | ||||
| section = factory.SubFactory(SectionFactory) | section = factory.SubFactory(SectionFactory) | ||||
| file = factory.LazyFunction(generate_file_path) | file = factory.LazyFunction(generate_file_path) | ||||
| size_bytes = 0 | size_bytes = 0 | ||||
| storage_location = factory.SubFactory(StorageLocationFactory) | |||||
| class VideoFactory(DjangoModelFactory): | class VideoFactory(DjangoModelFactory): | ||||
| class Meta: | class Meta: | ||||
| model = Video | model = Video | ||||
| section = factory.SubFactory(SectionFactory) | section = factory.SubFactory(SectionFactory) | ||||
| file = factory.LazyFunction(generate_file_path) | file = factory.LazyFunction(generate_file_path) | ||||
| size_bytes = 0 | size_bytes = 0 | ||||
| duration = '00:05:00' | duration = '00:05:00' | ||||
| storage_location = factory.SubFactory(StorageLocationFactory) | |||||