[grid] Fixing RelayOptionsTest

diemol · diemol · commit e406790ce153 · 2022-04-01T16:41:30.000+02:00
diff --git a/java/test/org/openqa/selenium/grid/node/relay/RelayOptionsTest.java b/java/test/org/openqa/selenium/grid/node/relay/RelayOptionsTest.java
@@ -17,9 +17,6 @@
 
 package org.openqa.selenium.grid.node.relay;
 
-import static org.assertj.core.api.Assertions.assertThat;
-import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
-
 import org.junit.Test;
 import org.openqa.selenium.Capabilities;
 import org.openqa.selenium.grid.config.Config;
@@ -32,9 +29,13 @@
 import org.openqa.selenium.remote.tracing.Tracer;
 
 import java.io.StringReader;
+import java.time.Duration;
 import java.util.Collection;
 import java.util.Map;
 
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
+
 @SuppressWarnings("DuplicatedCode")
 public class RelayOptionsTest {
 
@@ -51,7 +52,10 @@ public void basicConfigurationIsParsedSuccessfully() {
     HttpClient.Factory httpClientFactory = networkOptions.getHttpClientFactory(tracer);
     RelayOptions relayOptions = new RelayOptions(config);
     Map<Capabilities, Collection<SessionFactory>>
-      sessionFactories = relayOptions.getSessionFactories(tracer, httpClientFactory);
+      sessionFactories = relayOptions.getSessionFactories(
+      tracer,
+      httpClientFactory,
+      Duration.ofSeconds(300));
 
     Capabilities chrome = sessionFactories
       .keySet()
@@ -105,7 +109,8 @@ public void missingConfigsThrowsConfigException() {
     Tracer tracer = DefaultTestTracer.createTracer();
     HttpClient.Factory httpClientFactory = networkOptions.getHttpClientFactory(tracer);
     assertThatExceptionOfType(ConfigException.class)
-      .isThrownBy(() -> new RelayOptions(config).getSessionFactories(tracer, httpClientFactory));
+      .isThrownBy(() -> new RelayOptions(config)
+        .getSessionFactories(tracer, httpClientFactory, Duration.ofSeconds(300)));
   }
 
   @Test
@@ -121,7 +126,8 @@ public void incompleteConfigsThrowsConfigException() {
     Tracer tracer = DefaultTestTracer.createTracer();
     HttpClient.Factory httpClientFactory = networkOptions.getHttpClientFactory(tracer);
     assertThatExceptionOfType(ConfigException.class)
-      .isThrownBy(() -> new RelayOptions(config).getSessionFactories(tracer, httpClientFactory));
+      .isThrownBy(() -> new RelayOptions(config)
+        .getSessionFactories(tracer, httpClientFactory, Duration.ofSeconds(300)));
   }
 
 }
