dc7db1d3ad
* feat: unpinned social-auth-core feat: unpinned social-auth-core * fix: updated to resolve failing tests * fix: resolved linter errors and failing tests * fix: updated get_attr signature according to new version
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
"""
|
|
Unit tests for third_party_auth SAML auth providers
|
|
"""
|
|
|
|
from unittest import mock
|
|
|
|
from django.utils.datastructures import MultiValueDictKeyError
|
|
from social_core.exceptions import AuthMissingParameter
|
|
|
|
from common.djangoapps.third_party_auth.dummy import DummyBackend
|
|
from common.djangoapps.third_party_auth.saml import EdXSAMLIdentityProvider, get_saml_idp_class, SAMLAuthBackend
|
|
from common.djangoapps.third_party_auth.tests.data.saml_identity_provider_mock_data import (
|
|
expected_user_details,
|
|
mock_attributes,
|
|
mock_conf
|
|
)
|
|
from common.djangoapps.third_party_auth.tests.testutil import SAMLTestCase
|
|
|
|
|
|
class TestEdXSAMLIdentityProvider(SAMLTestCase):
|
|
"""
|
|
Test EdXSAMLIdentityProvider.
|
|
"""
|
|
|
|
@mock.patch('common.djangoapps.third_party_auth.saml.log')
|
|
def test_get_saml_idp_class_with_fake_identifier(self, log_mock):
|
|
error_mock = log_mock.error
|
|
idp_class = get_saml_idp_class('fake_idp_class_option')
|
|
error_mock.assert_called_once_with(
|
|
'[THIRD_PARTY_AUTH] Invalid EdXSAMLIdentityProvider subclass--'
|
|
'using EdXSAMLIdentityProvider base class. Provider: {provider}'.format(provider='fake_idp_class_option')
|
|
)
|
|
assert idp_class is EdXSAMLIdentityProvider
|
|
|
|
def test_get_user_details(self):
|
|
""" test get_attr and get_user_details of EdXSAMLIdentityProvider"""
|
|
edx_saml_identity_provider = EdXSAMLIdentityProvider(DummyBackend, 'demo', **mock_conf)
|
|
assert edx_saml_identity_provider.get_user_details(mock_attributes) == expected_user_details
|
|
|
|
|
|
class TestSAMLAuthBackend(SAMLTestCase):
|
|
""" Tests for the SAML backend. """
|
|
|
|
@mock.patch('common.djangoapps.third_party_auth.saml.SAMLAuth.auth_complete')
|
|
def test_saml_auth_complete(self, super_auth_complete):
|
|
super_auth_complete.side_effect = MultiValueDictKeyError('RelayState')
|
|
backend = SAMLAuthBackend()
|
|
with self.assertRaises(AuthMissingParameter) as cm:
|
|
backend.auth_complete()
|
|
|
|
assert cm.exception.parameter == 'RelayState'
|