From 2159b718704bc65ea75172f6e7ba67985354895c Mon Sep 17 00:00:00 2001 From: Clinton Blackburn Date: Thu, 25 Aug 2016 13:41:16 -0400 Subject: [PATCH] Factory Cleanup - Corrected import ordering - Using a constant for the User password ECOM-4896 --- common/djangoapps/student/tests/factories.py | 28 +++++++++++--------- 1 file changed, 16 insertions(+), 12 deletions(-) diff --git a/common/djangoapps/student/tests/factories.py b/common/djangoapps/student/tests/factories.py index 6f246dc413..129b22fc2c 100644 --- a/common/djangoapps/student/tests/factories.py +++ b/common/djangoapps/student/tests/factories.py @@ -1,28 +1,31 @@ """Provides factories for student models.""" 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 uuid import uuid4 + import factory +from django.contrib.auth.models import Group, AnonymousUser from factory import lazy_attribute from factory.django import DjangoModelFactory -from uuid import uuid4 -from pytz import UTC 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 # pylint: disable=missing-docstring +USER_PASSWORD = 'test' + class GroupFactory(DjangoModelFactory): class Meta(object): model = Group - django_get_or_create = ('name', ) + django_get_or_create = ('name',) name = factory.Sequence(u'group{0}'.format) @@ -39,7 +42,7 @@ class UserStandingFactory(DjangoModelFactory): class UserProfileFactory(DjangoModelFactory): class Meta(object): model = UserProfile - django_get_or_create = ('user', ) + django_get_or_create = ('user',) user = None name = factory.LazyAttribute(u'{0.user.first_name} {0.user.last_name}'.format) @@ -83,7 +86,7 @@ class UserFactory(DjangoModelFactory): username = factory.Sequence(u'robot{0}'.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) last_name = 'Test' is_staff = False @@ -155,6 +158,7 @@ class PendingEmailChangeFactory(DjangoModelFactory): new_email: sequence of new+email+{}@edx.org activation_key: sequence of integers, padded to 30 characters """ + class Meta(object): model = PendingEmailChange