ADd notebook impersonate login auth

Author: mmourafiq

polyaxon_client/api/auth.py

@@ -192,3 +192,19 @@ def login_job_impersonate_token(self,
                                              internal_token=internal_token,
                                              set_token=set_token,
                                              persist_token=persist_token)
+
+    def login_notebook_impersonate_token(self,
+                                         username,
+                                         project_name,
+                                         internal_token,
+                                         set_token=True,
+                                         persist_token=True):
+        request_url = self.build_url(self._get_http_url('/'),
+                                     username,
+                                     project_name,
+                                     'notebook',
+                                     'imporsonatetoken')
+        return self._login_impersonate_token(request_url=request_url,
+                                             internal_token=internal_token,
+                                             set_token=set_token,
+                                             persist_token=persist_token)

tests/test_api/test_auth.py

@@ -230,3 +230,45 @@ def test_login_job_impersonate_token(self):
             persist_token=True)
         assert self.api_config.token == token
         assert os.path.exists(settings.CONTEXT_AUTH_TOKEN_PATH) is True
+
+    @httpretty.activate
+    def test_login_notebook_impersonate_token(self):
+        token = uuid.uuid4().hex
+        httpretty.register_uri(
+            httpretty.POST,
+            BaseApiHandler.build_url(
+                self.api_config.base_url,
+                '/',
+                'user',
+                'project',
+                'notebook',
+                'imporsonatetoken'
+            ),
+            body=json.dumps({'token': token}),
+            content_type='application/json', status=200)
+
+        # Login without updating the token and without persistence
+        if os.path.exists(settings.CONTEXT_AUTH_TOKEN_PATH):
+            os.remove(settings.CONTEXT_AUTH_TOKEN_PATH)
+        assert self.api_config.token == 'token'
+        assert token == self.api_handler.login_notebook_impersonate_token(
+            username='user',
+            project_name='project',
+            internal_token='foo',
+            set_token=False,
+            persist_token=False)
+        assert self.api_config.token == 'token'
+        assert os.path.exists(settings.CONTEXT_AUTH_TOKEN_PATH) is False
+
+        # Login and update the token and persistence
+        if os.path.exists(settings.CONTEXT_AUTH_TOKEN_PATH):
+            os.remove(settings.CONTEXT_AUTH_TOKEN_PATH)
+        assert self.api_config.token == 'token'
+        assert token == self.api_handler.login_notebook_impersonate_token(
+            username='user',
+            project_name='project',
+            internal_token='foo',
+            set_token=True,
+            persist_token=True)
+        assert self.api_config.token == token
+        assert os.path.exists(settings.CONTEXT_AUTH_TOKEN_PATH) is True