o
    ”^©i0  ã                   @   sV   d dl Z d dlZdd„ Zdd„ Zdd„ Zdd	„ Zd
d„ Zej dddg¡dd„ ƒZ	dS )é    Nc                 C   s.   dd l }| ¡ }|D ]
}t|d ƒsJ ‚q
d S )Nr   Úmodule)Újupyter_lspÚ_jupyter_server_extension_pathsÚ
__import__)Úappr   ÚpathsÚpath© r	   ú[/home/ubuntu/hpml_nyu/venv/lib/python3.10/site-packages/jupyter_lsp/tests/test_extension.pyÚtest_serverextension_path   s
   ÿr   c                 C   s\   |   dg¡ | jsJ ‚d}| jjjD ]}|jjD ]}dt|jjƒv r$d}qq|s,J dƒ‚d S )NúE--ServerApp.jpserver_extensions={'jupyter_lsp.serverextension': True}Fz/lsp/Tz)apparently didn't install the /lsp/ route)	Ú
initializeÚlanguage_server_managerÚweb_appÚdefault_routerÚrulesÚtargetÚstrÚmatcherÚregex)r   Ú	found_lspÚrÚrrr	   r	   r
   Útest_serverextension   s   ÿ
€þr   c                 C   s    |   dg¡ | jjdksJ ‚d S )Nr   z.virtual_documents©r   r   Úvirtual_documents_dir)r   r	   r	   r
   Ú"test_default_virtual_documents_dir   s   ÿr   c                 C   s*   d}|   dd| g¡ | jj|ksJ ‚d S )Nú.custom_virtual_dirr   z8--ServerApp.LanguageServerManager.virtual_documents_dir=r   )r   Ú
custom_dirr	   r	   r
   Ú!test_virtual_documents_dir_config#   s   þÿr   c                 C   sL   d t jd< }|  dg¡ | jj|ksJ ‚| jjd }|d  |¡s$J ‚d S )Nr   ÚJP_LSP_VIRTUAL_DIRr   Úpage_config_dataÚvirtualDocumentsUri)ÚosÚenvironr   r   r   r   ÚsettingsÚendswith)r   r   Úpage_configr	   r	   r
   Útest_virtual_documents_dir_env.   s   ÿr(   Úinvalid_pathÚ Ú.c                 C   s8   |t jd< |  dg¡ | jjd }|d  d¡sJ ‚d S )Nr    r   r!   r"   z/.virtual_documents)r#   r$   r   r   r%   r&   )r   r)   r'   r	   r	   r
   Ú$test_virtual_documents_dir_env_empty8   s   
ÿr,   )
r#   Úpytestr   r   r   r   r(   ÚmarkÚparametrizer,   r	   r	   r	   r
   Ú<module>   s    
