o
tf, @ sR d Z ddlZddlZddlZddlZddlZddlZzddlmZ W n e y1 ddl
mZ Y nw ddlm Z ddl
mZ ddlmZmZmZ ddlmZ ddlmZ ddlZddlZdd lmZ dd
lm Z dd l m Z dd l m Z e !e "e#Z$d
d Z%dd Z&dd Z'dd Z(dd Z)G dd dZ*G dd de Z+G dd d e Z,dS ) z7A lab app that runs a sub process for a demo or a test. N)files)path)join)S_IRGRPS_IROTHS_IRUSR)TemporaryDirectory)patch)write_kernel_spec) ServerApp)
ProcessApp)defaultc C sR t jdd} t| d}t|d}|d W d | S 1 s"w Y | S )N mock_staticprefixz
index.htmlwa
{% block title %}Jupyter Lab Test{% endblock %}
{% block meta %}
{% endblock %}
JupyterLab Test Application
{% block site %}
{% endblock site %}
{% block after_site %}
{% endblock after_site %}
)tempfilemkdtempospr openwrite) template_dirZindex_filepathfid r b/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/jupyterlab/tests/test_app.py_create_template_dir" s
r c C s t jdd} | S )Nr r )r r )
static_dirr r r _create_static_dir@ s r c s t jdd t dd} t| tj}d}t| |
}t t| dd}| |
d W d n1 sQ z%_create_schemas_dir..)r r r r osmakedirsjupyterlab_server__name__r joinpath
read_bytesr r decodeatexitregister)Z
extension_dirZschema_packageZ schema_paththemesr r r# r _create_schemas_dirE s
r1 c " t jdd t fdd S ),Create a temporary directory for workspaces.Zmock_user_settingsr c r r r r r# r r r% X r& z+_create_user_settings_dir..r r r. r/ r r r# r _create_user_settings_dirU r5 c r2 )r3 Zmock_workspacesr c r r r r r# r r r% _ r& z(_create_workspaces_dir..r4 r r r# r _create_workspaces_dir\ r6 r7 c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) TestEnvzuSet Jupyter path variables to a temporary directory
Useful as a context manager or with explicit start/stop
c
C s t | _}ttjt|jdt|jdt|jdt|jdd| _| j tj
t j t|jddgt|jdddgt|jddgt|jdddgd | _
| j
d S )
NjupyterZ jupyter_dataZjupyter_runtimeipython)ZJUPYTER_CONFIG_DIRZJUPYTER_DATA_DIRZJUPYTER_RUNTIME_DIRZ
IPYTHONDIRshareenvetc)ZSYSTEM_JUPYTER_PATHZENV_JUPYTER_PATHZSYSTEM_CONFIG_PATHZENV_CONFIG_PATH)r test_dirr dictr' environpjoinname env_patchstartZmultiple jupyter_corepaths
path_patch)selftdr r r rD i s$
z
TestEnv.startc C s: | j | j z| j W d S ty Y d S w N)rC stoprG r> cleanupOSErrorrH r r r rK ~ s
z TestEnv.stopc C s | | jjS rJ )rD r> rB rN r r r __enter__ s zTestEnv.__enter__c G s | d S rJ )rK )rH exc_infor r r __exit__ s zTestEnv.__exit__N)r*
__module__ __qualname____doc__rD rK rO rQ r r r r r8 c s r8 c sD e Zd ZdZdZdd Z fddZdd Zd d
Zd d Z Z
S )
ProcessTestAppzDA process app for running tests, includes a mock contents directory.*c C s t g| _t g| _d S rJ )r static_pathsr template_pathsrN r r r initialize_templates s
z#ProcessTestApp.initialize_templatesc sz t | _| j t| tj| jd< | jd | _ | j
d | _ t | _
t | _t | _| d| jd _t d S )N allow_originr echoZkernel_manager)r8 rC rD r __init__rU rZ settingsrW r rX r r1 Z schemas_dirr5 Zuser_settings_dirr7 Zworkspaces_dir_install_default_kernelsZdefault_kernel_namesuperinitialize_settingsrN __class__r r r` s
z"ProcessTestApp.initialize_settingsc C sf t j}t| d|}t| tt|dd}|t | W d dS 1 s,w Y dS )zInstall a kernel spec to the data directory.
Parameters
----------
kernel_name: str
Name of the kernel.
kernel_spec: dict
The kernel spec for the kernel
kernelsz kernel.jsonr N)
rE rF rA jupyter_data_dirr' r( r r jsondumps)rH kernel_name kernel_specrF Z
kernel_dirfr r r _install_kernel s
"z ProcessTestApp._install_kernelc C sD | j dtjddddgdddd tj}t| d d
}t| d S ) Nr[ z-mz jupyterlab.tests.echo_kernelz-fz{connection_file}z Echo Kernel)argvZ display_namelanguage)rg rh rc r: )rj sys
executablerE rF rA rd r
)rH rF Z
ipykernel_dirr r r r^ s z'ProcessTestApp._install_default_kernelsc
C sx | j j | j j | j z
t| W d S ty; } z| j
t | td W Y d }~d S d }~ww )N ) Z serverappZ http_serverrK Zio_looprC r' _exitresult Exceptionlogerrorstr)rH futureer r r _process_finished s
z ProcessTestApp._process_finished) r* rR rS rT rZ rY r` rj r^ rx
__classcell__r r ra r rU s rU c @ s e Zd Zeddd ZdS )RootedServerAppr$ c s t jdd tt d tt ddd
}|d W d n1 s)w Y t dd}t|d
}|d W d n1 sJw Y t|t t
B t B t
fd
d S ) z6Create a temporary directory with some file structure.Z mock_rootr srcztemp.txtr ZhelloNzreadonly-temp.txtzhello from a readonly filec r r r r r# r r r% r& z3RootedServerApp._default_root_dir..)r r r' mkdirr r r r chmodr r r r. r/ )rH r Zreadonly_filepathr r# r _default_root_dir s z!RootedServerApp._default_root_dirN)r* rR rS r
r~ r r r r rz s rz )-rT r. re r' r! rm r importlib.resourcesr ImportErrorZimportlib_resourcesr r Zos.pathr rA statr r r r Z
unittest.mockr rE r) Zipykernel.kernelspecr
Zjupyter_server.serverappr Z jupyterlab_server.process_appr Z traitletsr
realpathdirname__file__ZHEREr r r1 r5 r7 r8 rU rz r r r r s@ +K