61 lines
2.6 KiB
Python
61 lines
2.6 KiB
Python
"""
|
|
Unit tests for the gating feature in Studio
|
|
"""
|
|
from milestones.tests.utils import MilestonesTestCaseMixin
|
|
from mock import patch
|
|
|
|
from contentstore.signals.handlers import handle_item_deleted
|
|
from openedx.core.lib.gating import api as gating_api
|
|
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
|
|
from xmodule.modulestore.tests.factories import CourseFactory, ItemFactory
|
|
|
|
|
|
class TestHandleItemDeleted(ModuleStoreTestCase, MilestonesTestCaseMixin):
|
|
"""
|
|
Test case for handle_score_changed django signal handler
|
|
"""
|
|
ENABLED_SIGNALS = ['course_published']
|
|
|
|
def setUp(self):
|
|
"""
|
|
Initial data setup
|
|
"""
|
|
super(TestHandleItemDeleted, self).setUp()
|
|
|
|
self.course = CourseFactory.create()
|
|
self.course.enable_subsection_gating = True
|
|
self.course.save()
|
|
self.chapter = ItemFactory.create(
|
|
parent=self.course,
|
|
category="chapter",
|
|
display_name="Chapter"
|
|
)
|
|
self.open_seq = ItemFactory.create(
|
|
parent=self.chapter,
|
|
category='sequential',
|
|
display_name="Open Sequential"
|
|
)
|
|
self.gated_seq = ItemFactory.create(
|
|
parent=self.chapter,
|
|
category='sequential',
|
|
display_name="Gated Sequential"
|
|
)
|
|
gating_api.add_prerequisite(self.course.id, self.open_seq.location)
|
|
gating_api.set_required_content(self.course.id, self.gated_seq.location, self.open_seq.location, 100)
|
|
|
|
@patch('contentstore.signals.handlers.gating_api.set_required_content')
|
|
@patch('contentstore.signals.handlers.gating_api.remove_prerequisite')
|
|
def test_chapter_deleted(self, mock_remove_prereq, mock_set_required):
|
|
""" Test gating milestone data is cleanup up when course content item is deleted """
|
|
handle_item_deleted(usage_key=self.chapter.location, user_id=0)
|
|
mock_remove_prereq.assert_called_with(self.open_seq.location)
|
|
mock_set_required.assert_called_with(self.open_seq.location.course_key, self.open_seq.location, None, None)
|
|
|
|
@patch('contentstore.signals.handlers.gating_api.set_required_content')
|
|
@patch('contentstore.signals.handlers.gating_api.remove_prerequisite')
|
|
def test_sequential_deleted(self, mock_remove_prereq, mock_set_required):
|
|
""" Test gating milestone data is cleanup up when course content item is deleted """
|
|
handle_item_deleted(usage_key=self.open_seq.location, user_id=0)
|
|
mock_remove_prereq.assert_called_with(self.open_seq.location)
|
|
mock_set_required.assert_called_with(self.open_seq.location.course_key, self.open_seq.location, None, None)
|