a
    E¿bd  ã                   @   sR   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)ÚKeys)ÚWebDriverWaité
   c                    s    t | |ƒ}‡ fdd„}| |¡S )Nc                    s   |   d¡}|  d|¡}|ˆ kS )NZnotebook_namezreturn arguments[0].innerText)Zfind_element_by_idÚexecute_script)ÚbrowserÚelemZcurrent_name©Únbname© úLlib/python3.9/site-packages/notebook/tests/selenium/test_save_as_notebook.pyÚnotebook_renamed   s    
z)wait_for_rename.<locals>.notebook_renamed)r   Zuntil)r   r
   ZtimeoutÚwaitr   r   r	   r   Úwait_for_rename   s    
r   c                 C   s   d}| j  |¡S )Nz#Jupyter.notebook.save_notebook_as()©r   r   ©ÚnbÚJSr   r   r   Úsave_as   s    r   c                 C   s   d}| j  |¡S )Nz%return Jupyter.notebook.notebook_namer   r   r   r   r   Úget_notebook_name   s    r   c                 C   s   d  |¡}| j |¡ d S )NzJupyter.notebook.rename("{}"))Úformatr   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 d¡}| d¡ | tj	¡ t| jdƒ t| ƒdkszJ ‚d| jj
v sŠJ ‚d S )Nz	nb1.ipynb)r   Znb1z.save-messagez//input[@data-testid="save-as"]znew_notebook.ipynbZnew_notebook)r   r   r   r   r   r   Zfind_element_by_xpathZ	send_keysr   ZRETURNZcurrent_url)ZnotebookZinpr   r   r   Útest_save_notebook_as   s    
r   N)r   )Znotebook.tests.selenium.utilsr   Zselenium.webdriver.common.keysr   Zselenium.webdriver.support.uir   r   r   r   r   r   r   r   r   r   Ú<module>   s   
