def test_not_valid_xml(self, mock_b64decode):
a = SAMLAuthenticator()
fake_data = {a.login_post_field: 'this string isn\'t important'}
mock_b64decode.return_value = 'bad xml string'
assert a._get_saml_doc_etree(fake_data) is None
def test_get_saml_doc_different_location(self):
a = SAMLAuthenticator()
a.login_post_field = 'test'
fake_data = {
a.login_post_field: test_constants.b64encoded_response_xml
}
faked_etree = a._get_saml_doc_etree(fake_data)
real_etree = etree.fromstring(test_constants.sample_response_xml)
assert etree.tostring(faked_etree) == etree.tostring(real_etree)
def test_get_saml_doc_etree(self):
# We expect the SAML Response to be coming in base 64 encoded
a = SAMLAuthenticator()
fake_data = {
a.login_post_field: test_constants.b64encoded_response_xml
}
faked_etree = a._get_saml_doc_etree(fake_data)
real_etree = etree.fromstring(test_constants.sample_response_xml)
assert etree.tostring(faked_etree) == etree.tostring(real_etree)
def test_bad_decode(self):
a = SAMLAuthenticator()
fake_data = {a.login_post_field: 'this is not base 64 encoded data'}
assert a._get_saml_doc_etree(fake_data) is None
def test_with_failed_get(self):
a = SAMLAuthenticator()
fake_data = {}
assert a._get_saml_doc_etree(fake_data) is None
