[grid] Fixing a couple of failing tests.

diemol · diemol · commit adb59828eb3d · 2022-05-25T15:44:16.000+02:00
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);
 
