From 25dfdc0a694299fe2a37f6edfb2dc39cb6c57e10 Mon Sep 17 00:00:00 2001
From: Jeff Tchang <jeff.tchang@reciprocitylabs.com>
Date: Fri, 26 Jul 2019 12:16:14 -0700
Subject: [PATCH] Don't clean xsd and xsi namespaces

---
 src/onelogin/saml2/constants.py | 1 +
 src/onelogin/saml2/xml_utils.py | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/onelogin/saml2/constants.py b/src/onelogin/saml2/constants.py
index 7b3f0085..e0778bd2 100644
--- 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
index c7274298..d966e615 100644
--- 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: