Unify opening browser windows.

Author: Tracy-Walker

l10n_CM/Unified/test_demo_ad_email_phone_captured_in_doorhanger_and_stored.py

@@ -42,7 +42,9 @@ def test_demo_ad_email_phone_captured_in_doorhanger_and_stored(
     if expected_phone:
         with driver.context(driver.CONTEXT_CHROME):
             actual_phone = autofill_popup.get_element("address-doorhanger-phone").text
-        normalize_expected = util.normalize_regional_phone_numbers(expected_phone, region)
+        normalize_expected = util.normalize_regional_phone_numbers(
+            expected_phone, region
+        )
         normalized_actual = util.normalize_regional_phone_numbers(actual_phone, region)
         assert normalized_actual == normalize_expected, (
             f"Phone number mismatch for {region} | Expected: {normalize_expected}, Got: {normalized_actual}"

modules/page_base.py

@@ -3,6 +3,7 @@
 import os
 import platform
 import re
+import sys
 import time
 from copy import deepcopy
 from pathlib import Path
@@ -678,20 +679,55 @@ def wait_for_num_tabs(self, num_tabs: int) -> Page:
 
     def switch_to_new_tab(self) -> Page:
         """Get list of all window handles, switch to the newly opened tab"""
-        self.driver.switch_to.window(self.driver.window_handles[-1])
+        with self.driver.context(self.driver.CONTEXT_CONTENT):
+            self.driver.switch_to.window(self.driver.window_handles[-1])
         return self
 
     def switch_to_new_window(self) -> Page:
         """Switch to the most recently opened window. Can be a standard or private window"""
-        all_window_handles = self.driver.window_handles
-        self.driver.switch_to.window(all_window_handles[-1])
+        with self.driver.context(self.driver.CONTEXT_CONTENT):
+            all_window_handles = self.driver.window_handles
+            self.driver.switch_to.window(all_window_handles[-1])
         return self
 
     def wait_for_num_windows(self, num: int) -> Page:
         """Wait for the number of open tabs + windows to equal given int"""
         with self.driver.context(self.driver.CONTEXT_CONTENT):
             return self.wait_for_num_tabs(num)
 
+    def open_and_switch_to_new_window(self, browser_window: str) -> Page:
+        """
+        Opens a new browser window of the given type, then switches to it.
+
+        Parameters:
+            browser_window: Can be a standard 'window', 'tab' or 'private' browser window.
+        """
+        if browser_window == "private":
+            self.open_and_switch_to_private_window()
+        else:
+            self.driver.switch_to.new_window(browser_window)
+        return self
+
+    def open_and_switch_to_private_window(self) -> Page:
+        """
+        Opens a new private browsing window via keyboard shortcut and switch to it
+        """
+        # Keep track of window count to ensure we get a new one to switch to
+        window_count = len(self.driver.window_handles)
+
+        with self.driver.context(self.driver.CONTEXT_CHROME):
+            os_name = sys.platform
+            mod_key = Keys.COMMAND if os_name == "darwin" else Keys.CONTROL
+            self.actions.key_down(mod_key)
+            self.actions.key_down(Keys.SHIFT)
+            self.actions.send_keys("p")
+            self.actions.key_up(Keys.SHIFT)
+            self.actions.key_up(mod_key).perform()
+        expected_window_count = window_count + 1
+        self.wait_for_num_windows(expected_window_count)
+        self.switch_to_new_window()
+        return self
+
     def switch_to_frame(self, frame: str, labels=[]) -> Page:
         """Switch to inline document frame"""
         with self.driver.context(self.driver.CONTEXT_CHROME):

modules/util.py

@@ -499,15 +499,17 @@ def normalize_regional_phone_numbers(self, phone: str, region: str) -> str:
         digits = re.sub(r"\D", "", phone)
 
         # Determine country code
-        country_code = country_codes.get(region, "1")  # Default to "1" (US/CA) if region is unknown
+        country_code = country_codes.get(
+            region, "1"
+        )  # Default to "1" (US/CA) if region is unknown
         local_number = digits
 
         # Check if phone already contains a valid country code
         for code in country_codes.values():
             if digits.startswith(code):
                 country_code = code
                 # Remove country code from local number
-                local_number = digits[len(code):]
+                local_number = digits[len(code) :]
                 break
 
         # Handle leading zero in local numbers (France & Germany)

tests/bookmarks_and_history/test_opened_website_in_new_window_present_in_toolbar_history.py

@@ -27,13 +27,8 @@ def test_the_website_opened_in_new_window_is_present_in_history_menu(driver: Fir
     for url in WEBSITES:
         GenericPage(driver, url=url).open()
 
-    tabs = TabBar(driver)
     panel_ui = PanelUi(driver)
-
-    panel_ui.open_panel_menu()
-    panel_ui.select_panel_setting("panel-ui-new-window")
-    tabs.wait_for_num_tabs(2)
-    tabs.switch_to_new_window()
+    panel_ui.open_and_switch_to_new_window("window")
 
     page = GenericPage(driver, url=YOUTUBE_URL)
     page.open()

tests/bookmarks_and_history/test_private_window_website_not_in_history.py

@@ -19,8 +19,7 @@ def test_opened_website_in_private_window_not_captured_in_history_list(driver: F
     """
 
     panel_ui = PanelUi(driver)
-    panel_ui.open_private_window()
-    panel_ui.switch_to_new_window()
+    panel_ui.open_and_switch_to_new_window("private")
 
     GenericPage(driver, url=YOUTUBE_URL).open()
 

tests/networking/test_http_site.py

@@ -31,13 +31,18 @@ def set_prefs():
 def test_http_site(driver: Firefox):
     """C2300294 Check that HTTP is allowed when appropriate"""
 
+    # Instantiate objects
+
     # Basic functionality
     prefs = AboutPrefs(driver, category="privacy")
+    nav = Navigation(driver)
+    panel_ui = PanelUi(driver)
+
+    # Basic functionality
     prefs.open()
     prefs.select_https_only_setting(prefs.HTTPS_ONLY_STATUS.HTTPS_ONLY_DISABLED)
-    driver.switch_to.new_window("tab")
+    panel_ui.open_and_switch_to_new_window("tab")
     driver.get(HTTP_SITE)
-    nav = Navigation(driver)
     nav.element_attribute_contains("lock-icon", "tooltiptext", CONNECTION_NOT_SECURE)
 
     # Blocking
@@ -58,9 +63,8 @@ def test_http_site(driver: Firefox):
     nav.element_attribute_contains("lock-icon", "tooltiptext", CONNECTION_NOT_SECURE)
 
     # Private browsing - blocked
-    hamburger = PanelUi(driver)
-    hamburger.open_private_window()
-    nav.switch_to_new_window()
+    panel_ui.open_and_switch_to_new_window("private")
+
     try:
         driver.get(HTTP_SITE)
         assert "badssl" not in driver.current_url, "Site should not be displayed"

tests/scrolling_panning_zooming/test_zoom_text_only.py

@@ -97,15 +97,16 @@ def test_zoom_text_only_from_settings(
     """
     # Initializing objects
     nav = Navigation(driver)
+    panel_ui = PanelUi(driver)
 
     # Save the original positions of elements for comparison
-    driver.switch_to.new_window("tab")
+    panel_ui.open_and_switch_to_new_window("tab")
     nav.search(WEBSITE_2)
     web_page.wait.until(lambda _: web_page.title_contains("DuckDuckGo"))
     original_positions = save_original_positions(driver, web_page)
 
     # Set the pref to zoom text only
-    driver.switch_to.new_window("tab")
+    panel_ui.open_and_switch_to_new_window("tab")
     about_prefs = AboutPrefs(driver, category="General").open()
     about_prefs.click_on("zoom-text-only")
 
@@ -132,15 +133,16 @@ def test_zoom_text_only_after_restart(
     """
     # Initializing objects
     nav = Navigation(driver)
+    panel_ui = PanelUi(driver)
 
     # Save the original positions of elements for comparison
-    driver.switch_to.new_window("tab")
+    panel_ui.open_and_switch_to_new_window("tab")
     nav.search(WEBSITE_2)
     web_page.wait.until(lambda _: web_page.title_contains("DuckDuckGo"))
     original_positions = save_original_positions(driver, web_page)
 
     # Set default zoom level
-    driver.switch_to.new_window("tab")
+    panel_ui.open_and_switch_to_new_window("tab")
     about_prefs = AboutPrefs(driver, category="General").open()
     about_prefs.set_default_zoom_level(110)
 

tests/security_and_privacy/test_cookies_not_saved_private_browsing.py

@@ -18,13 +18,11 @@ def test_cookies_not_saved_private_browsing(driver: Firefox):
     # Instantiate objs
     about_prefs = AboutPrefs(driver, category="privacy")
     panel_ui = PanelUi(driver)
-    nav = Navigation(driver)
     google_search = GoogleSearch(driver)
     ba = BrowserActions(driver)
 
     # Open new private window
-    panel_ui.open_private_window()
-    nav.switch_to_new_window()
+    panel_ui.open_and_switch_to_new_window("private")
 
     # Open the Google page and perform a search
     google_search.open()

tests/security_and_privacy/test_downloads_from_private_not_leaked.py

@@ -35,12 +35,10 @@ def test_downloads_from_private_not_leaked(driver: Firefox, delete_files, screen
 
     # We've deleted relevant downloads_file just to be safe
     non_private_window = driver.current_window_handle
-    panelui = PanelUi(driver)
-    panelui.open_panel_menu()
-    panelui.select_panel_setting("new-private-window-option")
-
+    panel_ui = PanelUi(driver)
     nav = Navigation(driver)
-    nav.switch_to_new_window()
+
+    panel_ui.open_and_switch_to_new_window("private")
 
     about_downloads = AboutDownloads(driver)
     about_downloads.open()

tests/security_and_privacy/test_https_enabled_private_browsing.py

@@ -2,7 +2,7 @@
 from selenium.webdriver import Firefox
 from selenium.webdriver.support.wait import WebDriverWait
 
-from modules.browser_object import Navigation, PanelUi
+from modules.browser_object import PanelUi
 from modules.page_object import AboutPrefs
 
 
@@ -19,15 +19,15 @@ def test_https_first_mode_in_private_browsing(driver: Firefox):
     C1362731 Check that https First Mode is properly enabled and working in Private Browsing
     """
 
-    # Navigate to the HTTP Site in a Private Window
+    # Instantiate objects
     prefs = AboutPrefs(driver, category="privacy")
+    panel_ui = PanelUi(driver)
+
+    # Navigate to the HTTP Site in a Private Window
     prefs.open()
     prefs.select_https_only_setting(prefs.HTTPS_ONLY_STATUS.HTTPS_ONLY_PRIVATE)
-    hamburger = PanelUi(driver)
-    hamburger.open_private_window()
-
-    nav = Navigation(driver)
-    nav.switch_to_new_window()
+    panel_ui.open_and_switch_to_new_window("private")
+    driver.get("about:blank")
     driver.get(HTTP_SITE)
 
     # Wait for the URL to be redirected to HTTPS