fix: point to new models in channel_migrations app

Author: Sameen Fatima

openedx/core/djangoapps/user_api/accounts/tests/test_retirement_views.py

@@ -11,6 +11,7 @@
 from consent.models import DataSharingConsent
 from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
 from django.contrib.sites.models import Site
+from django.conf import settings
 from django.core import mail
 from django.core.cache import cache
 from django.test import TestCase
@@ -21,7 +22,11 @@
     EnterpriseCustomerUser,
     PendingEnterpriseCustomerUser
 )
-from integrated_channels.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
+# This is a temporary import path while we transition from integrated_channels to channel_integrations
+if getattr(settings, 'ENABLE_LEGACY_INTEGRATED_CHANNELS', True):
+    from integrated_channels.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
+else:
+    from channel_integrations.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
 from opaque_keys.edx.keys import CourseKey
 from rest_framework import status
 from social_django.models import UserSocialAuth

openedx/core/djangoapps/user_api/accounts/views.py

@@ -26,8 +26,13 @@
 from edx_rest_framework_extensions.auth.jwt.authentication import JwtAuthentication
 from edx_rest_framework_extensions.auth.session.authentication import SessionAuthenticationAllowInactiveUser
 from enterprise.models import EnterpriseCourseEnrollment, EnterpriseCustomerUser, PendingEnterpriseCustomerUser
-from integrated_channels.degreed.models import DegreedLearnerDataTransmissionAudit
-from integrated_channels.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
+# This is a temporary import path while we transition from integrated_channels to channel_integrations
+if getattr(settings, 'ENABLE_LEGACY_INTEGRATED_CHANNELS', True):
+    from integrated_channels.degreed.models import DegreedLearnerDataTransmissionAudit
+    from integrated_channels.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
+else:
+    from channel_integrations.degreed2.models import Degreed2LearnerDataTransmissionAudit as DegreedLearnerDataTransmissionAudit
+    from channel_integrations.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
 from rest_framework import permissions, status
 from rest_framework.authentication import SessionAuthentication
 from rest_framework.exceptions import UnsupportedMediaType

openedx/core/djangoapps/user_api/management/commands/create_user_gdpr_testing.py

@@ -11,14 +11,19 @@
 
 from consent.models import DataSharingConsent
 from django.contrib.auth.models import User  # lint-amnesty, pylint: disable=imported-auth-user
+from django.conf import settings
 from django.core.management.base import BaseCommand
 from enterprise.models import (
     EnterpriseCourseEnrollment,
     EnterpriseCustomer,
     EnterpriseCustomerUser,
     PendingEnterpriseCustomerUser
 )
-from integrated_channels.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
+# This is a temporary import path while we transition from integrated_channels to channel_integrations
+if getattr(settings, 'ENABLE_LEGACY_INTEGRATED_CHANNELS', True):
+    from integrated_channels.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
+else:
+    from channel_integrations.sap_success_factors.models import SapSuccessFactorsLearnerDataTransmissionAudit
 from opaque_keys.edx.keys import CourseKey
 from zoneinfo import ZoneInfo
 

openedx/features/enterprise_support/signals.py

@@ -10,10 +10,17 @@
 from django.db.models.signals import post_save, pre_save
 from django.dispatch import receiver
 from enterprise.models import EnterpriseCourseEnrollment, EnterpriseCustomer
-from integrated_channels.integrated_channel.tasks import (
-    transmit_single_learner_data,
-    transmit_single_subsection_learner_data
-)
+# This is a temporary import path while we transition from integrated_channels to channel_integrations
+if getattr(settings, 'ENABLE_LEGACY_INTEGRATED_CHANNELS', True):
+    from integrated_channels.integrated_channel.tasks import (
+        transmit_single_learner_data,
+        transmit_single_subsection_learner_data
+    )
+else:
+    from channel_integrations.integrated_channel.tasks import (
+        transmit_single_learner_data,
+        transmit_single_subsection_learner_data
+    )
 from slumber.exceptions import HttpClientError
 
 from common.djangoapps.student.signals import UNENROLL_DONE

openedx/features/enterprise_support/tests/test_signals.py

@@ -6,6 +6,7 @@
 from unittest.mock import patch
 
 import ddt
+from django.conf import settings
 from django.test.utils import override_settings
 from django.utils.timezone import now
 from edx_django_utils.cache import TieredCache
@@ -196,7 +197,9 @@ def test_handle_enterprise_learner_passing_grade(self):
         Test to assert transmit_single_learner_data is called when COURSE_GRADE_NOW_PASSED signal is fired
         """
         with patch(
-            'integrated_channels.integrated_channel.tasks.transmit_single_learner_data.apply_async',
+            'integrated_channels.integrated_channel.tasks.transmit_single_subsection_learner_data.apply_async' 
+            if getattr(settings, 'ENABLE_LEGACY_INTEGRATED_CHANNELS', True) else
+            'channel_integrations.integrated_channel.tasks.transmit_single_subsection_learner_data.apply_async',
             return_value=None
         ) as mock_task_apply:
             course_key = CourseKey.from_string(self.course_id)
@@ -218,7 +221,9 @@ def test_handle_enterprise_learner_subsection(self):
         Test to assert transmit_subsection_learner_data is called when COURSE_ASSESSMENT_GRADE_CHANGED signal is fired.
         """
         with patch(
-            'integrated_channels.integrated_channel.tasks.transmit_single_subsection_learner_data.apply_async',
+            'integrated_channels.integrated_channel.tasks.transmit_single_subsection_learner_data.apply_async' 
+            if getattr(settings, 'ENABLE_LEGACY_INTEGRATED_CHANNELS', True) else
+            'channel_integrations.integrated_channel.tasks.transmit_single_subsection_learner_data.apply_async',
             return_value=None
         ) as mock_task_apply:
             course_key = CourseKey.from_string(self.course_id)