17 lines
466 B
Python
17 lines
466 B
Python
import random
|
|
|
|
import factory
|
|
|
|
from core.factories import fake
|
|
from rooms.models import Room
|
|
from user.factories import UserGroupFactory
|
|
|
|
|
|
class RoomFactory(factory.django.DjangoModelFactory):
|
|
class Meta:
|
|
model = Room
|
|
|
|
title = factory.LazyAttribute(lambda x: fake.sentence(nb_words=random.randint(4, 8)))
|
|
user_group = factory.SubFactory(UserGroupFactory)
|
|
appearance = factory.LazyAttribute(lambda x: random.choice(['red', 'green', 'brown']))
|