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=a2c5d0ccf7aecee55641d84b81f552c6044fff3c;hb=3dcb574d91d50c6bd1dceb8d6a403cb01d833908;hp=25de95f8b8bc6d845d8c6918ea56fc28e9d81bfa;hpb=a9ab3a5ef5b84e0447e5794d409e940c6d2254a9;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 25de95f..a2c5d0c 100644
--- a/src/test/java/de/juplo/demo/WebDriverExtension.java
+++ b/src/test/java/de/juplo/demo/WebDriverExtension.java
@@ -3,6 +3,8 @@ package de.juplo.demo;
 
 import org.junit.jupiter.api.extension.BeforeEachCallback;
 import org.junit.jupiter.api.extension.ExtensionContext;
+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.testcontainers.containers.BrowserWebDriverContainer;
@@ -14,7 +16,7 @@ import org.testcontainers.containers.BrowserWebDriverContainer;
  * {@link https://www.testcontainers.org/modules/webdriver_containers/ Testcontainers}.
  * @author Kai Moritz
  */
-public class WebDriverExtension implements BeforeEachCallback
+public class WebDriverExtension implements BeforeEachCallback, ParameterResolver
 {
   BrowserWebDriverContainer chrome;
 
@@ -24,4 +26,20 @@ public class WebDriverExtension implements BeforeEachCallback
   {
     chrome = new BrowserWebDriverContainer().withCapabilities(DesiredCapabilities.chrome());
   }
+
+  @Override
+  public boolean supportsParameter(
+      ParameterContext parameterContext,
+      ExtensionContext extensionContext)
+  {
+    return parameterContext.getParameter().getType() == BrowserWebDriverContainer.class;
+  }
+
+  @Override
+  public Object resolveParameter(
+      ParameterContext parameterContext,
+      ExtensionContext extensionContext)
+  {
+    return chrome;
+  }
 }