o
    i                     @   s\   d dl mZ d dlmZ d dlmZ d dlmZ g dZ	dZ
dZdZd	Zd
d Zdd ZdS )    )By)expected_conditions)WebDriverWait)wait_for_selector)z#restart_kernelz#restart_clear_outputz#restart_run_allz#notification_kernel > spanz#shutdown_kernelz.btn-dangerz".modal-footer button:first-of-typec                 C   s   | j }|tjd}ttg D ].}|  t||ddd  t|tddd  t	|d
ttjdf |  s=J qdS )zFClick each of the restart options, then cancel the confirmation dialog
kernellinkTvisiblesingle   .modal-backdropN)browserfind_elementr   IDrestart_selectorsshutdown_selectorclickr   cancel_selectorr   untilECinvisibility_of_elementCSS_SELECTORis_kernel_runningnotebookr   kernel_menu	menu_item r   c/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/tests/selenium/test_kernel_menu.pytest_cancel_restart_or_shutdown   s   
r   c                    s    j }|tjd}tD ]V}|  t|tddd  t|tddd  t	|dj
 fdddd t	|d
ttjd	f |  t||ddd  t	|d
j
 fddd|dd qd S )Nr   Tr   r
   c                    s
       S Nr   br   r   r   <lambda>,   s   
 z!test_menu_items.<locals>.<lambda>z$Kernel did not shut down as expected)messager   
   c                    s      S r   r    r!   r#   r   r   r$   9   s    z	Restart (z%) after shutdown did not start kernel)r   r   r   r   r   r   r   r   confirm_selectorr   r   r   r   r   r   r   r#   r   test_menu_items   s(   





r(   N)selenium.webdriver.common.byr   selenium.webdriver.supportr   r   selenium.webdriver.support.uir   notebook.tests.selenium.utilsr   r   notify_interactionr   r'   r   r   r(   r   r   r   r   <module>   s    