add utils tests

Author: asadali145

src/ol_openedx_course_sync/tests/test_utils.py

@@ -0,0 +1,48 @@
+"""
+Tests for ol-openedx-course-sync utils.
+"""
+
+from unittest import mock
+
+from ddt import data, ddt
+from ol_openedx_course_sync.utils import copy_course_content
+from openedx.core.djangolib.testing.utils import skip_unless_cms
+from tests.utils import OLOpenedXCourseSyncTestCase
+from xmodule.modulestore import ModuleStoreEnum
+from xmodule.modulestore.django import modulestore
+
+
+@ddt
+class TestUtils(OLOpenedXCourseSyncTestCase):
+    """
+    Test the ol_openedx_course_sync utils.
+    """
+
+    @skip_unless_cms
+    @data(ModuleStoreEnum.BranchName.draft, ModuleStoreEnum.BranchName.published)
+    def test_copy_course_content(self, branch):
+        """
+        Test the copy_course_content function.
+        """
+        with mock.patch(
+            "ol_openedx_course_sync.utils.modulestore"
+        ) as mixed_modulestore_mock:
+            split_modulestore_mock = mock.Mock()
+            split_modulestore_mock.copy = mock.Mock()
+            mixed_modulestore_mock.return_value = mock.Mock(
+                make_course_usage_key=mock.Mock(
+                    return_value=modulestore().make_course_usage_key(
+                        self.source_course.usage_key.course_key
+                    )
+                ),
+                _get_modulestore_for_courselike=mock.Mock(
+                    side_effect=[split_modulestore_mock, split_modulestore_mock]
+                ),
+            )
+            copy_course_content(
+                None,
+                self.source_course.usage_key.course_key,
+                self.target_course.usage_key.course_key,
+                branch,
+            )
+            split_modulestore_mock.copy.assert_called_once()

src/ol_openedx_course_sync/tests/utils.py

@@ -0,0 +1,91 @@
+"""
+Utils for ol-openedx-course-sync tests.
+"""
+
+from datetime import datetime
+
+from pytz import UTC
+from xblock.runtime import DictKeyValueStore, KvsFieldData
+from xblock.test.tools import TestRuntime
+from xmodule.modulestore import ModuleStoreEnum
+from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
+from xmodule.modulestore.tests.factories import BlockFactory, CourseFactory
+
+
+class OLOpenedXCourseSyncTestCase(ModuleStoreTestCase):
+    """
+    Base test case for ol-openedx-course-sync tests.
+    """
+
+    def setUp(self):
+        super().setUp()
+
+        key_store = DictKeyValueStore()
+        field_data = KvsFieldData(key_store)
+        self.runtime = TestRuntime(services={"field-data": field_data})
+
+        self.source_course = CourseFactory.create(
+            default_store=ModuleStoreEnum.Type.split
+        )
+        self.source_course_block = BlockFactory.create(
+            parent_location=self.source_course.location,
+            category="course",
+            display_name="Source course",
+        )
+        self.create_course_blocks(self.source_course_block)
+
+        self.target_course = CourseFactory.create(
+            default_store=ModuleStoreEnum.Type.split
+        )
+        self.target_course_block = BlockFactory.create(
+            parent_location=self.target_course.location,
+            category="course",
+            display_name="Target course",
+        )
+
+    def create_course_blocks(self, course_block):
+        """
+        Create a set of blocks for the course.
+        """
+        chapter = BlockFactory.create(
+            parent_location=course_block.location,
+            category="chapter",
+            display_name="Week 1",
+            publish_item=True,
+            start=datetime(2015, 3, 1, tzinfo=UTC),
+        )
+        sequential = BlockFactory.create(
+            parent_location=chapter.location,
+            category="sequential",
+            display_name="Lesson 1",
+            publish_item=True,
+            start=datetime(2015, 3, 1, tzinfo=UTC),
+        )
+        vertical = BlockFactory.create(
+            parent_location=sequential.location,
+            category="vertical",
+            display_name="Subsection 1",
+            publish_item=True,
+            start=datetime(2015, 4, 1, tzinfo=UTC),
+        )
+
+        BlockFactory.create(
+            category="problem",
+            parent_location=vertical.location,
+            display_name="A Problem Block",
+            weight=1,
+            user_id=self.user.id,
+            publish_item=False,
+        )
+        BlockFactory.create(
+            parent_location=vertical.location,
+            category="video",
+            display_name="My Video",
+            user_id=self.user.id,
+        )
+        BlockFactory.create(
+            category="html",
+            parent_location=vertical.location,
+            display_name="An HTML Block",
+            user_id=self.user.id,
+        )