Create test for persistence of search terms

Author: Tracy-Walker

tests/address_bar_and_search/test_search_term_persists.py

@@ -0,0 +1,85 @@
+import pytest
+from selenium.webdriver import Firefox
+from selenium.webdriver.support import expected_conditions as EC
+
+from modules.browser_object import Navigation
+from modules.browser_object_tabbar import TabBar
+from modules.page_object import AboutConfig
+
+
+# Set search region
+@pytest.fixture()
+def add_prefs():
+    return [
+        ("browser.search.region", "US"),
+        ("browser.urlbar.showSearchTerms.enabled", True),
+        ("browser.urlbar.showSearchTerms.featureGate", True),
+    ]
+
+
+def test_search_term_persists(driver: Firefox):
+    """
+    C2153943 - Persist search term basic functionality
+    """
+
+    # Create objects
+    nav = Navigation(driver).open()
+    tab = TabBar(driver)
+
+    # Set values
+    search1 = "cheetah"
+    result1 = "https://www.google.com/search?client=firefox-b-1-d&q=cheetah"
+    search2 = "lion"
+    result2 = "https://www.google.com/search?client=firefox-b-1-d&q=lion"
+
+    def toggle_old_search_bar():
+        tab.new_tab_by_button()
+        window_handles = driver.window_handles
+        driver.switch_to.window(window_handles[-1])
+        ac = AboutConfig(driver)
+        pref = "browser.search.widget.inNavBar"
+        ac.toggle_true_false_config(pref)
+        nav.set_chrome_context()
+        x_icon = tab.get_element("tab-x-icon", multiple=True)
+        x_icon[1].click()
+        driver.switch_to.window(window_handles[0])
+
+    # Perform a search using the URL bar.
+    nav.search(search1)
+    nav.expect(EC.title_contains("Google Search"))
+    nav.set_chrome_context()
+    address_bar_text = nav.get_element("awesome-bar").get_attribute("value")
+    print("address bar value: ", address_bar_text)
+    assert search1 == address_bar_text
+
+    # Add the search bar to toolbar
+    toggle_old_search_bar()
+
+    # Search term should be replaced with full url
+    address_bar_text = nav.get_element("awesome-bar").get_attribute("value")
+    print("address bar value: ", address_bar_text)
+    assert result1 == address_bar_text
+    nav.clear_awesome_bar()
+
+    # Perform a new awesome bar search, full url should be present
+    # First, navigate away from Google
+    nav.set_content_context()
+    driver.get("about:robots")
+    # Then perform another search
+    nav.search(search2)
+    nav.expect(EC.title_contains("Google Search"))
+    nav.set_chrome_context()
+    address_bar_text = nav.get_element("awesome-bar").get_attribute("value")
+    print("address bar value: ", address_bar_text)
+    assert result2 == address_bar_text
+
+    # Disable the old search bar
+    toggle_old_search_bar()
+
+    # Again, perform a search using the URL bar.
+    nav.search(search1)
+    nav.expect(EC.title_contains("Google Search"))
+    nav.set_chrome_context()
+    address_bar_text = nav.get_element("awesome-bar").get_attribute("value")
+    print("address bar value: ", address_bar_text)
+    assert search1 == address_bar_text