Refactor Enterprise class to use AuthenticationBase

jjcp · jjcp · commit 10f72ab0428d · 2015-12-14T00:51:31.000-03:00
diff --git a/auth0/v2/authentication/enterprise.py b/auth0/v2/authentication/enterprise.py
@@ -1,18 +1,17 @@
-import requests
+from .base import AuthenticationBase
 
 
-class Enterprise(object):
+class Enterprise(AuthenticationBase):
 
     def __init__(self, domain):
         self.domain = domain
 
-    def saml_login(self, client_id, connection):
-        """
-        """
+    def saml_metadata(self, client_id):
+        return self.get(url='https://%s/samlp/metadata/%s' % (self.domain,
+                                                              client_id))
 
-        return requests.get(
-            'https://%s/samlp/%s' % (self.domain, client_id),
-            params={'connection': connection}
-        )
+    def wsfed_metadata(self):
+        url = 'https://%s/wsfed/FederationMetadata' \
+              '/2007-06/FederationMetadata.xml'
 
-    def saml_metadata(self,
+        return self.get(url=url % self.domain)
