From b448578c6061657fef58612943d8a1f372cab136 Mon Sep 17 00:00:00 2001
From: Aryan-Verma <aryanverma@google.com>
Date: Thu, 31 Aug 2023 13:55:08 +0530
Subject: [PATCH] Fix chrome driver issue

---
 pom.xml                                             | 2 +-
 src/main/java/io/cdap/e2e/utils/SeleniumDriver.java | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/pom.xml b/pom.xml
index 4697837ba..1e0aa309b 100644
--- a/pom.xml
+++ b/pom.xml
@@ -340,7 +340,7 @@ the License.-->
     <dependency>
       <groupId>org.seleniumhq.selenium</groupId>
       <artifactId>selenium-chrome-driver</artifactId>
-      <version>4.0.0-rc-1</version>
+      <version>4.11.0</version>
     </dependency>
 
     <dependency>
diff --git a/src/main/java/io/cdap/e2e/utils/SeleniumDriver.java b/src/main/java/io/cdap/e2e/utils/SeleniumDriver.java
index 11dfe4c51..9ab96613a 100644
--- a/src/main/java/io/cdap/e2e/utils/SeleniumDriver.java
+++ b/src/main/java/io/cdap/e2e/utils/SeleniumDriver.java
@@ -16,9 +16,9 @@
 
 package io.cdap.e2e.utils;
 
-import io.github.bonigarcia.wdm.WebDriverManager;
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.chrome.ChromeDriver;
+import org.openqa.selenium.chrome.ChromeDriverService;
 import org.openqa.selenium.chrome.ChromeOptions;
 import org.openqa.selenium.remote.HttpCommandExecutor;
 import org.openqa.selenium.remote.SessionId;
@@ -43,7 +43,7 @@ public class SeleniumDriver {
   private static ChromeDriver chromeDriver;
 
   SeleniumDriver() throws IOException {
-    WebDriverManager.chromedriver().setup();
+    ChromeDriverService service = new ChromeDriverService.Builder().build();
     ChromeOptions chromeOptions = new ChromeOptions();
     chromeOptions.addArguments("--no-sandbox");
     chromeOptions.addArguments("--disable-setuid-sandbox");
@@ -52,7 +52,7 @@ public class SeleniumDriver {
     chromeOptions.addArguments("--disable-gpu");
     chromeOptions.addArguments("--disable-dev-shm-usage");
     chromeOptions.addArguments("--disable-features=VizDisplayCompositor");
-    chromeDriver = new ChromeDriver(chromeOptions);
+    chromeDriver = new ChromeDriver(service, chromeOptions);
     chromeDriver.manage().window().maximize();
     HttpCommandExecutor executor = (HttpCommandExecutor) chromeDriver.getCommandExecutor();
     url = executor.getAddressOfRemoteServer();