o
i, @ 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_dirindex_filepathfid r T/var/www/edux/Edux_v2/venv/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)
extension_dirschema_package schema_paththemesr r r% r _create_schemas_dirE s
r6 c " t jdd t fdd S ),Create a temporary directory for workspaces.mock_user_settingsr c r r! r" r r% r r r' X r( z+_create_user_settings_dir..r r r0 r1 r r r% r _create_user_settings_dirU r; c r7 )r8 mock_workspacesr c r r! r" r r% r r r' _ r( z(_create_workspaces_dir..r: r r r% r _create_workspaces_dir\ r< r> 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 )
Njupyter jupyter_datajupyter_runtimeipython)JUPYTER_CONFIG_DIRJUPYTER_DATA_DIRJUPYTER_RUNTIME_DIR
IPYTHONDIRshareenvetc)SYSTEM_JUPYTER_PATHENV_JUPYTER_PATHSYSTEM_CONFIG_PATHENV_CONFIG_PATH)r test_dirr dictr) environpjoinname env_patchstartmultiple jupyter_corepaths
path_patch)selftdr r r rU i s$
z
TestEnv.startc C s: | j | j z| j W d S ty Y d S w N)rT stoprY rO cleanupOSErrorrZ r r r r] ~ s
z TestEnv.stopc C s | | jjS r\ )rU rO rS r` r r r __enter__ s zTestEnv.__enter__c G s | d S r\ )r] )rZ exc_infor r r __exit__ s zTestEnv.__exit__N)r,
__module__ __qualname____doc__rU r] ra rc r r r r r? c s r? 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 r\ )r static_pathsr template_pathsr` 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 echokernel_manager)r? rT rU r __init__rg rl settingsri r rj r r6 schemas_dirr; user_settings_dirr> workspaces_dir_install_default_kernelsdefault_kernel_namesuperinitialize_settingsr` __class__r r rw 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)
rW rX rR jupyter_data_dirr) r* r r jsondumps)rZ kernel_name kernel_specrX
kernel_dirfr r r _install_kernel s
"z ProcessTestApp._install_kernelc C sD | j dtjddddgdddd tj}t| d d
}t| d S ) Nrm z-mz jupyterlab.tests.echo_kernelz-fz{connection_file}z Echo Kernel)argv display_namelanguage)r~ r rz rC )r sys
executablerW rX rR r{ r
)rZ rX
ipykernel_dirr r r rt 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 ) serverapp http_serverr] io_looprT r) _exitresult Exceptionlogerrorstr)rZ futureer r r _process_finished s
z ProcessTestApp._process_finished) r, rd re rf rl rk rw r rt r
__classcell__r r rx r rg s rg 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. mock_rootr srcztemp.txtr helloNzreadonly-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 r0 r1 )rZ r readonly_filepathr r% r _default_root_dir s z!RootedServerApp._default_root_dirN)r, rd re r
r r r r r r s r )-rf r0 r| r) r# r r importlib.resourcesr ImportErrorimportlib_resourcesr r os.pathr rR statr r r r
unittest.mockr rW r+ ipykernel.kernelspecr
jupyter_server.serverappr jupyterlab_server.process_appr traitletsr
realpathdirname__file__HEREr r r6 r; r> r? rg r r r r r s@ +K