106 lines
4.2 KiB
Python
106 lines
4.2 KiB
Python
"""
|
|
Unit tests for course tools.
|
|
"""
|
|
|
|
import crum
|
|
import datetime
|
|
|
|
from mock import patch
|
|
from nose.plugins.attrib import attr
|
|
import pytz
|
|
from django.test import RequestFactory
|
|
|
|
from course_modes.models import CourseMode
|
|
from course_modes.tests.factories import CourseModeFactory
|
|
from courseware.course_tools import VerifiedUpgradeTool
|
|
from courseware.models import DynamicUpgradeDeadlineConfiguration
|
|
from openedx.core.djangoapps.content.course_overviews.models import CourseOverview
|
|
from openedx.core.djangoapps.schedules.config import CREATE_SCHEDULE_WAFFLE_FLAG
|
|
from openedx.core.djangoapps.site_configuration.tests.factories import SiteFactory
|
|
from openedx.core.djangoapps.waffle_utils.testutils import override_waffle_flag
|
|
from student.tests.factories import CourseEnrollmentFactory, UserFactory
|
|
from xmodule.modulestore.tests.django_utils import SharedModuleStoreTestCase
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
|
|
|
|
@attr(shard=3)
|
|
class VerifiedUpgradeToolTest(SharedModuleStoreTestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(VerifiedUpgradeToolTest, cls).setUpClass()
|
|
cls.now = datetime.datetime.now(pytz.UTC)
|
|
|
|
cls.course = CourseFactory.create(
|
|
org='edX',
|
|
number='test',
|
|
display_name='Test Course',
|
|
self_paced=True,
|
|
start=cls.now - datetime.timedelta(days=30),
|
|
)
|
|
cls.course_overview = CourseOverview.get_from_id(cls.course.id)
|
|
|
|
@override_waffle_flag(CREATE_SCHEDULE_WAFFLE_FLAG, True)
|
|
def setUp(self):
|
|
super(VerifiedUpgradeToolTest, self).setUp()
|
|
|
|
self.course_verified_mode = CourseModeFactory(
|
|
course_id=self.course.id,
|
|
mode_slug=CourseMode.VERIFIED,
|
|
expiration_datetime=self.now + datetime.timedelta(days=30),
|
|
)
|
|
|
|
patcher = patch('openedx.core.djangoapps.schedules.signals.get_current_site')
|
|
mock_get_current_site = patcher.start()
|
|
self.addCleanup(patcher.stop)
|
|
mock_get_current_site.return_value = SiteFactory.create()
|
|
|
|
DynamicUpgradeDeadlineConfiguration.objects.create(enabled=True)
|
|
|
|
self.enrollment = CourseEnrollmentFactory(
|
|
course_id=self.course.id,
|
|
mode=CourseMode.AUDIT,
|
|
course=self.course_overview,
|
|
)
|
|
self.request = RequestFactory().request()
|
|
self.request.user = self.enrollment.user
|
|
crum.set_current_request(self.request)
|
|
|
|
def test_tool_visible(self):
|
|
self.assertTrue(VerifiedUpgradeTool().is_enabled(self.request, self.course.id))
|
|
|
|
def test_not_visible_when_no_enrollment_exists(self):
|
|
self.enrollment.delete()
|
|
|
|
request = RequestFactory().request()
|
|
request.user = UserFactory()
|
|
self.assertFalse(VerifiedUpgradeTool().is_enabled(self.request, self.course.id))
|
|
|
|
def test_not_visible_when_using_deadline_from_course_mode(self):
|
|
DynamicUpgradeDeadlineConfiguration.objects.create(enabled=False)
|
|
self.assertFalse(VerifiedUpgradeTool().is_enabled(self.request, self.course.id))
|
|
|
|
def test_not_visible_when_enrollment_is_inactive(self):
|
|
self.enrollment.is_active = False
|
|
self.enrollment.save()
|
|
self.assertFalse(VerifiedUpgradeTool().is_enabled(self.request, self.course.id))
|
|
|
|
def test_not_visible_when_already_verified(self):
|
|
self.enrollment.mode = CourseMode.VERIFIED
|
|
self.enrollment.save()
|
|
self.assertFalse(VerifiedUpgradeTool().is_enabled(self.request, self.course.id))
|
|
|
|
def test_not_visible_when_no_verified_track(self):
|
|
self.course_verified_mode.delete()
|
|
self.assertFalse(VerifiedUpgradeTool().is_enabled(self.request, self.course.id))
|
|
|
|
def test_not_visible_when_course_deadline_has_passed(self):
|
|
self.course_verified_mode.expiration_datetime = self.now - datetime.timedelta(days=1)
|
|
self.course_verified_mode.save()
|
|
self.assertFalse(VerifiedUpgradeTool().is_enabled(self.request, self.course.id))
|
|
|
|
def test_not_visible_when_course_mode_has_no_deadline(self):
|
|
self.course_verified_mode.expiration_datetime = None
|
|
self.course_verified_mode.save()
|
|
self.assertFalse(VerifiedUpgradeTool().is_enabled(self.request, self.course.id))
|