WIP

[demos/spring-boot] / src / test / java / de / juplo / demo / WebDriverExtension.java

diff --git a/src/test/java/de/juplo/demo/WebDriverExtension.java b/src/test/java/de/juplo/demo/WebDriverExtension.java
index a2c5d0c..f53bcd4 100644 (file)
--- a/src/test/java/de/juplo/demo/WebDriverExtension.java
+++ b/src/test/java/de/juplo/demo/WebDriverExtension.java
@@ -1,12 +1,14 @@
 package de.juplo.demo;
 
 
+import java.io.File;
 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.openqa.selenium.remote.RemoteWebDriver;
 import org.testcontainers.containers.BrowserWebDriverContainer;
 
 
@@ -24,7 +26,10 @@ public class WebDriverExtension implements BeforeEachCallback, ParameterResolver
   @Override
   public void beforeEach(ExtensionContext context) throws Exception
   {
-    chrome = new BrowserWebDriverContainer().withCapabilities(DesiredCapabilities.chrome());
+    chrome =
+        new BrowserWebDriverContainer()
+            .withCapabilities(DesiredCapabilities.chrome())
+            .withRecordingMode(BrowserWebDriverContainer.VncRecordingMode.RECORD_ALL, new File("./target/"));
   }
 
   @Override
@@ -32,7 +37,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 +45,6 @@ public class WebDriverExtension implements BeforeEachCallback, ParameterResolver
       ParameterContext parameterContext,
       ExtensionContext extensionContext)
   {
-    return chrome;
+    return chrome.getWebDriver();
   }
 }