From 2dc901ef6ff3686df2fe886fa15cc46fbe3a4b8f Mon Sep 17 00:00:00 2001
From: labkey-tchad <tchad@labkey.com>
Date: Tue, 4 Feb 2025 12:29:30 -0800
Subject: [PATCH] Add test property to set the browser time zone

---
 src/org/labkey/test/TestProperties.java   |  5 +++++
 src/org/labkey/test/WebDriverWrapper.java | 18 ++++++++++++++----
 test.properties.template                  |  2 ++
 3 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/src/org/labkey/test/TestProperties.java b/src/org/labkey/test/TestProperties.java
index a7b984402c..4f8eeb88f1 100644
--- a/src/org/labkey/test/TestProperties.java
+++ b/src/org/labkey/test/TestProperties.java
@@ -162,6 +162,11 @@ public static boolean isDumpBrowserConsole()
         return getBooleanProperty("webtest.dump.browser.console", false);
     }
 
+    public static String getBrowserTimeZone()
+    {
+        return System.getProperty("webtest.browser.tz");
+    }
+
     public static double getTimeoutMultiplier()
     {
         return Math.max(0, getDoubleProperty("webtest.timeout.multiplier", 1.0));
diff --git a/src/org/labkey/test/WebDriverWrapper.java b/src/org/labkey/test/WebDriverWrapper.java
index b7308ebb6b..97ca573bd1 100644
--- a/src/org/labkey/test/WebDriverWrapper.java
+++ b/src/org/labkey/test/WebDriverWrapper.java
@@ -96,6 +96,7 @@
 import org.openqa.selenium.firefox.FirefoxProfile;
 import org.openqa.selenium.firefox.GeckoDriverService;
 import org.openqa.selenium.ie.InternetExplorerDriver;
+import org.openqa.selenium.ie.InternetExplorerDriverService;
 import org.openqa.selenium.interactions.Actions;
 import org.openqa.selenium.remote.RemoteWebDriver;
 import org.openqa.selenium.remote.service.DriverService;
@@ -210,6 +211,11 @@ protected Pair<WebDriver, DriverService> createNewWebDriver(@NotNull Pair<WebDri
         WebDriver newWebDriver = null;
         DriverService oldDriverService = oldDriverAndService.getRight();
         DriverService newDriverService = null;
+        Map<String, String> browserEnv = new HashMap<>();
+        if (TestProperties.getBrowserTimeZone() != null)
+        {
+            browserEnv.put("TZ", TestProperties.getBrowserTimeZone());
+        }
 
         switch (browserType)
         {
@@ -236,7 +242,8 @@ protected Pair<WebDriver, DriverService> createNewWebDriver(@NotNull Pair<WebDri
                 }
                 if (oldWebDriver == null)
                 {
-                    newWebDriver = new InternetExplorerDriver();
+                    newDriverService = new InternetExplorerDriverService.Builder().withEnvironment(browserEnv).build();
+                    newWebDriver = new InternetExplorerDriver((InternetExplorerDriverService) newDriverService);
                 }
                 break;
             }
@@ -276,7 +283,7 @@ protected Pair<WebDriver, DriverService> createNewWebDriver(@NotNull Pair<WebDri
                         options.addArguments("headless");
                     }
 
-                    newDriverService = ChromeDriverService.createDefaultService();
+                    newDriverService = new ChromeDriverService.Builder().withEnvironment(browserEnv).build();
                     newWebDriver = new ChromeDriver((ChromeDriverService) newDriverService, options);
                 }
                 break;
@@ -373,7 +380,10 @@ protected Pair<WebDriver, DriverService> createNewWebDriver(@NotNull Pair<WebDri
                     capabilities.setUnhandledPromptBehaviour(UnexpectedAlertBehaviour.IGNORE);
                     FirefoxOptions firefoxOptions = new FirefoxOptions(capabilities);
 
-                    newDriverService = GeckoDriverService.createDefaultService();
+                    GeckoDriverService.Builder driverServiceBuilder = new GeckoDriverService.Builder()
+                        .withEnvironment(browserEnv);
+
+                    newDriverService = driverServiceBuilder.build();
                     try
                     {
                         newWebDriver = new FirefoxDriver((FirefoxDriverService) newDriverService, firefoxOptions);
@@ -384,7 +394,7 @@ protected Pair<WebDriver, DriverService> createNewWebDriver(@NotNull Pair<WebDri
                         {
                             retry.printStackTrace(System.err);
                             newDriverService.stop();
-                            newDriverService = GeckoDriverService.createDefaultService();
+                            newDriverService = driverServiceBuilder.build();
                             sleep(10000);
                             newWebDriver = new FirefoxDriver((FirefoxDriverService) newDriverService, firefoxOptions);
                         }
diff --git a/test.properties.template b/test.properties.template
index 66e7f534e8..90aed9ef8b 100644
--- a/test.properties.template
+++ b/test.properties.template
@@ -88,6 +88,8 @@ selenium.reuseWebDriver=true
 #webtest.webdriver.headless=false
 ## Customize browser window size
 #webtest.window.size=1280x1024
+## Set browser time zone
+#webtest.browser.tz=America/New_York
 
 
 #==============================================================================