parametrize test for multiple locales

Author: navin772

py/test/selenium/webdriver/common/bidi_emulation_tests.py

@@ -225,37 +225,60 @@ def test_set_geolocation_override_with_error(driver, pages):
     assert "error" in result, f"Expected geolocation error, got: {result}"
 
 
-def test_set_locale_override_with_contexts(driver, pages):
+@pytest.mark.parametrize(
+    "locale,expected_locale",
+    [
+        # Locale with Unicode extension keyword for collation.
+        ("de-DE-u-co-phonebk", "de-DE"),
+        # Lowercase language and region.
+        ("fr-ca", "fr-CA"),
+        # Uppercase language and region (should be normalized by Intl.Locale).
+        ("FR-CA", "fr-CA"),
+        # Mixed case language and region (should be normalized by Intl.Locale).
+        ("fR-cA", "fr-CA"),
+        # Locale with transform extension (simple case).
+        ("en-t-zh", "en"),
+    ],
+)
+def test_set_locale_override_with_contexts(driver, pages, locale, expected_locale):
     """Test setting locale override with browsing contexts."""
     context_id = driver.current_window_handle
 
-    # Set locale override to French
-    test_locale = "fr-FR"
-    driver.emulation.set_locale_override(locale=test_locale, contexts=[context_id])
+    driver.emulation.set_locale_override(locale=locale, contexts=[context_id])
 
     driver.browsing_context.navigate(context_id, pages.url("formPage.html"), wait="complete")
 
     current_locale = get_browser_locale(driver)
-    assert current_locale == test_locale, f"Expected locale {test_locale}, got {current_locale}"
-
-
-def test_set_locale_override_with_user_contexts(driver, pages):
+    assert current_locale == expected_locale, f"Expected locale {expected_locale}, got {current_locale}"
+
+
+@pytest.mark.parametrize(
+    "value",
+    [
+        # Simple language code (2-letter).
+        "en",
+        # Language and region (both 2-letter).
+        "en-US",
+        # Language and script (4-letter).
+        "sr-Latn",
+        # Language, script, and region.
+        "zh-Hans-CN",
+    ],
+)
+def test_set_locale_override_with_user_contexts(driver, pages, value):
     """Test setting locale override with user contexts."""
     user_context = driver.browser.create_user_context()
 
     context_id = driver.browsing_context.create(type=WindowTypes.TAB, user_context=user_context)
 
     driver.switch_to.window(context_id)
 
-    # Set locale override to Spanish
-    test_locale = "es-ES"
-    driver.emulation.set_locale_override(locale=test_locale, user_contexts=[user_context])
+    driver.emulation.set_locale_override(locale=value, user_contexts=[user_context])
 
-    url = pages.url("formPage.html")
-    driver.browsing_context.navigate(context_id, url, wait="complete")
+    driver.browsing_context.navigate(context_id, pages.url("formPage.html"), wait="complete")
 
     current_locale = get_browser_locale(driver)
-    assert current_locale == test_locale, f"Expected locale {test_locale}, got {current_locale}"
+    assert current_locale == value, f"Expected locale {value}, got {current_locale}"
 
     driver.browsing_context.close(context_id)
     driver.browser.remove_user_context(user_context)