Add remote proxy test, which is not truly remote, but exercises that

class.

Author: ian-r-rose

tests/resources/jupyter_server_config.py

@@ -11,4 +11,8 @@
         'port': 54321,
     },
 }
+
+import sys
+sys.path.append('./tests/resources')
+c.NotebookApp.nbserver_extensions = { 'proxyextension': True }
 #c.Application.log_level = 'DEBUG'

tests/resources/proxyextension.py

@@ -0,0 +1,47 @@
+from jupyter_server_proxy.handlers import ProxyHandler
+from notebook.utils import url_path_join
+
+class NewHandler(ProxyHandler):
+    async def http_get(self):
+        return await self.proxy()
+
+    async def open(self):
+        host = '127.0.0.1'
+        port = 54321
+        return await super().proxy_open(host, port)
+
+    def post(self):
+        return self.proxy()
+
+    def put(self):
+        return self.proxy()
+
+    def delete(self):
+        return self.proxy()
+
+    def head(self):
+        return self.proxy()
+
+    def patch(self):
+        return self.proxy()
+
+    def options(self):
+        return self.proxy()
+
+    def proxy(self):
+        host = '127.0.0.1'
+        port = 54321
+        proxied_path = ''
+        return super().proxy(host, port, proxied_path)
+
+
+def _jupyter_server_extension_paths():
+    return [{"module": "dask_labextension"}]
+
+
+def load_jupyter_server_extension(nb_server_app):
+    web_app = nb_server_app.web_app
+    base_url = web_app.settings["base_url"]
+    proxy_path = url_path_join(base_url, "newproxy/" + "?")
+    handlers = [(proxy_path, NewHandler)]
+    web_app.add_handlers(".*$", handlers)

tests/test_proxies.py

@@ -5,8 +5,8 @@
 TOKEN = os.getenv('JUPYTER_TOKEN', 'secret')
 
 
-def request_get(port, path, token):
-    h = HTTPConnection('localhost', port, 10)
+def request_get(port, path, token, host='localhost'):
+    h = HTTPConnection(host, port, 10)
     h.request('GET', '{}?token={}'.format(path, token))
     return h.getresponse()
 
@@ -57,3 +57,7 @@ def test_server_proxy_port_absolute():
     assert s.startswith('GET /proxy/absolute/54321/nmo?token=')
     assert 'X-Forwarded-Context' not in s
     assert 'X-Proxycontextpath' not in s
+
+def test_server_proxy_remote():
+    r = request_get(PORT, '/newproxy', TOKEN, host='127.0.0.1')
+    assert r.code == 200