179 lines
4.6 KiB
Python
179 lines
4.6 KiB
Python
"""Provides factories for student models."""
|
|
import random
|
|
from datetime import datetime
|
|
from uuid import uuid4
|
|
|
|
import factory
|
|
from django.contrib.auth.models import AnonymousUser, Group, Permission
|
|
from django.contrib.contenttypes.models import ContentType
|
|
from factory import lazy_attribute
|
|
from factory.django import DjangoModelFactory
|
|
from opaque_keys.edx.keys import CourseKey
|
|
from pytz import UTC
|
|
|
|
from course_modes.models import CourseMode
|
|
from student.models import (
|
|
CourseAccessRole,
|
|
CourseEnrollment,
|
|
CourseEnrollmentAllowed,
|
|
PendingEmailChange,
|
|
Registration,
|
|
User,
|
|
UserProfile,
|
|
UserStanding
|
|
)
|
|
|
|
# Factories are self documenting
|
|
# pylint: disable=missing-docstring
|
|
|
|
TEST_PASSWORD = 'test'
|
|
|
|
|
|
class GroupFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = Group
|
|
django_get_or_create = ('name', )
|
|
|
|
name = factory.Sequence(u'group{0}'.format)
|
|
|
|
|
|
class UserStandingFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = UserStanding
|
|
|
|
user = None
|
|
account_status = None
|
|
changed_by = None
|
|
|
|
|
|
class UserProfileFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = UserProfile
|
|
django_get_or_create = ('user', )
|
|
|
|
user = None
|
|
name = factory.LazyAttribute(u'{0.user.first_name} {0.user.last_name}'.format)
|
|
level_of_education = None
|
|
gender = u'm'
|
|
mailing_address = None
|
|
goals = u'Learn a lot'
|
|
allow_certificate = True
|
|
|
|
|
|
class RegistrationFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = Registration
|
|
|
|
user = None
|
|
activation_key = uuid4().hex.decode('ascii')
|
|
|
|
|
|
class UserFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = User
|
|
django_get_or_create = ('email', 'username')
|
|
|
|
_DEFAULT_PASSWORD = 'test'
|
|
|
|
username = factory.Sequence(u'robot{0}'.format)
|
|
email = factory.Sequence(u'robot+test+{0}@edx.org'.format)
|
|
password = factory.PostGenerationMethodCall('set_password', _DEFAULT_PASSWORD)
|
|
first_name = factory.Sequence(u'Robot{0}'.format)
|
|
last_name = 'Test'
|
|
is_staff = False
|
|
is_active = True
|
|
is_superuser = False
|
|
last_login = datetime(2012, 1, 1, tzinfo=UTC)
|
|
date_joined = datetime(2011, 1, 1, tzinfo=UTC)
|
|
|
|
@factory.post_generation
|
|
def profile(obj, create, extracted, **kwargs): # pylint: disable=unused-argument, no-self-argument
|
|
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
|
|
|
|
@factory.post_generation
|
|
def groups(self, create, extracted, **kwargs):
|
|
if extracted is None:
|
|
return
|
|
|
|
if isinstance(extracted, basestring):
|
|
extracted = [extracted]
|
|
|
|
for group_name in extracted:
|
|
self.groups.add(GroupFactory.simple_generate(create, name=group_name))
|
|
|
|
|
|
class AnonymousUserFactory(factory.Factory):
|
|
class Meta(object):
|
|
model = AnonymousUser
|
|
|
|
|
|
class AdminFactory(UserFactory):
|
|
is_staff = True
|
|
|
|
|
|
class SuperuserFactory(UserFactory):
|
|
is_superuser = True
|
|
|
|
|
|
class CourseEnrollmentFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = CourseEnrollment
|
|
|
|
user = factory.SubFactory(UserFactory)
|
|
course = factory.SubFactory(
|
|
'openedx.core.djangoapps.content.course_overviews.tests.factories.CourseOverviewFactory',
|
|
)
|
|
|
|
|
|
class CourseAccessRoleFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = CourseAccessRole
|
|
|
|
user = factory.SubFactory(UserFactory)
|
|
course_id = CourseKey.from_string('edX/toy/2012_Fall')
|
|
role = 'TestRole'
|
|
|
|
|
|
class CourseEnrollmentAllowedFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = CourseEnrollmentAllowed
|
|
|
|
email = 'test@edx.org'
|
|
course_id = CourseKey.from_string('edX/toy/2012_Fall')
|
|
|
|
|
|
class PendingEmailChangeFactory(DjangoModelFactory):
|
|
"""Factory for PendingEmailChange objects
|
|
|
|
user: generated by UserFactory
|
|
new_email: sequence of new+email+{}@edx.org
|
|
activation_key: sequence of integers, padded to 30 characters
|
|
"""
|
|
class Meta(object):
|
|
model = PendingEmailChange
|
|
|
|
user = factory.SubFactory(UserFactory)
|
|
new_email = factory.Sequence(u'new+email+{0}@edx.org'.format)
|
|
activation_key = factory.Sequence(u'{:0<30d}'.format)
|
|
|
|
|
|
class ContentTypeFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = ContentType
|
|
|
|
app_label = factory.Faker('app_name')
|
|
|
|
|
|
class PermissionFactory(DjangoModelFactory):
|
|
class Meta(object):
|
|
model = Permission
|
|
|
|
codename = factory.Faker('codename')
|
|
content_type = factory.SubFactory(ContentTypeFactory)
|