add tests for extension URLs with base_url

Author: Zsailer

jupyter_server/extension/application.py

@@ -1,16 +1,15 @@
 import sys
 import re
 import logging
+from urllib.parse import urljoin
 
 from jinja2 import Environment, FileSystemLoader
 
 from traitlets import (
     Unicode,
     List,
     Dict,
-    Bool,
-    default,
-    validate
+    default
 )
 from traitlets.config import Config
 from tornado.log import LogFormatter
@@ -191,8 +190,10 @@ def _default_log_format(self):
 
     @property
     def static_url_prefix(self):
-        return "/static/{name}/".format(
-            name=self.name)
+        static_url = "static/{name}".format(
+            name=self.name
+        )
+        return urljoin(self.serverapp.base_url, static_url)
 
     static_paths = List(Unicode(),
         help="""paths to search for serving static files.
@@ -288,6 +289,7 @@ def _prepare_handlers(self):
         # Add static endpoint for this extension, if static paths are given.
         if len(self.static_paths) > 0:
             # Append the extension's static directory to server handlers.
+            print(self.static_url_prefix)
             static_url = url_path_join(self.static_url_prefix, "(.*)")
 
             # Construct handler.

jupyter_server/extension/handler.py

@@ -1,3 +1,4 @@
+from urllib.parse import urljoin
 from jupyter_server.base.handlers import FileFindHandler
 
 
@@ -55,10 +56,10 @@ def base_url(self):
 
     @property
     def static_url_prefix(self):
-        return "{base_url}/static/{name}/".format(
-            name=self.name,
-            base_url=self.base_url
+        static_url = "static/{name}".format(
+            name=self.name
         )
+        return urljoin(self.serverapp.base_url, static_url)
 
     @property
     def static_path(self):

tests/extension/mockextensions/app.py

@@ -1,3 +1,4 @@
+import os
 from traitlets import Unicode, List
 
 from jupyter_server.base.handlers import JupyterHandler
@@ -10,6 +11,8 @@
     ExtensionHandlerJinjaMixin
 )
 
+STATIC_PATH = os.path.join(os.path.dirname(__file__), "static")
+
 
 class MockExtensionHandler(ExtensionHandlerMixin, JupyterHandler):
 
@@ -31,6 +34,7 @@ class MockExtensionApp(ExtensionAppJinjaMixin, ExtensionApp):
 
     name = 'mockextension'
     template_paths = List().tag(config=True)
+    static_paths = [STATIC_PATH]
     mock_trait = Unicode('mock trait', config=True)
     loaded = False
 

tests/extension/mockextensions/static/mock.txt

@@ -0,0 +1 @@
+mock static content

tests/extension/test_handler.py

@@ -72,3 +72,42 @@ async def test_handler_argv(fetch):
     )
     assert r.code == 200
     assert r.body.decode() == 'test mock trait'
+
+
+@pytest.mark.parametrize(
+    'server_config',
+    [
+        {
+            "ServerApp": {
+                "jpserver_extensions": {
+                    "tests.extension.mockextensions": True
+                },
+                # Move extension handlers behind a url prefix
+                "base_url": "test_prefix"
+            },
+            "MockExtensionApp": {
+                # Change a trait in the MockExtensionApp using
+                # the following config value.
+                "mock_trait": "test mock trait"
+            }
+        }
+    ]
+)
+async def test_base_url(fetch):
+    # Test that the extension's handlers were properly prefixed
+    r = await fetch(
+        'test_prefix', 'mock',
+        method='GET'
+    )
+    assert r.code == 200
+    assert r.body.decode() == 'test mock trait'
+
+    # Test that the static namespace was prefixed by base_url
+    r = await fetch(
+        'test_prefix',
+        'static', 'mockextension', 'mock.txt',
+        method='GET'
+    )
+    assert r.code == 200
+    body = r.body.decode()
+    assert "mock static content" in body