3d1f3cea64
Ran pyupgrade on lms/djangoapps/course_goals Ran pyugprade on lms/djangoapps/course_home_api
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
"""
|
|
Base classes or util functions for use in Course Home API tests
|
|
"""
|
|
|
|
import unittest
|
|
from datetime import datetime
|
|
|
|
from django.conf import settings
|
|
|
|
from common.djangoapps.course_modes.models import CourseMode
|
|
from common.djangoapps.course_modes.tests.factories import CourseModeFactory
|
|
from lms.djangoapps.courseware.tests.helpers import MasqueradeMixin
|
|
from lms.djangoapps.verify_student.models import VerificationDeadline
|
|
from openedx.core.djangoapps.content.course_overviews.tests.factories import CourseOverviewFactory
|
|
from xmodule.modulestore.tests.django_utils import ( # lint-amnesty, pylint: disable=unused-import
|
|
TEST_DATA_SPLIT_MODULESTORE,
|
|
ModuleStoreTestCase
|
|
)
|
|
from xmodule.modulestore.tests.factories import CourseFactory, ItemFactory
|
|
|
|
|
|
@unittest.skipUnless(settings.ROOT_URLCONF == 'lms.urls', 'Test only valid in lms')
|
|
class BaseCourseHomeTests(ModuleStoreTestCase, MasqueradeMixin):
|
|
"""
|
|
Base class for Course Home API tests.
|
|
|
|
Creates a course to
|
|
"""
|
|
MODULESTORE = TEST_DATA_SPLIT_MODULESTORE
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
self.course = CourseFactory.create(
|
|
start=datetime(2020, 1, 1),
|
|
end=datetime(2028, 1, 1),
|
|
enrollment_start=datetime(2020, 1, 1),
|
|
enrollment_end=datetime(2028, 1, 1),
|
|
emit_signals=True,
|
|
modulestore=self.store,
|
|
)
|
|
chapter = ItemFactory(parent=self.course, category='chapter')
|
|
ItemFactory(parent=chapter, category='sequential')
|
|
|
|
CourseModeFactory(course_id=self.course.id, mode_slug=CourseMode.AUDIT)
|
|
CourseModeFactory(
|
|
course_id=self.course.id,
|
|
mode_slug=CourseMode.VERIFIED,
|
|
expiration_datetime=datetime(2028, 1, 1),
|
|
min_price=149,
|
|
sku='ABCD1234',
|
|
)
|
|
VerificationDeadline.objects.create(course_key=self.course.id, deadline=datetime(2028, 1, 1))
|
|
|
|
CourseOverviewFactory.create(run='1T2020')
|
|
|
|
self.staff_user = self.user
|
|
self.user, password = self.create_non_staff_user()
|
|
self.client.login(username=self.user.username, password=password)
|
|
|
|
def switch_to_staff(self):
|
|
self.user = self.staff_user
|
|
self.client.login(username=self.user.username, password=self.user_password)
|