Call remove method

burakkaygusuz · Nov 11, 2024 · 80c8f71 · 80c8f71
1 parent 498b124
commit 80c8f71
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/main/java/io/github/burakkaygusuz/config/WebDriverBuilder.java b/src/main/java/io/github/burakkaygusuz/config/WebDriverBuilder.java
@@ -63,9 +63,13 @@ public WebDriverBuilder enableHeadless() {
     }
 
     public WebDriver build() {
-        DRIVER_THREAD_LOCAL.set(url != null ? new RemoteWebDriver(url, options, isTracingEnabled)
-                : new RemoteWebDriver(options, isTracingEnabled));
-        WebDriver original = DRIVER_THREAD_LOCAL.get();
-        return new EventFiringDecorator<>(new CustomWebDriverListener()).decorate(original);
+        try {
+            DRIVER_THREAD_LOCAL.set(url != null ? new RemoteWebDriver(url, options, isTracingEnabled)
+                    : new RemoteWebDriver(options, isTracingEnabled));
+            WebDriver original = DRIVER_THREAD_LOCAL.get();
+            return new EventFiringDecorator<>(new CustomWebDriverListener()).decorate(original);
+        } finally {
+            DRIVER_THREAD_LOCAL.remove();
+        }
     }
 }