o
    i                     @   s^   d dl mZ d dlmZ d dlmZ d dlmZ dddZdd	 Z	d
d Z
dd Zdd ZdS )    )wait_for_selector)By)Keys)WebDriverWait
   c                    s    t | |} fdd}||S )Nc                    s"   |  tjd}| d|}| kS )Nnotebook_namezreturn arguments[0].innerText)find_elementr   IDexecute_script)browserelemcurrent_namenbname h/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/tests/selenium/test_save_as_notebook.pynotebook_renamed	   s   z)wait_for_rename.<locals>.notebook_renamed)r   until)r   r   timeoutwaitr   r   r   r   wait_for_rename   s   

r   c                 C      d}| j |S )Nz#Jupyter.notebook.save_notebook_as()r   r
   nbJSr   r   r   save_as      r   c                 C   r   )Nz%return Jupyter.notebook.notebook_namer   r   r   r   r   get_notebook_name   r   r   c                 C   s   d| d}| j | d S )NzJupyter.notebook.rename("z")r   )r   namer   r   r   r   set_notebook_name   s   r    c                 C   s   t | dd t| jd t| dksJ t|  t| jd | jtjd}|	d |	t
j t| jd t| dks?J d| jjv sGJ d S )Nz	nb1.ipynb)r   nb1z.save-messagez//input[@data-testid="save-as"]znew_notebook.ipynbnew_notebook)r    r   r   r   r   r   r   r   XPATH	send_keysr   RETURNcurrent_url)notebookinpr   r   r   test_save_notebook_as   s   
r)   N)r   )notebook.tests.selenium.utilsr   selenium.webdriver.common.byr   selenium.webdriver.common.keysr   selenium.webdriver.support.uir   r   r   r   r    r)   r   r   r   r   <module>   s    
