[grid] Fixing a couple of failing tests.

SeleniumHQ · May 25, 2022 · adb5982 · adb5982
1 parent 70b0bb8
commit adb5982
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/java/test/org/openqa/selenium/grid/router/SessionCleanUpTest.java b/java/test/org/openqa/selenium/grid/router/SessionCleanUpTest.java
@@ -221,12 +221,13 @@ public void shouldRemoveSessionAfterNodeIsShutDownGracefully() {
     Optional<Map<String, Object>> maybeResponse =
       Optional.ofNullable(Values.get(httpResponse, Map.class));
 
+    assertThat(maybeResponse.isPresent()).isTrue();
     String rawResponse = JSON.toJson(maybeResponse.get().get("sessionId"));
     SessionId id = JSON.toType(rawResponse, SessionId.class);
 
     Session session = sessions.get(id);
 
-    assertThat(session.getCapabilities()).isEqualTo(capabilities);
+    assertThat(session.getCapabilities().getBrowserName()).isEqualTo(capabilities.getBrowserName());
 
     nodeServer.stop();
 
@@ -296,7 +297,7 @@ public void shouldRemoveSessionAfterNodeIsDown() throws URISyntaxException {
     SessionId id = result.right().getSession().getId();
     Session session = sessions.get(id);
 
-    assertThat(session.getCapabilities()).isEqualTo(capabilities);
+    assertThat(session.getCapabilities().getBrowserName()).isEqualTo(capabilities.getBrowserName());
 
     availability.set(DOWN);