31 lines
1.2 KiB
Python
31 lines
1.2 KiB
Python
"""
|
|
Unit tests for third_party_auth SAML auth providers
|
|
"""
|
|
|
|
import mock
|
|
|
|
from third_party_auth.tests.testutil import SAMLTestCase
|
|
from third_party_auth.saml import EdXSAMLIdentityProvider, get_saml_idp_class
|
|
from third_party_auth.tests.data.saml_identity_provider_mock_data import mock_conf, mock_attributes,\
|
|
expected_user_details
|
|
|
|
|
|
class TestEdXSAMLIdentityProvider(SAMLTestCase):
|
|
"""
|
|
Test EdXSAMLIdentityProvider.
|
|
"""
|
|
@mock.patch('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(
|
|
u'%s is not a valid EdXSAMLIdentityProvider subclass; using EdXSAMLIdentityProvider base class.',
|
|
'fake_idp_class_option'
|
|
)
|
|
self.assertIs(idp_class, EdXSAMLIdentityProvider)
|
|
|
|
def test_get_user_details(self):
|
|
""" test get_attr and get_user_details of EdXSAMLIdentityProvider"""
|
|
edx_saml_identity_provider = EdXSAMLIdentityProvider('demo', **mock_conf)
|
|
self.assertEqual(edx_saml_identity_provider.get_user_details(mock_attributes), expected_user_details)
|