Files
the-littlest-jupyterhub/integration-tests/test_extensions.py

48 lines
1.3 KiB
Python
Raw Normal View History

import subprocess
def test_serverextensions():
"""
Validate serverextensions we want are installed
"""
# jupyter-serverextension writes to stdout and stderr weirdly
proc = subprocess.run(
["/opt/tljh/user/bin/jupyter-serverextension", "list", "--sys-prefix"],
stderr=subprocess.PIPE,
)
extensions = [
2023-08-01 06:28:39 +00:00
"jupyterlab",
"nbgitpuller 1.",
"jupyter_resource_usage",
]
for e in extensions:
2020-02-03 11:07:49 +02:00
assert e in proc.stderr.decode()
2023-08-01 06:43:49 +00:00
def test_labextensions():
"""
2023-08-01 06:43:49 +00:00
Validate JupyterLab extensions we want are installed & enabled
"""
2023-08-01 06:43:49 +00:00
# jupyter-labextension writes to stdout and stderr weirdly
proc = subprocess.run(
2023-08-01 06:43:49 +00:00
["/opt/tljh/user/bin/jupyter-labextension", "list"],
stderr=subprocess.PIPE,
stdout=subprocess.PIPE,
)
extensions = [
2023-08-01 06:43:49 +00:00
"@jupyter-server/resource-usage",
# This is what ipywidgets lab extension is called
"@jupyter-widgets/jupyterlab-manager",
]
for e in extensions:
assert f"{e} \x1b[32m enabled \x1b[0m" in proc.stdout.decode()
# Ensure we have 'OK' messages in our stdout, to make sure everything is importable
assert proc.stderr.decode() == " - Validating: \x1b[32mOK\x1b[0m\n" * len(
extensions
)