class BlogEntryFactory(factory.django.DjangoModelFactory):
"""Factory for ``BlogEntry`` model."""
date = factory.sequence(lambda x: timezone.now() + timedelta(days=x))
description = factory.sequence(lambda x: f'SEO description {x}')
show_gallery = True
subtitle = factory.Faker('sentence')
text = factory.Faker('text')
title = factory.sequence(lambda x: f'Блог {x}')
video = factory.SubFactory(VideoFileFactory)
slug = factory.sequence(lambda x: f'blog-{x}')
@factory.post_generation
def create_images(self, create, extracted, **kwargs):
"""Create images for blog."""
if not create:
# Simple build, do nothing.
return
if extracted:
# Create 6 images
BlogImageFactory.create_batch(2, blog=self)
class Meta:
model = models.BlogEntry
class RegionFactory(SQLAlchemyModelFactory):
class Meta:
model = Region
sqlalchemy_session = cm.Session
name = factory.sequence(lambda n: 'region_%s' % n)
location = factory.sequence(lambda n: 'POINT(1%s 1%s)' % (n, n))
class LegacyAccountFactory(factory.DjangoModelFactory):
class Meta:
model = Eestecer
tshirt_size = "mxxl"
passport_number = factory.sequence(lambda x: "asdad" + str(x))
#Information important for companies
field_of_study = "ee"
first_name = u"Łukasz"
middle_name = factory.sequence(lambda x: "asdad" + str(x))
last_name = u"Knüppel"
second_last_name = u"Goméz"
password = factory.PostGenerationMethodCall('set_password',
'defaultpassword')
email = factory.sequence(lambda x: str(x) + "*****@*****.**" + str(x))
gender = "m"
thumbnail = factory.django.ImageField(color="blue")
description = "description"
date_of_birth = datetime.datetime.now().date()
allergies = "allergies"
activation_link = "activation_link"
food_preferences = "vegan"
""" Food preferences, for example vegetarian or no pork. """
curriculum_vitae = factory.django.ImageField(name="lol.txt")
class ParkboxFactory(factory.DjangoModelFactory):
class Meta:
model = Parkbox
uuid = factory.sequence(lambda n: uuid4())
major = factory.sequence(lambda n: 10000 + n)
minor = factory.sequence(lambda n: 10000 + n)
class ZaakKenmerkFactory(factory.django.DjangoModelFactory):
kenmerk = factory.sequence(lambda n: 'Identificatie {0}'.format(n))
kenmerk_bron = factory.sequence(
lambda n: 'Waar het kenmerk op slaat {0}'.format(n))
class Meta:
model = ZaakKenmerk
class TenderDocumentFactory(factory.DjangoModelFactory):
class Meta:
model = TenderDocument
name = factory.sequence(lambda n: 'Doc_%s' % n)
download_url = factory.sequence(lambda n: "http://test.org/%s" % n)
tender = factory.SubFactory(TenderFactory)
class UserFactory(factory.DjangoModelFactory):
username = factory.lazy_attribute(lambda obj: "{}-{}".format(obj.first_name, obj.last_name))
email = factory.lazy_attribute(lambda obj: "{}@gmail.com".format(obj.username))
first_name = factory.sequence(lambda n: fake.first_name())
last_name = factory.sequence(lambda n: fake.last_name())
profile = factory.RelatedFactory(UserProfileFactory, 'user')
password = '******'
@factory.post_generation
def groups(self, create, extracted, **kwargs):
if extracted:
self.groups.add(*extracted)
@factory.post_generation
def permissions(self, create, extracted, **kwargs):
if extracted:
self.user_permissions.add(*extracted)
@classmethod
def _create(cls, model_class, *args, **kwargs):
user = get_user_model().objects.filter(username=kwargs.get('username')).first()
if user:
return user
else:
is_superuser = kwargs.get('is_superuser')
if is_superuser:
return get_user_model().objects.create_superuser(*args, **kwargs)
else:
return get_user_model().objects.create_user(*args, **kwargs)
class Meta:
model = get_user_model()
class UserFactory(factory.django.DjangoModelFactory):
"""
Factory for generating Django user instances.
"""
password = "******"
username = factory.sequence(lambda n: "user{n}".format(n=n))
email = factory.sequence(lambda n: "test{n}@example.com".format(n=n))
phone_number = factory.sequence(lambda n: "+25471594357{n}".format(n=n))
class Meta(object):
model = get_user_model()
@classmethod
def _create(cls, model_class, *args, **kwargs):
"""
Create a new user instance
Args:
model_class:
The type of model to create an instance of.
args:
Positional arguements to create the instance with.
kwargs:
Keyword arguements to create the instance with.
Returns:
A new user instance of the type specified by
``model_class``.
"""
manager = cls._get_manager(model_class)
user = manager.create_user(*args, **kwargs)
user.save()
return user
class PostFactory(DjangoModelFactory):
class Meta:
model = Post
title = factory.sequence('Title #{}'.format)
body = factory.sequence('Body #{}'.format)
author = factory.SubFactory(UserFactory)
class ProductFactory(factory.DjangoModelFactory):
client = factory.SubFactory('jupiter_auth.factories.UserFactory')
name = factory.sequence(lambda n: fake.sentence(nb_words=1))
description = factory.sequence(lambda n: fake.sentence(nb_words=10))
class Meta:
model = Product
class DomainFactory(factory.django.DjangoModelFactory):
class Meta:
model = Domain
domain_name = factory.sequence(lambda n: "domain_name %02d" % n)
domain_description = factory.sequence(
lambda n: "domain_description %02d" % n)
class UserFactory(SQLAlchemyModelFactory):
class Meta:
model = User
sqlalchemy_session = db.session
id = factory.sequence(lambda n: 'user{0}'.format(n + 1))
name = factory.sequence(lambda n: 'user {0}'.format(n + 1))
password = '******'
class ClusterFactory(factory.DjangoModelFactory):
FACTORY_FOR = models.Cluster
name = factory.sequence(lambda n: 'cluster {0}'.format(n))
status = factory.sequence(lambda n: 'cluster status {0}'.format(n))
notes = factory.sequence(lambda n: 'cluster notes {0}'.format(n))
virtual_environment = factory.SubFactory(VirtualEnvironmentFactory)
class DomainDTOFactory(factory.Factory):
class Meta:
model = DomainDTO
domain_id = factory.sequence(lambda n: n + 1)
domain_name = factory.Sequence(lambda n: "domain_{0}".format(n + 1))
description = factory.sequence(
lambda n: "domain_{0} description".format(n + 1))
class PostFactory(MongoEngineFactory):
external_id = sequence(lambda x: x)
title = sequence(lambda x: f'title_{x}')
url = sequence(lambda x: f'http://{x}.example.com')
created = FuzzyDateTime(start_dt=datetime(2000, 1, 1, tzinfo=UTC))
class Meta:
model = Post
class UserFactory(factory.DjangoModelFactory):
class Meta:
model = get_user_model()
username = factory.sequence(lambda x: f"user_{x}")
email = factory.sequence(lambda x: f"user{x}@email.com")
password = '******'
is_verified = True
class FileCategoryFactory(factory.django.DjangoModelFactory):
"""Factory for ``FileCategory`` model."""
name = factory.sequence(lambda x: f'Category {x}')
order = factory.sequence(lambda x: x)
class Meta:
model = models.FileCategory
class AnderZaakObjectFactory(factory.django.DjangoModelFactory):
ander_zaakobject_aanduiding = factory.sequence(
lambda n: 'Identificerende beschrijving {0}'.format(n))
ander_zaakobject_omschrijving = factory.sequence(
lambda n: 'Korte omschrijving {0}'.format(n))
class Meta:
model = AnderZaakObject
class SectionF(factory.django.DjangoModelFactory):
"""Section model factory."""
class Meta:
model = Section
name = factory.sequence(lambda n: u'Test section name %s' % n)
notes = factory.sequence(lambda n: u'Test section notes %s' % n)
project = factory.SubFactory(ProjectF)
class BuildingPostFactory(factory.django.DjangoModelFactory):
building = factory.SubFactory(BuildingFactory)
creator = factory.SubFactory(UserFactory)
title = factory.sequence(lambda s: f'title-{s}')
content = factory.sequence(lambda s: f'content-{s}')
class Meta:
model = 'building_post.BuildingPost'
class UserFactory(factory.django.DjangoModelFactory):
username = factory.sequence(lambda num: str(num).zfill(13))
first_name = 'John'
last_name = 'Doe'
gender = models.User.Gender.MALE
email = factory.sequence(lambda num: f'email_{num}@training.com')
class Meta:
model = models.User
class CountryFactory(factory.django.DjangoModelFactory):
country_code = factory.sequence(lambda s: f'K{s}')
country_name = factory.sequence(lambda s: f'대한민국-{s}')
class Meta:
model = 'core.Country'
django_get_or_create = (
'country_code',
)
class TicketFactory(factory.DjangoModelFactory):
class Meta:
model = Ticket
priority = factory.sequence(lambda i: i)
title = factory.sequence(lambda i: f'ticket_{i}')
body = factory.sequence(lambda i: f'ticket_body_{i}')
due_date = factory.sequence(lambda i: date(2000 + i, 1, 1))
category = factory.SubFactory(CategoryFactory)
class OAuth2Credentials(ModelFactory):
class Meta(object):
model = models.OAuth2Credentials
sqlalchemy_session_persistence = 'flush'
client_id = factory.sequence(lambda n: 'TEST_CLIENT_ID_' + str(n))
client_secret = factory.sequence(lambda n: 'TEST_CLIENT_SECRET_' + str(n))
authorization_server = 'TEST_AUTHORIZATION_SERVER'
class BlogEntryFactory(factory.DjangoModelFactory):
"""Factory for ``BlogEntry`` model."""
date = factory.sequence(lambda x: timezone.now() + timedelta(days=x))
description = factory.sequence(lambda x: f'SEO description {x}')
show_gallery = True
subtitle = factory.Faker('sentence')
text = factory.Faker('text')
title = factory.sequence(lambda x: f'Фильм {x}. Часть 2')
video = factory.SubFactory(VideoFileFactory)
wide_image = factory.django.ImageField(color='#555')
@factory.post_generation
def create_images(self, create, extracted, **kwargs):
"""Create images for blog."""
if not create:
# Simple build, do nothing.
return
if extracted:
# Create 6 images
BlogImageFactory.create_batch(2, blog=self)
@factory.post_generation
def create_items(self, create, extracted, **kwargs):
"""Create extra blogs for blog."""
if not create:
# Simple build, do nothing.
return
if extracted:
# Create next part
self.next_part = BlogEntryFactory(
title=self.title[:-1] + '3',
date=self.date + timedelta(hours=1),
prev_part=self,
create_images=True,
)
# Create previous part
self.prev_part = BlogEntryFactory(title=self.title[:-1] + '1',
date=self.date -
timedelta(hours=1),
next_part=self,
create_images=True)
BlogRelationFactory(blog=self, related_blog=self.next_part)
BlogRelationFactory(blog=self, related_blog=self.prev_part)
BlogRelationFactory(blog=self.prev_part, related_blog=self)
BlogRelationFactory(blog=self.prev_part,
related_blog=self.next_part)
BlogRelationFactory(blog=self.next_part, related_blog=self)
BlogRelationFactory(blog=self.next_part,
related_blog=self.prev_part)
class Meta:
model = models.BlogEntry
class MusicVideoTypeFactory(factory.django.DjangoModelFactory):
"""Factory for ``MusicVideoType`` model."""
name = factory.sequence(lambda x: f'Name {x}')
slug = factory.sequence(lambda x: f'name-{x}')
description = factory.Faker('sentence')
class Meta:
model = MusicVideoType
class BlogImageFactory(factory.django.DjangoModelFactory):
"""Factory for ``BlogImage`` model."""
description = factory.sequence(lambda x: f'Title {x}')
image = factory.django.ImageField(color='#333')
order = factory.sequence(lambda x: x)
class Meta:
model = models.BlogImage
class SpecificationF(factory.django.DjangoModelFactory):
"""Specification model factory."""
class Meta:
model = Specification
worksheet = factory.SubFactory(WorksheetF)
title = factory.sequence(lambda n: u'Test title %s' % n)
value = factory.sequence(lambda n: u'Test value %s' % n)
notes = factory.sequence(lambda n: u'Test section notes %s' % n)
class SubjectFactory(factory.django.DjangoModelFactory):
name = factory.sequence(lambda n: f"subject{n}")
semester = factory.LazyAttribute(lambda n: randint(1, 6))
general_description = "description"
module_code = factory.sequence(lambda n: f"module{n}")
field_of_study = factory.SubFactory(FieldOfStudyFactory)
class Meta:
model = Subject
class OAuth2AccessToken(ModelFactory):
class Meta(object):
model = models.OAuth2AccessToken
sqlalchemy_session_persistence = 'flush'
credentials = factory.SubFactory(OAuth2Credentials)
access_token = factory.sequence(lambda n: 'TEST_ACCESS_TOKEN_' + str(n))
refresh_token = factory.sequence(lambda n: 'TEST_REFRESH_TOKEN_' + str(n))
