Modified save button selectors

jupyter · Jan 24, 2023 · 255046e · 255046e
1 parent 0a4d713
commit 255046e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/nbclassic/tests/end_to_end/test_save_as_notebook.py b/nbclassic/tests/end_to_end/test_save_as_notebook.py
@@ -30,12 +30,13 @@ def test_save_notebook_as(notebook_frontend):
     notebook_frontend.wait_for_selector('.save-message', page=EDITOR_PAGE)
 
     # TODO: Add a function for locator assertions to FrontendElement
-    locator_element = notebook_frontend.locate_and_focus('//input[@data-testid="save-as"]', page=EDITOR_PAGE)
+    dialog_element = notebook_frontend.locate_and_focus(".modal-footer", page=EDITOR_PAGE)
+    locator_element = dialog_element.locate('text=Save')
     locator_element.wait_for('visible')
 
     notebook_frontend.insert_text('new_notebook.ipynb', page=EDITOR_PAGE)
-    notebook_frontend.try_click_selector('//html//body//div[8]//div//div//div[3]//button[2]', page=EDITOR_PAGE)
-    
+    locator_element.click()
+
     locator_element.expect_not_to_be_visible()
 
     assert get_notebook_name(notebook_frontend) == "new_notebook.ipynb"