summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/saml2/__init__.py3
-rw-r--r--tests/test_82_pefim.py2
2 files changed, 2 insertions, 3 deletions
diff --git a/src/saml2/__init__.py b/src/saml2/__init__.py
index 7a7079e5..12eabaa8 100644
--- a/src/saml2/__init__.py
+++ b/src/saml2/__init__.py
@@ -570,6 +570,7 @@ class SamlBase(ExtensionContainer):
uri, tag = elem.tag[1:].split("}")
return uri
return None
+
def get_ns_map(self, elements, uri_set):
for elem in elements:
@@ -608,7 +609,6 @@ class SamlBase(ExtensionContainer):
return ElementTree.tostring(tree, encoding="UTF-8")
-
def set_prefixes(self, elem, prefix_map):
# check if this is a tree wrapper
@@ -626,7 +626,6 @@ class SamlBase(ExtensionContainer):
for elem in elem.getiterator():
self.fixup_element_prefixes(elem, uri_map, memo)
-
def fixup_element_prefixes(self, elem, uri_map, memo):
def fixup(name):
try:
diff --git a/tests/test_82_pefim.py b/tests/test_82_pefim.py
index d6175660..95bca6a3 100644
--- a/tests/test_82_pefim.py
+++ b/tests/test_82_pefim.py
@@ -48,5 +48,5 @@ _elem = extension_elements_to_elements(parsed.extensions.extension_elements,
assert len(_elem) == 1
_spcertenc = _elem[0]
-_cert = _spcertenc.key_info[0].x509_data[0].x509_certificate.text
+_cert = _spcertenc.x509_data[0].x509_certificate.text
assert cert == _cert