[grid] Fixing DistributorTest

diemol · diemol · commit 3f44bc8f5365 · 2022-05-25T17:28:41.000+02:00
diff --git a/java/test/org/openqa/selenium/grid/distributor/DistributorTest.java b/java/test/org/openqa/selenium/grid/distributor/DistributorTest.java
@@ -233,7 +233,8 @@ public void shouldBeAbleToAddANodeAndCreateASession() {
       distributor.newSession(createRequest(sessionCaps));
     assertThatEither(result).isRight();
     Session session = result.right().getSession();
-    assertThat(session.getCapabilities()).isEqualTo(sessionCaps);
+    assertThat(session.getCapabilities().getCapability("sausages"))
+      .isEqualTo(sessionCaps.getCapability("sausages"));
     assertThat(session.getUri()).isEqualTo(routableUri);
   }
 
@@ -275,7 +276,8 @@ public void creatingASessionAddsItToTheSessionMap() {
     assertThatEither(result).isRight();
     Session returned = result.right().getSession();
     Session session = sessions.get(returned.getId());
-    assertThat(session.getCapabilities()).isEqualTo(sessionCaps);
+    assertThat(session.getCapabilities().getCapability("sausages"))
+      .isEqualTo(sessionCaps.getCapability("sausages"));
     assertThat(session.getUri()).isEqualTo(routableUri);
   }
 
