Commit 2159b718 by Clinton Blackburn

Factory Cleanup

- Corrected import ordering
- Using a constant for the User password

ECOM-4896
parent 35c16546
"""Provides factories for student models.""" """Provides factories for student models."""
import random import random
from student.models import (User, UserProfile, Registration,
CourseEnrollmentAllowed, CourseEnrollment,
PendingEmailChange, UserStanding,
CourseAccessRole)
from course_modes.models import CourseMode
from django.contrib.auth.models import Group, AnonymousUser
from datetime import datetime from datetime import datetime
from uuid import uuid4
import factory import factory
from django.contrib.auth.models import Group, AnonymousUser
from factory import lazy_attribute from factory import lazy_attribute
from factory.django import DjangoModelFactory from factory.django import DjangoModelFactory
from uuid import uuid4
from pytz import UTC
from opaque_keys.edx.locations import SlashSeparatedCourseKey from opaque_keys.edx.locations import SlashSeparatedCourseKey
from pytz import UTC
from course_modes.models import CourseMode
from student.models import (
User, UserProfile, Registration, CourseEnrollmentAllowed, CourseEnrollment, PendingEmailChange, UserStanding,
CourseAccessRole
)
# Factories are self documenting # Factories are self documenting
# pylint: disable=missing-docstring # pylint: disable=missing-docstring
USER_PASSWORD = 'test'
class GroupFactory(DjangoModelFactory): class GroupFactory(DjangoModelFactory):
class Meta(object): class Meta(object):
model = Group model = Group
django_get_or_create = ('name', ) django_get_or_create = ('name',)
name = factory.Sequence(u'group{0}'.format) name = factory.Sequence(u'group{0}'.format)
...@@ -39,7 +42,7 @@ class UserStandingFactory(DjangoModelFactory): ...@@ -39,7 +42,7 @@ class UserStandingFactory(DjangoModelFactory):
class UserProfileFactory(DjangoModelFactory): class UserProfileFactory(DjangoModelFactory):
class Meta(object): class Meta(object):
model = UserProfile model = UserProfile
django_get_or_create = ('user', ) django_get_or_create = ('user',)
user = None user = None
name = factory.LazyAttribute(u'{0.user.first_name} {0.user.last_name}'.format) name = factory.LazyAttribute(u'{0.user.first_name} {0.user.last_name}'.format)
...@@ -83,7 +86,7 @@ class UserFactory(DjangoModelFactory): ...@@ -83,7 +86,7 @@ class UserFactory(DjangoModelFactory):
username = factory.Sequence(u'robot{0}'.format) username = factory.Sequence(u'robot{0}'.format)
email = factory.Sequence(u'robot+test+{0}@edx.org'.format) email = factory.Sequence(u'robot+test+{0}@edx.org'.format)
password = factory.PostGenerationMethodCall('set_password', 'test') password = factory.PostGenerationMethodCall('set_password', USER_PASSWORD)
first_name = factory.Sequence(u'Robot{0}'.format) first_name = factory.Sequence(u'Robot{0}'.format)
last_name = 'Test' last_name = 'Test'
is_staff = False is_staff = False
...@@ -155,6 +158,7 @@ class PendingEmailChangeFactory(DjangoModelFactory): ...@@ -155,6 +158,7 @@ class PendingEmailChangeFactory(DjangoModelFactory):
new_email: sequence of new+email+{}@edx.org new_email: sequence of new+email+{}@edx.org
activation_key: sequence of integers, padded to 30 characters activation_key: sequence of integers, padded to 30 characters
""" """
class Meta(object): class Meta(object):
model = PendingEmailChange model = PendingEmailChange
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment