Don't clean xsd and xsi namespaces

tachang · tachang · commit 25dfdc0a6942 · 2019-07-26T12:16:14.000-07:00
diff --git a/src/onelogin/saml2/constants.py b/src/onelogin/saml2/constants.py
@@ -53,6 +53,7 @@ class OneLogin_Saml2_Constants(object):
     NS_PREFIX_MD = 'md'
     NS_PREFIX_XS = 'xs'
     NS_PREFIX_XSI = 'xsi'
+    NS_PREFIX_XSD = 'xsd'
     NS_PREFIX_XENC = 'xenc'
     NS_PREFIX_DS = 'ds'
 
diff --git a/src/onelogin/saml2/xml_utils.py b/src/onelogin/saml2/xml_utils.py
@@ -147,7 +147,9 @@ def cleanup_namespaces(tree_or_element, top_nsmap=None, keep_ns_prefixes=None):
         :rtype: etree.Element
         """
         all_prefixes_to_keep = [
-            OneLogin_Saml2_Constants.NS_PREFIX_XS
+            OneLogin_Saml2_Constants.NS_PREFIX_XS,
+            OneLogin_Saml2_Constants.NS_PREFIX_XSI,
+            OneLogin_Saml2_Constants.NS_PREFIX_XSD
         ]
 
         if keep_ns_prefixes:
