25 lines
750 B
Python
25 lines
750 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
|
|
from opaque_keys.edx.locations import SlashSeparatedCourseKey
|
|
|
|
# Factories don't have __init__ methods, and are self documenting
|
|
# pylint: disable=W0232, C0111
|
|
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
|