File size: 4,844 Bytes
d1ceb73 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 |
# Copyright (c) Jupyter Development Team.
# Distributed under the terms of the Modified BSD License.
"""pytest fixtures."""
from __future__ import annotations
import json
import os
import os.path as osp
import shutil
from os.path import join as pjoin
from pathlib import Path
from typing import Any, Callable
import pytest
from jupyter_server.serverapp import ServerApp
from jupyterlab_server import LabServerApp
pytest_plugins = ["pytest_jupyter.jupyter_server"]
def mkdir(tmp_path: Path, *parts: str) -> Path:
"""Util for making a directory."""
path = tmp_path.joinpath(*parts)
if not path.exists():
path.mkdir(parents=True)
return path
HERE = os.path.abspath(os.path.dirname(__file__))
app_settings_dir = pytest.fixture(lambda tmp_path: mkdir(tmp_path, "app_settings"))
user_settings_dir = pytest.fixture(lambda tmp_path: mkdir(tmp_path, "user_settings"))
schemas_dir = pytest.fixture(lambda tmp_path: mkdir(tmp_path, "schemas"))
workspaces_dir = pytest.fixture(lambda tmp_path: mkdir(tmp_path, "workspaces"))
labextensions_dir = pytest.fixture(lambda tmp_path: mkdir(tmp_path, "labextensions_dir"))
@pytest.fixture
def make_labserver_extension_app(
jp_root_dir: Path,
jp_template_dir: Path,
app_settings_dir: Path,
user_settings_dir: Path,
schemas_dir: Path,
workspaces_dir: Path,
labextensions_dir: Path,
) -> Callable[..., LabServerApp]:
"""Return a factory function for a labserver extension app."""
def _make_labserver_extension_app(**kwargs: Any) -> LabServerApp: # noqa: ARG001
"""Factory function for lab server extension apps."""
return LabServerApp(
static_dir=str(jp_root_dir),
templates_dir=str(jp_template_dir),
app_url="/lab",
app_settings_dir=str(app_settings_dir),
user_settings_dir=str(user_settings_dir),
schemas_dir=str(schemas_dir),
workspaces_dir=str(workspaces_dir),
extra_labextensions_path=[str(labextensions_dir)],
)
# Create the index files.
index = jp_template_dir.joinpath("index.html")
index.write_text(
"""
<!DOCTYPE html>
<html>
<head>
<title>{{page_config['appName'] | e}}</title>
</head>
<body>
{# Copy so we do not modify the page_config with updates. #}
{% set page_config_full = page_config.copy() %}
{# Set a dummy variable - we just want the side effect of the update. #}
{% set _ = page_config_full.update(baseUrl=base_url, wsUrl=ws_url) %}
<script id="jupyter-config-data" type="application/json">
{{ page_config_full | tojson }}
</script>
<script src="{{page_config['fullStaticUrl'] | e}}/bundle.js" main="index"></script>
<script type="text/javascript">
/* Remove token from URL. */
(function () {
var parsedUrl = new URL(window.location.href);
if (parsedUrl.searchParams.get('token')) {
parsedUrl.searchParams.delete('token');
window.history.replaceState({ }, '', parsedUrl.href);
}
})();
</script>
</body>
</html>
"""
)
# Copy the schema files.
src = pjoin(HERE, "test_data", "schemas", "@jupyterlab")
dst = pjoin(str(schemas_dir), "@jupyterlab")
if os.path.exists(dst):
shutil.rmtree(dst)
shutil.copytree(src, dst)
# Create the federated extensions
for name in ["apputils-extension", "codemirror-extension"]:
target_name = name + "-federated"
target = pjoin(str(labextensions_dir), "@jupyterlab", target_name)
src = pjoin(HERE, "test_data", "schemas", "@jupyterlab", name)
dst = pjoin(target, "schemas", "@jupyterlab", target_name)
if osp.exists(dst):
shutil.rmtree(dst)
shutil.copytree(src, dst)
with open(pjoin(target, "package.orig.json"), "w") as fid:
data = dict(name=target_name, jupyterlab=dict(extension=True))
json.dump(data, fid)
# Copy the overrides file.
src = pjoin(HERE, "test_data", "app-settings", "overrides.json")
dst = pjoin(str(app_settings_dir), "overrides.json")
if os.path.exists(dst):
os.remove(dst)
shutil.copyfile(src, dst)
# Copy workspaces.
ws_path = pjoin(HERE, "test_data", "workspaces")
for item in os.listdir(ws_path):
src = pjoin(ws_path, item)
dst = pjoin(str(workspaces_dir), item)
if os.path.exists(dst):
os.remove(dst)
shutil.copy(src, str(workspaces_dir))
return _make_labserver_extension_app
@pytest.fixture
def labserverapp(
jp_serverapp: ServerApp, make_labserver_extension_app: Callable[..., LabServerApp]
) -> LabServerApp:
"""A lab server app."""
app = make_labserver_extension_app()
app._link_jupyter_server_extension(jp_serverapp)
app.initialize() # type:ignore[no-untyped-call]
return app
|