Anca/Add search shortcut engine mix (#903)

* Add search shortcut engine mix

* Update id and test name and method

* Use the correct method name post merge main

Author: soncuteanca

modules/browser_object_navigation.py

@@ -2,6 +2,7 @@
 import re
 import time
 from typing import Literal
+from urllib.parse import urlparse
 
 from selenium.common.exceptions import StaleElementReferenceException, TimeoutException
 from selenium.webdriver import ActionChains, Firefox
@@ -407,6 +408,28 @@ def verify_search_mode_is_not_visible(self):
         self.element_not_visible("search-mode-chicklet")
         return self
 
+    @BasePage.context_chrome
+    def verify_search_mode_label(self, engine_name: str):
+        """Verify that the search mode chicklet displays the correct engine."""
+        chicklet = self.get_element("search-mode-chicklet")
+        chip_text = (
+            chicklet.text or chicklet.get_attribute("aria-label") or ""
+        ).lower()
+        assert engine_name.lower() in chip_text, (
+            f"Expected search mode engine '{engine_name}', got '{chip_text}'"
+        )
+        return self
+
+    @BasePage.context_chrome
+    def verify_plain_text_in_input_awesome_bar(self, expected_text: str):
+        """Verify the awesomebar input contains the exact literal text."""
+        input_el = self.get_element("awesome-bar")
+        value = input_el.get_attribute("value")
+        assert value == expected_text, (
+            f"Expected input '{expected_text}', got '{value}'"
+        )
+        return self
+
     def click_first_suggestion_row(self):
         """
         Clicks the first visible suggestion row in the list, using robust scrolling and fallback.
@@ -920,6 +943,27 @@ def verify_status_panel_url(self, expected_url: str):
             f"Expected '{expected_url}' in status panel URL, got '{actual_url}'"
         )
 
+    @BasePage.context_content
+    def verify_domain(self, expected_domain: str) -> None:
+        """
+        Verify that the current URL's domain matches the expected domain using urlparse.
+        This explicitly checks the domain (netloc) rather than just a substring match.
+        Uses content context to get the actual page URL.
+
+        Argument:
+            expected_domain: The expected domain (e.g., "wikipedia.org", "google.com")
+        """
+
+        def _domain_matches(_):
+            parsed = urlparse(self.driver.current_url)
+            return expected_domain in parsed.netloc
+
+        self.custom_wait(timeout=15).until(_domain_matches)
+        parsed_url = urlparse(self.driver.current_url)
+        assert expected_domain in parsed_url.netloc, (
+            f"Expected '{expected_domain}' domain, got '{parsed_url.netloc}'"
+        )
+
     @BasePage.context_chrome
     def verify_engine_returned(self, engine: str) -> None:
         """

tests/address_bar_and_search/test_use_different_search_shortcut_while_already_in_search_mode.py

@@ -0,0 +1,48 @@
+import pytest
+from selenium.webdriver import Firefox
+from selenium.webdriver.common.keys import Keys
+
+from modules.browser_object import Navigation
+
+SEARCH_ENGINE = "Wikipedia"
+SEARCH_TERM = "@amazon"
+URL_DOMAIN = "wikipedia.org"
+
+
+@pytest.fixture()
+def test_case():
+    return "3028851"
+
+
+def test_use_search_shortcut_for_a_different_search_engine_while_already_in_search_mode(
+    driver: Firefox,
+):
+    """
+    C3028851 - Verify that using a search shortcut for a different search engine while in search mode
+    for another engine works as expected.
+    """
+    # Instantiate objects
+    nav = Navigation(driver)
+
+    # Enter search mode for the desired search engine in a new tab
+    nav.open_and_switch_to_new_window("tab")
+    nav.open_usb_and_select_option(SEARCH_ENGINE)
+
+    # Verify search mode is entered for the corresponding engine
+    nav.verify_search_mode_is_visible()
+
+    # Type @engine shortcut
+    nav.type_in_awesome_bar(SEARCH_TERM)
+
+    # Verify the search mode label matches the selected search engine and the input contains the plain search term
+    nav.verify_search_mode_label(SEARCH_ENGINE)
+    nav.verify_plain_text_in_input_awesome_bar(SEARCH_TERM)
+
+    # Perform the search by pressing Enter and expect
+    nav.type_in_awesome_bar(Keys.ENTER)
+
+    # Verify the domain is explicitly wikipedia.org
+    nav.verify_domain(URL_DOMAIN)
+
+    # Verify search term is also present in the URL
+    nav.url_contains(SEARCH_TERM.lstrip("@"))