Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion common/djangoapps/course_modes/tests/factories.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from course_modes.models import CourseMode
from factory import DjangoModelFactory
from factory.django import DjangoModelFactory


# Factories don't have __init__ methods, and are self documenting
# pylint: disable=W0232
Expand Down
39 changes: 15 additions & 24 deletions common/djangoapps/student/tests/factories.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,15 +6,8 @@
from course_modes.models import CourseMode
from django.contrib.auth.models import Group, AnonymousUser
from datetime import datetime
from factory import (
DjangoModelFactory,
Factory,
LazyAttribute,
post_generation,
PostGenerationMethodCall,
Sequence,
SubFactory,
)
import factory
from factory.django import DjangoModelFactory
from uuid import uuid4
from pytz import UTC

Expand All @@ -26,8 +19,7 @@ class GroupFactory(DjangoModelFactory):
FACTORY_FOR = Group
FACTORY_DJANGO_GET_OR_CREATE = ('name', )

name = Sequence(u'group{0}'.format)

name = factory.Sequence(u'group{0}'.format)


class UserStandingFactory(DjangoModelFactory):
Expand All @@ -43,7 +35,7 @@ class UserProfileFactory(DjangoModelFactory):
FACTORY_DJANGO_GET_OR_CREATE = ('user', )

user = None
name = LazyAttribute(u'{0.user.first_name} {0.user.last_name}'.format)
name = factory.LazyAttribute(u'{0.user.first_name} {0.user.last_name}'.format)
level_of_education = None
gender = u'm'
mailing_address = None
Expand Down Expand Up @@ -72,19 +64,18 @@ class UserFactory(DjangoModelFactory):
FACTORY_FOR = User
FACTORY_DJANGO_GET_OR_CREATE = ('email', 'username')

username = Sequence(u'robot{0}'.format)
email = Sequence(u'robot+test+{0}@edx.org'.format)
password = PostGenerationMethodCall('set_password',
'test')
first_name = Sequence(u'Robot{0}'.format)
username = factory.Sequence(u'robot{0}'.format)
email = factory.Sequence(u'robot+test+{0}@edx.org'.format)
password = factory.PostGenerationMethodCall('set_password', 'test')
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)

@post_generation
@factory.post_generation
def profile(obj, create, extracted, **kwargs): # pylint: disable=unused-argument, no-self-argument
if create:
obj.save()
Expand All @@ -94,7 +85,7 @@ def profile(obj, create, extracted, **kwargs): # pylint: disable=unused-argumen
else:
return None

@post_generation
@factory.post_generation
def groups(self, create, extracted, **kwargs):
if extracted is None:
return
Expand All @@ -106,7 +97,7 @@ def groups(self, create, extracted, **kwargs):
self.groups.add(GroupFactory.simple_generate(create, name=group_name))


class AnonymousUserFactory(Factory):
class AnonymousUserFactory(factory.Factory):
FACTORY_FOR = AnonymousUser


Expand All @@ -117,7 +108,7 @@ class AdminFactory(UserFactory):
class CourseEnrollmentFactory(DjangoModelFactory):
FACTORY_FOR = CourseEnrollment

user = SubFactory(UserFactory)
user = factory.SubFactory(UserFactory)
course_id = u'edX/toy/2012_Fall'


Expand All @@ -137,6 +128,6 @@ class PendingEmailChangeFactory(DjangoModelFactory):
"""
FACTORY_FOR = PendingEmailChange

user = SubFactory(UserFactory)
new_email = Sequence(u'new+email+{0}@edx.org'.format)
activation_key = Sequence(u'{:0<30d}'.format)
user = factory.SubFactory(UserFactory)
new_email = factory.Sequence(u'new+email+{0}@edx.org'.format)
activation_key = factory.Sequence(u'{:0<30d}'.format)
22 changes: 11 additions & 11 deletions lms/djangoapps/courseware/tests/factories.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import json
from functools import partial

from factory import DjangoModelFactory, SubFactory, post_generation
import factory
from factory.django import DjangoModelFactory

# Imported to re-export
# pylint: disable=unused-import
Expand Down Expand Up @@ -40,7 +40,7 @@ class InstructorFactory(UserFactory):
"""
last_name = "Instructor"

@post_generation
@factory.post_generation
def course(self, create, extracted, **kwargs):
if extracted is None:
raise ValueError("Must specify a course location for a course instructor user")
Expand All @@ -54,7 +54,7 @@ class StaffFactory(UserFactory):
"""
last_name = "Staff"

@post_generation
@factory.post_generation
def course(self, create, extracted, **kwargs):
if extracted is None:
raise ValueError("Must specify a course location for a course staff user")
Expand All @@ -68,7 +68,7 @@ class BetaTesterFactory(UserFactory):
"""
last_name = "Beta-Tester"

@post_generation
@factory.post_generation
def course(self, create, extracted, **kwargs):
if extracted is None:
raise ValueError("Must specify a course location for a beta-tester user")
Expand All @@ -82,7 +82,7 @@ class OrgStaffFactory(UserFactory):
"""
last_name = "Org-Staff"

@post_generation
@factory.post_generation
def course(self, create, extracted, **kwargs):
if extracted is None:
raise ValueError("Must specify a course location for an org-staff user")
Expand All @@ -96,7 +96,7 @@ class OrgInstructorFactory(UserFactory):
"""
last_name = "Org-Instructor"

@post_generation
@factory.post_generation
def course(self, create, extracted, **kwargs):
if extracted is None:
raise ValueError("Must specify a course location for an org-instructor user")
Expand All @@ -109,7 +109,7 @@ class GlobalStaffFactory(UserFactory):
"""
last_name = "GlobalStaff"

@post_generation
@factory.post_generation
def set_staff(self, create, extracted, **kwargs):
GlobalStaff().add_users(self)

Expand All @@ -118,7 +118,7 @@ class StudentModuleFactory(DjangoModelFactory):
FACTORY_FOR = StudentModule

module_type = "problem"
student = SubFactory(UserFactory)
student = factory.SubFactory(UserFactory)
course_id = "MITx/999/Robot_Super_Course"
state = None
grade = None
Expand All @@ -139,7 +139,7 @@ class StudentPrefsFactory(DjangoModelFactory):

field_name = 'existing_field'
value = json.dumps('old_value')
student = SubFactory(UserFactory)
student = factory.SubFactory(UserFactory)
module_type = 'MockProblemModule'


Expand All @@ -148,4 +148,4 @@ class StudentInfoFactory(DjangoModelFactory):

field_name = 'existing_field'
value = json.dumps('old_value')
student = SubFactory(UserFactory)
student = factory.SubFactory(UserFactory)
2 changes: 1 addition & 1 deletion lms/djangoapps/django_comment_client/tests/factories.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from factory import DjangoModelFactory
from factory.django import DjangoModelFactory
from django_comment_common.models import Role, Permission


Expand Down
5 changes: 3 additions & 2 deletions lms/djangoapps/instructor_task/tests/factories.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import json

from factory import DjangoModelFactory, SubFactory
import factory
from factory.django import DjangoModelFactory
from student.tests.factories import UserFactory as StudentUserFactory
from instructor_task.models import InstructorTask
from celery.states import PENDING
Expand All @@ -16,4 +17,4 @@ class InstructorTaskFactory(DjangoModelFactory):
task_id = None
task_state = PENDING
task_output = None
requester = SubFactory(StudentUserFactory)
requester = factory.SubFactory(StudentUserFactory)
5 changes: 3 additions & 2 deletions lms/djangoapps/licenses/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,8 @@
from uuid import uuid4
from random import shuffle
from tempfile import NamedTemporaryFile
from factory import DjangoModelFactory, SubFactory
import factory
from factory.django import DjangoModelFactory

from django.test import TestCase
from django.test.client import Client
Expand Down Expand Up @@ -51,7 +52,7 @@ class UserLicenseFactory(DjangoModelFactory):
FACTORY_FOR = UserLicense

user = None
software = SubFactory(CourseSoftwareFactory)
software = factory.SubFactory(CourseSoftwareFactory)
serial = SERIAL_1


Expand Down
2 changes: 1 addition & 1 deletion lms/djangoapps/user_api/tests/factories.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from factory import DjangoModelFactory
from factory.django import DjangoModelFactory
from user_api.models import UserPreference


Expand Down
2 changes: 1 addition & 1 deletion requirements/edx/base.txt
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ transifex-client==0.9.1
# Used for testing
coverage==3.7
ddt==0.4.0
factory_boy==2.0.2
factory_boy==2.2.1
mock==1.0.1
nosexcover==1.0.7
pep8==1.4.5
Expand Down