ENT-3798 Multiple_SSO_Accounts_Association_to_SAML_User (#26170)

common/djangoapps/third_party_auth/tests/test_utils.py

@@ -9,7 +9,16 @@ from django.conf import settings
 
 from common.djangoapps.student.tests.factories import UserFactory
 from common.djangoapps.third_party_auth.tests.testutil import TestCase
-from common.djangoapps.third_party_auth.utils import user_exists, convert_saml_slug_provider_id
+from common.djangoapps.third_party_auth.utils import (
+    get_user_from_email,
+    is_enterprise_customer_user,
+    user_exists,
+    convert_saml_slug_provider_id,
+)
+from openedx.features.enterprise_support.tests.factories import (
+    EnterpriseCustomerIdentityProviderFactory,
+    EnterpriseCustomerUserFactory,
+)
 
 
 @unittest.skipUnless(settings.ROOT_URLCONF == 'lms.urls', 'Test only valid in lms')
@@ -53,3 +62,40 @@ class TestUtils(TestCase):
             self.assertEqual(
                 convert_saml_slug_provider_id(provider_names[provider_id]), provider_id
             )
+
+    def test_get_user(self):
+        """
+        Match the email and return user if exists.
+        """
+        # Create users from factory
+        UserFactory(username='test_user', email='test_user@example.com')
+        self.assertTrue(
+            get_user_from_email({'email': 'test_user@example.com'}),
+        )
+        self.assertFalse(
+            get_user_from_email({'email': 'invalid@example.com'}),
+        )
+
+    def test_is_enterprise_customer_user(self):
+        """
+        Verify that if user is an enterprise learner.
+        """
+        # Create users from factory
+
+        user = UserFactory(username='test_user', email='test_user@example.com')
+        other_user = UserFactory(username='other_user', email='other_user@example.com')
+        customer_idp = EnterpriseCustomerIdentityProviderFactory.create(
+            provider_id='the-provider',
+        )
+        customer = customer_idp.enterprise_customer
+        EnterpriseCustomerUserFactory.create(
+            enterprise_customer=customer,
+            user_id=user.id,
+        )
+
+        self.assertTrue(
+            is_enterprise_customer_user('the-provider', user),
+        )
+        self.assertFalse(
+            is_enterprise_customer_user('the-provider', other_user),
+        )