Files
the-littlest-jupyterhub/integration-tests/test_extensions.py
2023-08-01 06:28:39 +00:00

48 lines
1.3 KiB
Python

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 = [
"jupyterlab",
"nbgitpuller 1.",
"jupyter_resource_usage",
]
for e in extensions:
assert e in proc.stderr.decode()
def test_nbextensions():
"""
Validate nbextensions we want are installed & enabled
"""
# jupyter-nbextension writes to stdout and stderr weirdly
proc = subprocess.run(
["/opt/tljh/user/bin/jupyter-nbextension", "list", "--sys-prefix"],
stderr=subprocess.PIPE,
stdout=subprocess.PIPE,
)
extensions = [
"jupyter_resource_usage/main",
# This is what ipywidgets nbextension is called
"jupyter-js-widgets/extension",
]
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
)