Bookmark Selector Fix (#353)

mozilla · Feb 20, 2025 · f411eb8 · f411eb8
1 parent a07fe1d
commit f411eb8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/foxpuppet/windows/browser/bookmarks/bookmark.py b/foxpuppet/windows/browser/bookmarks/bookmark.py
@@ -67,7 +67,7 @@ def add_bookmark(
         """
         with self.selenium.context(self.selenium.CONTEXT_CHROME):
             if not is_detailed:
-                self.selenium.find_element(*BookmarkLocators.STAR_BUTTON).click()
+                self.selenium.find_element(*BookmarkLocators.STAR_BUTTON_IMAGE).click()
                 self.selenium.find_element(*BookmarkLocators.FOLDER_MENU).click()
                 self.selenium.find_element(*BookmarkLocators.OTHER_BOOKMARKS_STAR).click()
                 self.selenium.find_element(*BookmarkLocators.SAVE_BUTTON).click()
@@ -169,7 +169,9 @@ def delete_bookmark(
                     star_button_image
                     and star_button_image.get_attribute("starred") == "true"
                 ):
-                    self.selenium.find_element(*BookmarkLocators.STAR_BUTTON).click()
+                    self.selenium.find_element(
+                        *BookmarkLocators.STAR_BUTTON_IMAGE
+                    ).click()
                     self.selenium.find_element(*BookmarkLocators.REMOVE_BUTTON).click()
                 return
             self.actions.context_click(