Upgrade factory_boy

TNL-3179
This commit is contained in:
muhammad-ammar
2015-08-31 13:52:20 +05:00
parent 0a2cfbe31c
commit 6a5ce5d544
21 changed files with 113 additions and 57 deletions

View File

@@ -18,7 +18,8 @@ class CohortFactory(DjangoModelFactory):
"""
Factory for constructing mock cohorts.
"""
FACTORY_FOR = CourseUserGroup
class Meta(object): # pylint: disable=missing-docstring
model = CourseUserGroup
name = Sequence("cohort{}".format)
course_id = SlashSeparatedCourseKey("dummy", "dummy", "dummy")
@@ -37,7 +38,8 @@ class CourseCohortFactory(DjangoModelFactory):
"""
Factory for constructing mock course cohort.
"""
FACTORY_FOR = CourseCohort
class Meta(object): # pylint: disable=missing-docstring
model = CourseCohort
course_user_group = factory.SubFactory(CohortFactory)
assignment_type = 'manual'
@@ -47,7 +49,8 @@ class CourseCohortSettingsFactory(DjangoModelFactory):
"""
Factory for constructing mock course cohort settings.
"""
FACTORY_FOR = CourseCohortsSettings
class Meta(object): # pylint: disable=missing-docstring
model = CourseCohortsSettings
is_cohorted = False
course_id = SlashSeparatedCourseKey("dummy", "dummy", "dummy")

View File

@@ -10,7 +10,8 @@ from ..models import UserPreference, UserCourseTag, UserOrgTag
# Factories are self documenting
# pylint: disable=missing-docstring
class UserPreferenceFactory(DjangoModelFactory):
FACTORY_FOR = UserPreference
class Meta(object):
model = UserPreference
user = None
key = None
@@ -18,7 +19,8 @@ class UserPreferenceFactory(DjangoModelFactory):
class UserCourseTagFactory(DjangoModelFactory):
FACTORY_FOR = UserCourseTag
class Meta(object):
model = UserCourseTag
user = SubFactory(UserFactory)
course_id = SlashSeparatedCourseKey('org', 'course', 'run')
@@ -28,7 +30,8 @@ class UserCourseTagFactory(DjangoModelFactory):
class UserOrgTagFactory(DjangoModelFactory):
""" Simple factory class for generating UserOrgTags """
FACTORY_FOR = UserOrgTag
class Meta(object):
model = UserOrgTag
user = SubFactory(UserFactory)
org = 'org'