[grid] Fix tests for Grid status endpoint and UI

Signed-off-by: Viet Nguyen Duc <[email protected]>
SeleniumHQ · Oct 16, 2024 · 625bb94 · 625bb94
1 parent edff2f3
commit 625bb94
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 5 deletions.
diff --git a/java/src/org/openqa/selenium/grid/data/NodeStatus.java b/java/src/org/openqa/selenium/grid/data/NodeStatus.java
@@ -214,7 +214,8 @@ public boolean equals(Object o) {
     return Objects.equals(this.nodeId, that.nodeId)
         && Objects.equals(this.externalUri, that.externalUri)
         && this.maxSessionCount == that.maxSessionCount
-        && this.sessionTimeout == that.sessionTimeout
+        && this.sessionTimeout.compareTo(that.sessionTimeout) == 0
+        && this.heartbeatPeriod.compareTo(that.heartbeatPeriod) == 0
         && Objects.equals(this.slots, that.slots)
         && Objects.equals(this.availability, that.availability)
         && Objects.equals(this.version, that.version);

diff --git a/java/src/org/openqa/selenium/grid/router/GridStatusHandler.java b/java/src/org/openqa/selenium/grid/router/GridStatusHandler.java
@@ -135,7 +135,19 @@ public HttpResponse execute(HttpRequest req) {
 
       List<Map<String, Object>> nodeResults =
           status.getNodes().stream()
-              .map(node -> new ImmutableMap.Builder<String, Object>().putAll(node.toJson()).build())
+              .map(
+                  node ->
+                      new ImmutableMap.Builder<String, Object>()
+                          .put("id", node.getNodeId())
+                          .put("uri", node.getExternalUri())
+                          .put("maxSessions", node.getMaxSessionCount())
+                          .put("sessionTimeout", node.getSessionTimeout().toMillis())
+                          .put("osInfo", node.getOsInfo())
+                          .put("heartbeatPeriod", node.getHeartbeatPeriod().toMillis())
+                          .put("availability", node.getAvailability())
+                          .put("version", node.getVersion())
+                          .put("slots", node.getSlots())
+                          .build())
               .collect(toList());
 
       ImmutableMap.Builder<String, Object> value = ImmutableMap.builder();

diff --git a/java/test/org/openqa/selenium/grid/distributor/local/LocalDistributorTest.java b/java/test/org/openqa/selenium/grid/distributor/local/LocalDistributorTest.java
@@ -103,6 +103,7 @@ public void setUp() throws URISyntaxException {
             .add(caps, new TestSessionFactory((id, c) -> new Handler(c)))
             .maximumConcurrentSessions(2)
             .sessionTimeout(Duration.ofSeconds(30))
+            .heartbeatPeriod(Duration.ofSeconds(5))
             .build();
 
     wait =
@@ -145,6 +146,7 @@ void testAddNodeToDistributor() {
     assertThat(distributorNode.getNodeId()).isEqualByComparingTo(localNode.getId());
     assertThat(distributorNode.getExternalUri()).isEqualTo(uri);
     assertThat(distributorNode.getSessionTimeout()).isEqualTo(Duration.ofSeconds(30));
+    assertThat(distributorNode.getHeartbeatPeriod()).isEqualTo(Duration.ofSeconds(5));
   }
 
   @Test

diff --git a/java/test/org/openqa/selenium/grid/gridui/OverallGridTest.java b/java/test/org/openqa/selenium/grid/gridui/OverallGridTest.java
@@ -68,7 +68,7 @@ public void tearDown() {
 
   @Test
   void shouldReportConcurrencyZeroPercentWhenGridIsStartedWithoutLoad() {
-    driver.get(whereIs(server, "/ui#/sessions"));
+    driver.get(whereIs(server, "/ui/#/sessions"));
 
     WebElement concurrency =
         wait.until(
@@ -79,7 +79,7 @@ void shouldReportConcurrencyZeroPercentWhenGridIsStartedWithoutLoad() {
 
   @Test
   void shouldShowOneNodeRegistered() {
-    driver.get(whereIs(server, "/ui"));
+    driver.get(whereIs(server, "/ui/"));
 
     List<WebElement> nodeInfoIcons =
         wait.until(
@@ -93,7 +93,7 @@ void shouldIncrementSessionCountWhenSessionStarts() {
     WebDriver remoteWebDriver =
         new RemoteWebDriver(server.getUrl(), Browser.detect().getCapabilities());
     try {
-      driver.get(whereIs(server, "/ui#/sessions"));
+      driver.get(whereIs(server, "/ui/#/sessions"));
 
       wait.until(textToBe(By.cssSelector("div[data-testid='session-count']"), "1"));
     } finally {