add tests

Author: navin772

java/test/org/openqa/selenium/bidi/network/NetworkCommandsTest.java

@@ -25,6 +25,7 @@
 import java.time.Duration;
 import java.time.temporal.ChronoUnit;
 import java.util.Collections;
+import java.util.List;
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicInteger;
@@ -353,4 +354,160 @@ void throwsExceptionForInvalidContext() {
           .hasMessageContaining("no such frame");
     }
   }
+
+  @Test
+  @NeedsFreshDriver
+  void canAddDataCollector() {
+    try (Network network = new Network(driver)) {
+      AddDataCollectorParameters parameters =
+          new AddDataCollectorParameters(List.of(DataType.RESPONSE), 1024);
+
+      String collector = network.addDataCollector(parameters);
+      assertThat(collector).isNotNull();
+    }
+  }
+
+  @Test
+  @NeedsFreshDriver
+  void canAddDataCollectorWithCollectorType() {
+    try (Network network = new Network(driver)) {
+      AddDataCollectorParameters parameters =
+          new AddDataCollectorParameters(List.of(DataType.RESPONSE), 2048).collectorType("blob");
+
+      String collector = network.addDataCollector(parameters);
+      assertThat(collector).isNotNull();
+    }
+  }
+
+  @Test
+  @NeedsFreshDriver
+  void canAddDataCollectorWithContexts() {
+    try (Network network = new Network(driver)) {
+      String contextId = driver.getWindowHandle();
+      AddDataCollectorParameters parameters =
+          new AddDataCollectorParameters(List.of(DataType.RESPONSE), 1024)
+              .contexts(List.of(contextId));
+
+      String collector = network.addDataCollector(parameters);
+      assertThat(collector).isNotNull();
+    }
+  }
+
+  @Test
+  @NeedsFreshDriver
+  void canGetDataWithoutCollector() throws InterruptedException {
+    try (Network network = new Network(driver)) {
+      CountDownLatch latch = new CountDownLatch(1);
+
+      network.onResponseCompleted(
+          responseDetails -> {
+            String requestId = responseDetails.getRequest().getRequestId();
+            GetDataParameters parameters = new GetDataParameters(DataType.RESPONSE, requestId);
+
+            // browsers don't store body bytes for responses without a collector
+            assertThatThrownBy(() -> network.getData(parameters))
+                .isInstanceOf(BiDiException.class)
+                .hasMessageContaining("no such network data");
+
+            latch.countDown();
+          });
+
+      BrowsingContext browsingContext = new BrowsingContext(driver, driver.getWindowHandle());
+      browsingContext.navigate(
+          appServer.whereIs("/bidi/logEntryAdded.html"), ReadinessState.COMPLETE);
+
+      boolean countdown = latch.await(5, TimeUnit.SECONDS);
+      assertThat(countdown).isTrue();
+    }
+  }
+
+  @Test
+  @NeedsFreshDriver
+  void canGetDataWithCollector() throws InterruptedException {
+    try (Network network = new Network(driver)) {
+      AddDataCollectorParameters collectorParams =
+          new AddDataCollectorParameters(List.of(DataType.RESPONSE), 1024);
+      String collector = network.addDataCollector(collectorParams);
+
+      CountDownLatch latch = new CountDownLatch(1);
+
+      network.onResponseCompleted(
+          responseDetails -> {
+            String requestId = responseDetails.getRequest().getRequestId();
+            GetDataParameters parameters =
+                new GetDataParameters(DataType.RESPONSE, requestId).collector(collector);
+
+            BytesValue result = network.getData(parameters);
+            assertThat(result).isNotNull();
+            assertThat(result.getValue()).isNotNull();
+
+            latch.countDown();
+          });
+
+      BrowsingContext browsingContext = new BrowsingContext(driver, driver.getWindowHandle());
+      browsingContext.navigate(
+          appServer.whereIs("/bidi/logEntryAdded.html"), ReadinessState.COMPLETE);
+
+      boolean countdown = latch.await(5, TimeUnit.SECONDS);
+      assertThat(countdown).isTrue();
+    }
+  }
+
+  @Test
+  @NeedsFreshDriver
+  void canGetDataWithDisown() throws InterruptedException {
+    try (Network network = new Network(driver)) {
+      AddDataCollectorParameters collectorParams =
+          new AddDataCollectorParameters(List.of(DataType.RESPONSE), 2048);
+      String collector = network.addDataCollector(collectorParams);
+
+      CountDownLatch latch = new CountDownLatch(1);
+
+      network.onResponseCompleted(
+          responseDetails -> {
+            String requestId = responseDetails.getRequest().getRequestId();
+            GetDataParameters disownParams =
+                new GetDataParameters(DataType.RESPONSE, requestId)
+                    .collector(collector)
+                    .disown(true);
+
+            // First fetch should succeed
+            BytesValue result = network.getData(disownParams);
+            assertThat(result).isNotNull();
+            assertThat(result.getValue()).isNotNull();
+
+            // Second fetch with same collector should fail, because disowned
+            GetDataParameters secondParams =
+                new GetDataParameters(DataType.RESPONSE, requestId).collector(collector);
+
+            assertThatThrownBy(() -> network.getData(secondParams))
+                .isInstanceOf(BiDiException.class)
+                .hasMessageContaining("no such network data");
+
+            latch.countDown();
+          });
+
+      BrowsingContext browsingContext = new BrowsingContext(driver, driver.getWindowHandle());
+      browsingContext.navigate(
+          appServer.whereIs("/bidi/logEntryAdded.html"), ReadinessState.COMPLETE);
+
+      boolean countdown = latch.await(5, TimeUnit.SECONDS);
+      assertThat(countdown).isTrue();
+    }
+  }
+
+  @Test
+  @NeedsFreshDriver
+  void canRemoveDataCollector() {
+    try (Network network = new Network(driver)) {
+      AddDataCollectorParameters parameters =
+          new AddDataCollectorParameters(List.of(DataType.RESPONSE), 1024);
+
+      String collector = network.addDataCollector(parameters);
+      assertThat(collector).isNotNull();
+
+      // Remove the collector, should not throw any exception
+      network.removeDataCollector(collector);
+    }
+  }
 }