[java][grid]: video file name set independently in dynamic grid via s…

…e:videoName
NDViet · Jun 18, 2024 · d67abf6 · d67abf6
1 parent dec42aa
commit d67abf6
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/java/src/org/openqa/selenium/grid/node/docker/DockerSessionFactory.java b/java/src/org/openqa/selenium/grid/node/docker/DockerSessionFactory.java
@@ -373,13 +373,21 @@ private Map<String, String> getVideoContainerEnvVars(
     // Capabilities set to env vars with higher precedence
     setCapsToEnvVars(sessionRequestCapabilities, envVars);
     envVars.put("DISPLAY_CONTAINER_NAME", containerIp);
-    Optional<String> testName = ofNullable(getTestName(sessionRequestCapabilities));
-    testName.ifPresent(name -> envVars.put("SE_VIDEO_FILE_NAME", String.format("%s.mp4", name)));
+    Optional<String> videoName =
+        ofNullable(getVideoFileName(sessionRequestCapabilities, "se:videoName"));
+    if (videoName.isPresent()) {
+      envVars.put("SE_VIDEO_FILE_NAME", String.format("%s.mp4", videoName.get()));
+    } else {
+      Optional<String> testName =
+          ofNullable(getVideoFileName(sessionRequestCapabilities, "se:name"));
+      testName.ifPresent(name -> envVars.put("SE_VIDEO_FILE_NAME", String.format("%s.mp4", name)));
+    }
     return envVars;
   }
 
-  private String getTestName(Capabilities sessionRequestCapabilities) {
-    Optional<Object> testName = ofNullable(sessionRequestCapabilities.getCapability("se:name"));
+  private String getVideoFileName(Capabilities sessionRequestCapabilities, String capabilityName) {
+    Optional<Object> testName =
+        ofNullable(sessionRequestCapabilities.getCapability(capabilityName));
     if (testName.isPresent()) {
       String name = testName.get().toString();
       if (!name.isEmpty()) {