def setUpClass(cls):
super().setUpClass()
cls.user = User()
cls.user2 = User()
cls.url = '/api/users/'
cls.user_data = {
'display_name': faker.name(),
'first_name': faker.name(),
'last_name': faker.name(),
'email': faker.email(),
'password': faker.name(),
'address': faker.address(),
'latitude': faker.latitude(),
'longitude': faker.longitude()
}
class MockUser(Mock):
class Meta:
model = get_user_model()
strategy = CREATE_STRATEGY
is_active = True
is_staff = False
display_name = LazyAttribute(lambda _: faker.name())
first_name = LazyAttribute(lambda _: faker.name())
last_name = LazyAttribute(lambda _: faker.name())
email = LazyAttribute(lambda _: faker.email())
password = PostGeneration(
lambda obj, *args, **kwargs: obj.set_password(obj.display_name))
wall = SubFactory(MockWall)
class Store(DjangoModelFactory):
class Meta:
model = StoreModel
group = SubFactory(Group)
name = LazyAttribute(lambda x: faker.sentence(nb_words=4))
description = LazyAttribute(lambda x: faker.name())
class User(DjangoModelFactory):
class Meta:
model = get_user_model()
strategy = CREATE_STRATEGY
is_active = True
is_staff = False
display_name = LazyAttribute(lambda _: faker.name())
first_name = LazyAttribute(lambda _: faker.name())
last_name = LazyAttribute(lambda _: faker.name())
email = LazyAttribute(lambda _: faker.email())
# Use display_name as password, as it is readable
password = PostGeneration(lambda obj, *args, **kwargs: obj.set_password(obj.display_name))
wall = SubFactory(Wall)
def setUpClass(cls):
super().setUpClass()
cls.user = User()
cls.user2 = User()
cls.url = '/api/users/'
cls.user_data = {
'display_name': faker.name(),
'first_name': faker.name(),
'last_name': faker.name(),
'email': faker.email(),
'password': faker.name(),
'address': faker.address(),
'latitude': faker.latitude(),
'longitude': faker.longitude()
}
cls.group = Group(members=[cls.user, cls.user2])
cls.another_common_group = Group(members=[cls.user, cls.user2])
cls.user_in_another_group = User()
cls.another_group = Group(members=[cls.user_in_another_group, ])
def setUpClass(cls):
super().setUpClass()
cls.user = User()
cls.group = GroupFactory()
cls.url = '/api/groups/'
cls.group_data = {'name': faker.name(),
'description': faker.text(),
'address': faker.address(),
'latitude': faker.latitude(),
'longitude': faker.longitude()}
def setUpClass(cls):
super().setUpClass()
cls.user = UserFactory()
cls.user2 = UserFactory()
cls.verified_user = VerifiedUserFactory()
cls.url = '/api/users/'
cls.user_data = {
'email': faker.email(),
'password': faker.name(),
'display_name': faker.name(),
'address': faker.address(),
'latitude': faker.latitude(),
'longitude': faker.longitude()
}
cls.group = Group(members=[cls.user, cls.user2])
cls.another_common_group = Group(members=[cls.user, cls.user2])
cls.user_in_another_group = UserFactory()
cls.another_group = Group(members=[
cls.user_in_another_group,
])
def setUpClass(cls):
super().setUpClass()
cls.user = User()
cls.group = GroupFactory()
cls.url = '/api/groups/'
cls.group_data = {
'name': faker.name(),
'description': faker.text(),
'address': faker.address(),
'latitude': faker.latitude(),
'longitude': faker.longitude()
}
class MockUser(Mock):
class Meta:
model = "yunity.User"
strategy = CREATE_STRATEGY
is_active = True
is_staff = False
type = Category.objects.filter(name='user.default').first()
display_name = LazyAttribute(lambda _: faker.name())
email = LazyAttribute(lambda _: faker.email())
password = PostGeneration(
lambda obj, *args, **kwargs: obj.set_password(obj.display_name))
locations = LazyAttribute(lambda _: [faker.location() for _ in range(2)])
class Group(DjangoModelFactory):
class Meta:
model = GroupModel
@post_generation
def members(self, created, members, **kwargs):
if not created:
return
if members:
for member in members:
self.members.add(member)
name = LazyAttribute(lambda x: faker.sentence(nb_words=4))
description = LazyAttribute(lambda x: faker.name())
def setUpClass(cls):
super().setUpClass()
cls.url = '/api/stores/'
# group with two members and one store
cls.member = UserFactory()
cls.member2 = UserFactory()
cls.group = Group(members=[cls.member, cls.member2])
cls.store = Store(group=cls.group)
cls.store_url = cls.url + str(cls.store.id) + '/'
# not a member
cls.user = UserFactory()
# another store for above group
cls.store_data = {'name': faker.name(),
'description': faker.name(),
'group': cls.group.id,
'address': faker.address(),
'latitude': faker.latitude(),
'longitude': faker.longitude()}
# another group
cls.different_group = Group(members=[cls.member2, ])
def setUpClass(cls):
super().setUpClass()
cls.url = '/api/stores/'
# group with two members and one store
cls.member = User()
cls.member2 = User()
cls.group = Group(members=[cls.member, cls.member2])
cls.store = Store(group=cls.group)
cls.store_url = cls.url + str(cls.store.id) + '/'
# not a member
cls.user = User()
# another store for above group
cls.store_data = {'name': faker.name(),
'description': faker.name(),
'group': cls.group.id,
'address': faker.address(),
'latitude': faker.latitude(),
'longitude': faker.longitude()}
# another group
cls.different_group = Group(members=[cls.member2, ])
def setUpClass(cls):
super().setUpClass()
cls.user = UserFactory()
cls.member = UserFactory()
cls.group = GroupFactory(members=[
cls.member,
])
cls.group_with_password = GroupFactory(password='******')
cls.join_password_url = '/api/groups/{}/join/'.format(
cls.group_with_password.id)
cls.url = '/api/groups/'
cls.group_data = {
'name': faker.name(),
'description': faker.text(),
'address': faker.address(),
'latitude': faker.latitude(),
'longitude': faker.longitude()
}
def setUpClass(cls):
super().setUpClass()
cls.url = '/api/conversations/'
# A chat with 2 participants
cls.participant = User()
cls.participant2 = User()
cls.conversation = Conversation(participants=[cls.participant, cls.participant2])
cls.conversation_url = cls.url + str(cls.conversation.id) + '/'
# not a participant
cls.user = User()
# another chat
cls.conversation_data = {'topic': faker.name(),
'with_participants': [cls.participant2.id, ],
'message': faker.text()}
cls.patch_data = {'topic': 'new topic'}
cls.put_data = {'topic': 'new topic'}
class UserFactory(DjangoModelFactory):
class Meta:
model = get_user_model()
strategy = CREATE_STRATEGY
is_active = True
is_staff = False
display_name = LazyAttribute(lambda _: faker.name())
email = LazyAttribute(lambda _: faker.email())
description = LazyAttribute(lambda _: faker.text())
# Use display_name as password, as it is readable
password = PostGeneration(lambda obj, *args, **kwargs: obj.set_password(obj.display_name))
@classmethod
def _create(cls, model_class, *args, **kwargs):
user = super()._create(model_class, *args, **kwargs)
user._unverify_mail()
return user
def setUpClass(cls):
super().setUpClass()
cls.url = '/api/conversations/'
# A chat with 2 participants
cls.participant = User()
cls.participant2 = User()
cls.conversation = Conversation(
participants=[cls.participant, cls.participant2])
cls.conversation_url = cls.url + str(cls.conversation.id) + '/'
# not a participant
cls.user = User()
# another chat
cls.conversation_data = {
'topic': faker.name(),
'with_participants': [
cls.participant2.id,
],
'message': faker.text()
}
cls.patch_data = {'topic': 'new topic'}
cls.put_data = {'topic': 'new topic'}
