0f7378a171
This allows specification of profile parameters when creating a user. Because the profile contents are always accessed from the database, the user must be saved to the database before the profile is created. This means that the profile cannot be created if the user is merely being built (and not saved) rather than created.
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
from student.models import (User, UserProfile, Registration,
|
|
CourseEnrollmentAllowed, CourseEnrollment)
|
|
from django.contrib.auth.models import Group
|
|
from datetime import datetime
|
|
from factory import DjangoModelFactory, Factory, SubFactory, PostGenerationMethodCall, post_generation
|
|
from uuid import uuid4
|
|
|
|
|
|
class GroupFactory(DjangoModelFactory):
|
|
FACTORY_FOR = Group
|
|
|
|
name = 'staff_MITx/999/Robot_Super_Course'
|
|
|
|
|
|
class UserProfileFactory(DjangoModelFactory):
|
|
FACTORY_FOR = UserProfile
|
|
|
|
user = None
|
|
name = 'Robot Test'
|
|
level_of_education = None
|
|
gender = 'm'
|
|
mailing_address = None
|
|
goals = 'World domination'
|
|
|
|
|
|
class RegistrationFactory(DjangoModelFactory):
|
|
FACTORY_FOR = Registration
|
|
|
|
user = None
|
|
activation_key = uuid4().hex
|
|
|
|
|
|
class UserFactory(DjangoModelFactory):
|
|
FACTORY_FOR = User
|
|
|
|
username = 'robot'
|
|
email = 'robot+test@edx.org'
|
|
password = PostGenerationMethodCall('set_password',
|
|
'test')
|
|
first_name = 'Robot'
|
|
last_name = 'Test'
|
|
is_staff = False
|
|
is_active = True
|
|
is_superuser = False
|
|
last_login = datetime(2012, 1, 1)
|
|
date_joined = datetime(2011, 1, 1)
|
|
|
|
@post_generation
|
|
def profile(obj, create, extracted, **kwargs):
|
|
if create:
|
|
obj.save()
|
|
return UserProfileFactory.create(user=obj, **kwargs)
|
|
elif kwargs:
|
|
raise Exception("Cannot build a user profile without saving the user")
|
|
else:
|
|
return None
|
|
|
|
|
|
class AdminFactory(UserFactory):
|
|
is_staff = True
|
|
|
|
|
|
class CourseEnrollmentFactory(DjangoModelFactory):
|
|
FACTORY_FOR = CourseEnrollment
|
|
|
|
user = SubFactory(UserFactory)
|
|
course_id = 'edX/toy/2012_Fall'
|
|
|
|
|
|
class CourseEnrollmentAllowedFactory(DjangoModelFactory):
|
|
FACTORY_FOR = CourseEnrollmentAllowed
|
|
|
|
email = 'test@edx.org'
|
|
course_id = 'edX/test/2012_Fall'
|