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