Merge pull request #451 from mozilla/philimon/addon_suggestion

[Philimon] Fix for addon suggestion test

Author: Tracy-Walker

modules/browser_object_navigation.py

@@ -330,3 +330,8 @@ def open_searchmode_switcher_settings(self):
         self.click_on("searchmode-switcher")
         self.click_on("searchmode-switcher-settings")
         return self
+
+    def select_element_in_nav(self, element: str) -> BasePage:
+        with self.driver.context(self.driver.CONTEXT_CHROME):
+            self.get_element(element).click()
+            return self

modules/data/navigation.components.json

@@ -150,8 +150,8 @@
     },
 
     "addon-suggestion": {
-        "selectorData": "div.urlbarView-row[type='rust_amo'] span.urlbarView-title.urlbarView-overflowable",
-        "strategy": "css",
+        "selectorData": "urlbarView-row-body",
+        "strategy": "class",
         "groups": []
     },
 

tests/address_bar_and_search/test_addon_suggestion.py

@@ -35,20 +35,14 @@ def test_addon_suggestion_based_on_search_input(driver: Firefox):
 
     nav = Navigation(driver)
     nav.set_awesome_bar()
-    sleep(20)
+    sleep(2)
 
-    with driver.context(driver.CONTEXT_CHROME):
-        nav.awesome_bar.click()
+    for input_text, addon_name in input_to_addon_name.items():
+        nav.type_in_awesome_bar(input_text)
+        nav.element_visible("addon-suggestion")
+        nav.select_element_in_nav("addon-suggestion")
 
-        for input_text, addon_name in input_to_addon_name.items():
-            nav.awesome_bar.send_keys(input_text)
-            nav.element_visible("addon-suggestion")
-            nav.get_element("addon-suggestion").click()
-
-            # Construct the expected URL
-            expected_url = (
-                f"https://addons.mozilla.org/en-US/firefox/addon/{addon_name}/"
-            )
-            nav.expect_in_content(EC.url_contains(expected_url))
-
-            nav.awesome_bar.clear()
+        # Construct the expected URL
+        expected_url = f"https://addons.mozilla.org/en-US/firefox/addon/{addon_name}/"
+        nav.expect_in_content(EC.url_contains(expected_url))
+        nav.clear_awesome_bar()