Added login_url method to SAMLAuthenticator

Author: wesuuu

jupyterhub_saml_auth/authenticator.py

@@ -1,4 +1,5 @@
 from jupyterhub.auth import Authenticator
+from jupyterhub.utils import url_path_join
 from traitlets import Unicode, validate, Set, Callable
 import os
 
@@ -73,6 +74,9 @@ def _settings_files_exist(self, path):
                     in saml settings path. Path = {path}, \
                     contents = {dir_contents}')
 
+    def login_url(self, base_url):
+        return url_path_join(base_url, 'saml_login')
+
     def authenticate(self, handler, data):
         return data['name']
 

test/unit/test_authenticator.py

@@ -0,0 +1,6 @@
+from jupyterhub_saml_auth.authenticator import SAMLAuthenticator
+
+
+def test_login_url_sets():
+    samlauth = SAMLAuthenticator()
+    assert '/hub/saml_login' == samlauth.login_url('/hub')