test: fix tests after adding 'transaction.on_commit' running task

Author: NiedielnitsevIvan

cms/djangoapps/import_from_modulestore/tests/test_api.py

@@ -9,6 +9,7 @@
 
 from common.djangoapps.student.tests.factories import UserFactory
 from cms.djangoapps.import_from_modulestore.api import create_import, import_course_staged_content_to_library
+from cms.djangoapps.import_from_modulestore.data import ImportStatus
 from cms.djangoapps.import_from_modulestore.models import Import
 from openedx.core.djangoapps.content_libraries.tests import factories
 from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
@@ -32,18 +33,13 @@ def test_create_import(self):
         """
         course_id = "course-v1:edX+DemoX+Demo_Course"
         user = UserFactory()
-        with patch(
-            "cms.djangoapps.import_from_modulestore.api.save_legacy_content_to_staged_content_task"
-        ) as save_legacy_content_to_staged_content_task_mock:
-            create_import(course_id, user.id, self.library.learning_package_id)
+        create_import(course_id, user.id, self.library.learning_package_id)
 
         import_event = Import.objects.get()
         assert import_event.source_key == CourseKey.from_string(course_id)
         assert import_event.target == self.library.learning_package
         assert import_event.user_id == user.id
-        save_legacy_content_to_staged_content_task_mock.apply_async.assert_called_once_with(
-            kwargs={'import_uuid': import_event.uuid}
-        )
+        assert import_event.status == ImportStatus.PENDING
 
     def test_import_course_staged_content_to_library(self):
         """

cms/djangoapps/import_from_modulestore/tests/test_helpers.py

@@ -47,12 +47,12 @@ def setUp(self):
             display_name='Video',
             data="""<video youtube="1.00:3_yD_cEKoCk" url_name="SampleProblem"/>""",
         )
-
-        self.import_event = api.create_import(
-            source_key=self.course.id,
-            learning_package_id=self.library.learning_package.id,
-            user_id=self.user.id,
-        )
+        with self.captureOnCommitCallbacks(execute=True):
+            self.import_event = api.create_import(
+                source_key=self.course.id,
+                learning_package_id=self.library.learning_package.id,
+                user_id=self.user.id,
+            )
         self.parser = etree.XMLParser(strip_cdata=False)
 
     def test_import_from_staged_content(self):

cms/djangoapps/import_from_modulestore/views/v0/tests/test_views.py

@@ -44,11 +44,12 @@ def setUp(self):
         self.vertical = BlockFactory.create(category='vertical', parent=self.sequential)
         self.problem = BlockFactory.create(category='problem', parent=self.vertical)
 
-        self.import_event = api.create_import(
-            user_id=self.admin_user.pk,
-            learning_package_id=self.library.learning_package.id,
-            source_key=self.course.id,
-        )
+        with self.captureOnCommitCallbacks(execute=True):
+            self.import_event = api.create_import(
+                user_id=self.admin_user.pk,
+                learning_package_id=self.library.learning_package.id,
+                source_key=self.course.id,
+            )
 
 
 class ImportBlocksViewTest(TestCourseToLibraryImportViewsMixin):