Fixed imports

Author: Roland Hedberg

tests/test_13_validate.py

@@ -2,11 +2,9 @@
 # -*- coding: utf-8 -*-
 
 import saml2
-import xmldsig as ds
 
 from saml2 import saml
 from saml2 import samlp
-from saml2 import md
 from saml2.validate import valid_duration, MustValueError
 from saml2.validate import valid_unsigned_short
 from saml2.validate import valid_non_negative_integer

tests/test_20_assertion.py

@@ -26,8 +26,8 @@
 from saml2.extension import mdattr
 from saml2.extension import ui
 from saml2 import saml
-import xmldsig
-import xmlenc
+from saml2 import xmldsig
+from saml2 import xmlenc
 
 from pathutils import full_path
 

tests/test_22_mdie.py

@@ -10,8 +10,8 @@
 from saml2.extension import dri
 from saml2.extension import mdattr
 from saml2.extension import ui
-import xmldsig
-import xmlenc
+from saml2 import xmldsig
+from saml2 import xmlenc
 
 ONTS = {
     saml.NAMESPACE: saml,

tests/test_30_mdstore.py

@@ -26,8 +26,8 @@
 from saml2.extension import mdattr
 from saml2.extension import ui
 from saml2.s_utils import UnknownPrincipal
-import xmldsig
-import xmlenc
+from saml2 import xmldsig
+from saml2 import xmlenc
 
 from pathutils import full_path
 

tests/test_30_mdstore_old.py

@@ -2,8 +2,6 @@
 # -*- coding: utf-8 -*-
 import datetime
 import re
-from urllib import quote_plus
-from saml2.httpbase import HTTPBase
 
 from saml2.mdstore import MetadataStore, MetaDataMDX
 from saml2.mdstore import destinations
@@ -26,8 +24,8 @@
 from saml2.extension import mdattr
 from saml2.extension import ui
 from saml2.s_utils import UnknownPrincipal
-import xmldsig
-import xmlenc
+from saml2 import xmldsig
+from saml2 import xmlenc
 
 from pathutils import full_path
 

tests/test_37_entity_categories.py

@@ -12,8 +12,8 @@
 from pathutils import full_path
 from saml2.mdstore import MetadataStore
 from saml2.server import Server
-import xmldsig
-import xmlenc
+from saml2 import xmldsig
+from saml2 import xmlenc
 
 ONTS = {
     saml.NAMESPACE: saml,

tests/test_51_client.py

@@ -4,7 +4,7 @@
 import base64
 import urllib
 import urlparse
-from xmldsig import SIG_RSA_SHA256
+from saml2.xmldsig import SIG_RSA_SHA256
 from saml2 import BINDING_HTTP_POST
 from saml2 import BINDING_HTTP_REDIRECT
 from saml2 import config

tests/test_60_sp.py

@@ -5,7 +5,7 @@
 from saml2.authn_context import INTERNETPROTOCOLPASSWORD
 from saml2.saml import NAMEID_FORMAT_TRANSIENT
 from saml2.samlp import NameIDPolicy
-from s2repoze.plugins.sp import make_plugin
+from saml2.s2repoze.plugins.sp import make_plugin
 from saml2.server import Server
 
 ENV1 = {'SERVER_SOFTWARE': 'CherryPy/3.1.2 WSGI Server',

tests/test_76_metadata_in_mdb.py

@@ -20,8 +20,8 @@
 from saml2.extension import dri
 from saml2.extension import mdattr
 from saml2.extension import ui
-import xmldsig
-import xmlenc
+from saml2 import xmldsig
+from saml2 import xmlenc
 
 from pathutils import full_path
 

tests/test_82_pefim.py

@@ -1,4 +1,4 @@
-import xmldsig as ds
+from saml2 import xmldsig as ds
 from saml2 import config
 from saml2 import extension_elements_to_elements
 from saml2 import element_to_extension_element