Add integration test for user code execution with custom base_url

integration-tests/test_hub.py

@@ -18,6 +18,7 @@ from tljh.normalize import generate_system_username
 # This catches issues with PATH
 TLJH_CONFIG_PATH = ['sudo', 'tljh-config']
 
+
 def test_hub_up():
     r = requests.get('http://127.0.0.1')
     r.raise_for_status()
@@ -46,6 +47,34 @@ async def test_user_code_execute():
         assert pwd.getpwnam(f'jupyter-{username}') is not None
 
 
+@pytest.mark.asyncio
+async def test_user_code_execute_with_custom_base_url():
+    """
+    User logs in, starts a server with a custom base_url & executes code
+    """
+    # This *must* be localhost, not an IP
+    # aiohttp throws away cookies if we are connecting to an IP!
+    hub_url = 'http://localhost'
+    username = secrets.token_hex(8)
+
+    assert 0 == await (await asyncio.create_subprocess_exec(*TLJH_CONFIG_PATH, 'set', 'auth.type', 'dummyauthenticator.DummyAuthenticator')).wait()
+    assert 0 == await (await asyncio.create_subprocess_exec(*TLJH_CONFIG_PATH, 'set', 'base_url', '/custom-base')).wait()
+    assert 0 == await (await asyncio.create_subprocess_exec(*TLJH_CONFIG_PATH, 'reload')).wait()
+
+    async with User(username, hub_url, partial(login_dummy, password='')) as u:
+        await u.login()
+        await u.ensure_server_simulate()
+        await u.start_kernel()
+        await u.assert_code_output("5 * 4", "20", 5, 5)
+
+        # Assert that the user exists
+        assert pwd.getpwnam(f'jupyter-{username}') is not None
+
+        # unset base_url to avoid problems with other tests
+        assert 0 == await (await asyncio.create_subprocess_exec(*TLJH_CONFIG_PATH, 'unset', 'base_url')).wait()
+        assert 0 == await (await asyncio.create_subprocess_exec(*TLJH_CONFIG_PATH, 'reload')).wait()
+
+
 @pytest.mark.asyncio
 async def test_user_admin_add():
     """
@@ -68,7 +97,8 @@ async def test_user_admin_add():
         assert pwd.getpwnam(f'jupyter-{username}') is not None
 
         # Assert that the user has admin rights
-            assert f'jupyter-{username}' in grp.getgrnam('jupyterhub-admins').gr_mem
+        assert f'jupyter-{username}' in grp.getgrnam(
+            'jupyterhub-admins').gr_mem
 
 
 # FIXME: Make this test pass
@@ -97,7 +127,8 @@ async def test_user_admin_remove():
         assert pwd.getpwnam(f'jupyter-{username}') is not None
 
         # Assert that the user has admin rights
-            assert f'jupyter-{username}' in grp.getgrnam('jupyterhub-admins').gr_mem
+        assert f'jupyter-{username}' in grp.getgrnam(
+            'jupyterhub-admins').gr_mem
 
         assert 0 == await (await asyncio.create_subprocess_exec(*TLJH_CONFIG_PATH, 'remove-item', 'users.admin', username)).wait()
         assert 0 == await (await asyncio.create_subprocess_exec(*TLJH_CONFIG_PATH, 'reload')).wait()
@@ -106,7 +137,8 @@ async def test_user_admin_remove():
         await u.ensure_server_simulate()
 
         # Assert that the user does *not* have admin rights
-            assert f'jupyter-{username}' not in grp.getgrnam('jupyterhub-admins').gr_mem
+        assert f'jupyter-{username}' not in grp.getgrnam(
+            'jupyterhub-admins').gr_mem
 
 
 @pytest.mark.asyncio
@@ -229,6 +261,7 @@ async def test_idle_server_culled():
             timeout=100,
         )
 
+
 @pytest.mark.asyncio
 async def test_active_server_not_culled():
     """