Restore xml.etree.ElementTree after patch

Restore ``ElementTree`` attribute of ``xml.etree`` module after patching

Closes: https://github.com/tiran/defusedxml/issues/54
Co-authored-by: Marien Zwart <marienz@google.com>
Signed-off-by: Christian Heimes <christian@python.org>

1 files changed, 6 insertions, 0 deletions

diff --git a/tests.py b/tests.py
index f8b9274..89fb985 100644
--- a/tests.py
+++ b/tests.py
@@ -6,6 +6,7 @@ import sys
 import unittest
 import warnings
 
+from xml.etree import ElementTree as orig_elementtree
 from xml.sax.saxutils import XMLGenerator
 from xml.sax import SAXParseException
 from pyexpat import ExpatError
@@ -208,6 +209,11 @@ class TestDefusedElementTree(BaseTests):
         assert self.module.XMLParser is parser
         assert self.module.XMLParse is parser
 
+    def test_import_order(self):
+        from xml.etree import ElementTree as second_elementtree
+
+        self.assertIs(orig_elementtree, second_elementtree)
+
 
 class TestDefusedcElementTree(TestDefusedElementTree):
     module = cElementTree