[grid] Fixing DistributorTest

SeleniumHQ · May 25, 2022 · 3f44bc8 · 3f44bc8
1 parent adb5982
commit 3f44bc8
Showing 1 changed file with 4 additions and 2 deletions.
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);
   }