def test_get_invalid_xml_element(self):
a = SAMLAuthenticator()
a.metadata_content = test_constants.sample_metadata_xml
mock_handler_self = MagicMock()
with self.assertRaises(IndexError):
a._get_redirect_from_metadata_and_redirect('md:BadElement',
mock_handler_self)
def test_get_empty_metadata(self):
a = SAMLAuthenticator()
a.metadata_filepath = None
a.metadata_content = None
a.metadata_url = None
mock_handler_self = MagicMock()
with self.assertRaises(HTTPError):
a._get_redirect_from_metadata_and_redirect('md:BadElement',
mock_handler_self)
def test_get_valid_logout_redirect(self):
a = SAMLAuthenticator()
a.metadata_content = test_constants.sample_metadata_xml
mock_handler_self = MagicMock()
a._get_redirect_from_metadata_and_redirect('md:SingleLogoutService',
mock_handler_self)
mock_handler_self.redirect.assert_called_once_with(
'https://bluedata-test-before-deploy.onelogin.com/trust/saml2/http-redirect/slo/719630',
permanent=False)
