38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
import random
|
|
|
|
import factory
|
|
from django.contrib.auth import get_user_model
|
|
|
|
from book.factories import ModuleFactory
|
|
from core.factories import fake
|
|
from objectives.models import ObjectiveGroup, Objective
|
|
|
|
|
|
class ObjectiveGroupFactory(factory.django.DjangoModelFactory):
|
|
class Meta:
|
|
model = ObjectiveGroup
|
|
|
|
title = factory.Iterator(['Gesellschaft', 'Sprache und Kommunikation'])
|
|
|
|
module = factory.SubFactory(ModuleFactory)
|
|
user = factory.Iterator(get_user_model().objects.all())
|
|
|
|
|
|
class ObjectiveFactory(factory.django.DjangoModelFactory):
|
|
class Meta:
|
|
model = Objective
|
|
|
|
text = factory.LazyAttribute(lambda x: fake.sentence(nb_words=random.randint(4, 8)))
|
|
|
|
group = factory.SubFactory(ObjectiveGroupFactory)
|
|
|
|
|
|
# class ObjectiveProgressStatusFactory(factory.django.DjangoModelFactory):
|
|
# class Meta:
|
|
# model = ObjectiveProgressStatus
|
|
#
|
|
# objective = factory.SubFactory(ObjectiveFactory)
|
|
# user = factory.Iterator(get_user_model().objects.all())
|
|
#
|
|
# done = factory.Sequence([True, False])
|