37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
"""
|
|
Factories for Program Enrollment tests.
|
|
"""
|
|
from __future__ import absolute_import
|
|
|
|
from uuid import uuid4
|
|
|
|
import factory
|
|
from factory.django import DjangoModelFactory
|
|
from opaque_keys.edx.keys import CourseKey
|
|
|
|
from lms.djangoapps.program_enrollments import models
|
|
from student.tests.factories import CourseEnrollmentFactory, UserFactory
|
|
|
|
|
|
class ProgramEnrollmentFactory(DjangoModelFactory):
|
|
""" A Factory for the ProgramEnrollment model. """
|
|
class Meta(object):
|
|
model = models.ProgramEnrollment
|
|
|
|
user = factory.SubFactory(UserFactory)
|
|
external_user_key = None
|
|
program_uuid = uuid4()
|
|
curriculum_uuid = uuid4()
|
|
status = 'enrolled'
|
|
|
|
|
|
class ProgramCourseEnrollmentFactory(DjangoModelFactory):
|
|
""" A factory for the ProgramCourseEnrollment model. """
|
|
class Meta(object):
|
|
model = models.ProgramCourseEnrollment
|
|
|
|
program_enrollment = factory.SubFactory(ProgramEnrollmentFactory)
|
|
course_enrollment = factory.SubFactory(CourseEnrollmentFactory)
|
|
course_key = CourseKey.from_string("course-v1:edX+DemoX+Demo_Course")
|
|
status = 'active'
|