Differential D9127 Diff 29958 common/tests/factories/users.py

Changeset View

Standalone View

common/tests/factories/users.py

	import factory			import factory
	from django.contrib.auth.models import User			from django.contrib.auth.models import User
	from factory.django import DjangoModelFactory			from factory.django import DjangoModelFactory

	from blender_id_oauth_client.models import OAuthUserInfo, OAuthToken			from blender_id_oauth_client.models import OAuthUserInfo, OAuthToken


	class OAuthUserInfoFactory(DjangoModelFactory):			class OAuthUserInfoFactory(DjangoModelFactory):
	class Meta:			class Meta:
	model = OAuthUserInfo			model = OAuthUserInfo

	user = factory.SubFactory('common.tests.factories.users.UserFactory')			user = factory.SubFactory('common.tests.factories.users.UserFactory')


	class OAuthUserTokenFactory(DjangoModelFactory):			class OAuthUserTokenFactory(DjangoModelFactory):
	class Meta:			class Meta:
	model = OAuthToken			model = OAuthToken

	user = factory.SubFactory('common.tests.factories.users.UserFactory')			user = factory.SubFactory('common.tests.factories.users.UserFactory')


	class UserFactory(DjangoModelFactory):			class UserFactory(DjangoModelFactory):
	class Meta:			class Meta:
	model = User			model = User

	first_name = factory.Faker('first_name')			first_name = factory.Faker('first_name')
	last_name = factory.Faker('last_name')			last_name = factory.Faker('last_name')
	username = factory.LazyAttribute(lambda o: f'{o.first_name}_{o.last_name}')			username = factory.LazyAttribute(lambda o: f'{o.first_name}_{o.last_name}')
	password = 'pass'			password = 'pass'

	oauth_tokens = factory.RelatedFactory(OAuthUserTokenFactory, factory_related_name='user')			oauth_tokens = factory.RelatedFactory(OAuthUserTokenFactory, factory_related_name='user')
	oauth_info = factory.RelatedFactory(OAuthUserInfoFactory, factory_related_name='user')			oauth_info = factory.RelatedFactory(OAuthUserInfoFactory, factory_related_name='user')