X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fdemo%2FWebDriverExtension.java;fp=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Fdemo%2FWebDriverExtension.java;h=ada01f51c6112adc6a78a01818e45eedb3b69a51;hb=d968707a4bb1b520df07f363b7e8077061cb1ccf;hp=a2c5d0ccf7aecee55641d84b81f552c6044fff3c;hpb=436245ca8e2991d8b1e3ceb8d981cb166a06f9e8;p=demos%2Fspring-boot

diff --git a/src/test/java/de/juplo/demo/WebDriverExtension.java b/src/test/java/de/juplo/demo/WebDriverExtension.java
index a2c5d0c..ada01f5 100644
--- a/src/test/java/de/juplo/demo/WebDriverExtension.java
+++ b/src/test/java/de/juplo/demo/WebDriverExtension.java
@@ -7,6 +7,7 @@ import org.junit.jupiter.api.extension.ParameterContext;
 import org.junit.jupiter.api.extension.ParameterResolver;
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.remote.DesiredCapabilities;
+import org.openqa.selenium.remote.RemoteWebDriver;
 import org.testcontainers.containers.BrowserWebDriverContainer;
 
 
@@ -32,7 +33,7 @@ public class WebDriverExtension implements BeforeEachCallback, ParameterResolver
       ParameterContext parameterContext,
       ExtensionContext extensionContext)
   {
-    return parameterContext.getParameter().getType() == BrowserWebDriverContainer.class;
+    return parameterContext.getParameter().getType() == RemoteWebDriver.class;
   }
 
   @Override
@@ -40,6 +41,6 @@ public class WebDriverExtension implements BeforeEachCallback, ParameterResolver
       ParameterContext parameterContext,
       ExtensionContext extensionContext)
   {
-    return chrome;
+    return chrome.getWebDriver();
   }
 }