e07a0cc163
* feat: add comment to course reset model * feat: add comment info to list endpoint * feat: add comment to post endpoint * fixup! feat: add comment to post endpoint
31 lines
978 B
Python
31 lines
978 B
Python
""" Factories for course reset models """
|
|
import factory
|
|
from factory.django import DjangoModelFactory
|
|
from common.djangoapps.student.tests.factories import CourseEnrollmentFactory, UserFactory
|
|
|
|
|
|
from lms.djangoapps.support.models import (
|
|
CourseResetCourseOptIn,
|
|
CourseResetAudit
|
|
)
|
|
|
|
|
|
class CourseResetCourseOptInFactory(DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
|
class Meta:
|
|
model = CourseResetCourseOptIn
|
|
|
|
course_id = None
|
|
active = True
|
|
|
|
|
|
class CourseResetAuditFactory(DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
|
class Meta:
|
|
model = CourseResetAudit
|
|
|
|
course = factory.SubFactory(CourseResetCourseOptInFactory)
|
|
course_enrollment = factory.SubFactory(CourseEnrollmentFactory)
|
|
reset_by = factory.SubFactory(UserFactory)
|
|
status = CourseResetAudit.CourseResetStatus.ENQUEUED
|
|
comment = factory.Sequence(lambda i: f'comment {i}')
|
|
completed_at = None
|