From b7a4a307686b9d467578fe70eeb797ce9aa566fb Mon Sep 17 00:00:00 2001
From: Kamalpreet Kaur <kmlkaur73@gmail.com>
Date: Mon, 16 Jan 2023 14:28:12 +0530
Subject: [PATCH] update: use RemoteWebDriver Builder to increase connection
 timeout

---
 pom.xml                                               |  8 ++++++++
 .../java/com/browserstack/BrowserStackTestNGTest.java | 11 +++++++++--
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/pom.xml b/pom.xml
index 7e6608a..c300ad0 100644
--- a/pom.xml
+++ b/pom.xml
@@ -58,6 +58,14 @@
                     </suiteXmlFiles>
                 </configuration>
             </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <source>8</source>
+                    <target>8</target>
+                </configuration>
+            </plugin>
         </plugins>
     </build>
 
diff --git a/src/test/java/com/browserstack/BrowserStackTestNGTest.java b/src/test/java/com/browserstack/BrowserStackTestNGTest.java
index 40dcb45..06b9b33 100644
--- a/src/test/java/com/browserstack/BrowserStackTestNGTest.java
+++ b/src/test/java/com/browserstack/BrowserStackTestNGTest.java
@@ -2,6 +2,7 @@
 
 import java.io.FileReader;
 import java.net.URL;
+import java.time.Duration;
 import java.util.HashMap;
 import java.util.Iterator;
 import java.util.Map;
@@ -13,6 +14,7 @@
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.MutableCapabilities;
 import org.openqa.selenium.remote.RemoteWebDriver;
+import org.openqa.selenium.remote.http.ClientConfig;
 import org.testng.annotations.*;
 
 public class BrowserStackTestNGTest {
@@ -96,8 +98,13 @@ public void setUp(String config_file, String environment) throws Exception {
             bstackOptionsMap.put("source", "testng:sample-selenium-4:v1.1");
         }
 
-        driver = new RemoteWebDriver(
-                new URL("https://" + username + ":" + accessKey + "@" + config.get("server") + "/wd/hub"), capabilities);
+        ClientConfig customConfig = ClientConfig.defaultConfig().readTimeout(Duration.ofMinutes(15))
+            .connectionTimeout(Duration.ofMinutes(15));
+        driver = RemoteWebDriver.builder()
+            .config(customConfig)
+            .address(new URL("https://"+username+":"+accessKey+"@"+config.get("server")+"/wd/hub"))
+            .oneOf(capabilities)
+            .build();
     }
 
     @AfterMethod(alwaysRun = true)