36 lines
955 B
Python
36 lines
955 B
Python
"""Provides factories for User API models."""
|
|
from factory.django import DjangoModelFactory
|
|
from factory import SubFactory
|
|
from student.tests.factories import UserFactory
|
|
from user_api.models import UserPreference, UserCourseTag, UserOrgTag
|
|
from opaque_keys.edx.locations import SlashSeparatedCourseKey
|
|
|
|
|
|
# Factories are self documenting
|
|
# pylint: disable=missing-docstring
|
|
class UserPreferenceFactory(DjangoModelFactory):
|
|
FACTORY_FOR = UserPreference
|
|
|
|
user = None
|
|
key = None
|
|
value = "default test value"
|
|
|
|
|
|
class UserCourseTagFactory(DjangoModelFactory):
|
|
FACTORY_FOR = UserCourseTag
|
|
|
|
user = SubFactory(UserFactory)
|
|
course_id = SlashSeparatedCourseKey('org', 'course', 'run')
|
|
key = None
|
|
value = None
|
|
|
|
|
|
class UserOrgTagFactory(DjangoModelFactory):
|
|
""" Simple factory class for generating UserOrgTags """
|
|
FACTORY_FOR = UserOrgTag
|
|
|
|
user = SubFactory(UserFactory)
|
|
org = 'org'
|
|
key = None
|
|
value = None
|