[cdp][java] Continue requests without modification for know errors in…

… NetworkInterceptor Related to SeleniumHQ#13774
pujagani · Apr 22, 2024 · 8b8418b · 8b8418b
1 parent e85bf8d
commit 8b8418b
Show file tree

Hide file tree

Showing 6 changed files with 46 additions and 9 deletions.
diff --git a/java/src/org/openqa/selenium/devtools/idealized/Network.java b/java/src/org/openqa/selenium/devtools/idealized/Network.java
@@ -200,6 +200,12 @@ public void prepareToInterceptTraffic() {
         pausedRequest -> {
           try {
             String id = getRequestId(pausedRequest);
+
+            if (hasErrorResponse(pausedRequest)) {
+              devTools.send(continueWithoutModification(pausedRequest));
+              return;
+            }
+
             Either<HttpRequest, HttpResponse> message = createSeMessages(pausedRequest);
 
             if (message.isRight()) {
@@ -348,6 +354,8 @@ protected abstract Command<Void> continueWithAuth(
 
   protected abstract Either<HttpRequest, HttpResponse> createSeMessages(REQUESTPAUSED pausedReq);
 
+  protected abstract boolean hasErrorResponse(REQUESTPAUSED pausedReq);
+
   protected abstract Command<Void> continueWithoutModification(REQUESTPAUSED pausedReq);
 
   protected abstract Command<Void> continueRequest(REQUESTPAUSED pausedReq, HttpRequest req);

diff --git a/java/src/org/openqa/selenium/devtools/v121/v121Network.java b/java/src/org/openqa/selenium/devtools/v121/v121Network.java
@@ -114,8 +114,7 @@ public Event<RequestPaused> requestPausedEvent() {
 
   @Override
   public Either<HttpRequest, HttpResponse> createSeMessages(RequestPaused pausedReq) {
-    if (pausedReq.getResponseStatusCode().isPresent()
-        || pausedReq.getResponseErrorReason().isPresent()) {
+    if (pausedReq.getResponseStatusCode().isPresent()) {
       String body;
       boolean bodyIsBase64Encoded;
 
@@ -161,6 +160,11 @@ public Either<HttpRequest, HttpResponse> createSeMessages(RequestPaused pausedRe
     return Either.left(req);
   }
 
+  @Override
+  protected boolean hasErrorResponse(RequestPaused pausedReq) {
+    return pausedReq.getResponseErrorReason().isPresent();
+  }
+
   @Override
   protected String getRequestId(RequestPaused pausedReq) {
     return pausedReq.getRequestId().toString();

diff --git a/java/src/org/openqa/selenium/devtools/v122/v122Network.java b/java/src/org/openqa/selenium/devtools/v122/v122Network.java
@@ -114,8 +114,7 @@ public Event<RequestPaused> requestPausedEvent() {
 
   @Override
   public Either<HttpRequest, HttpResponse> createSeMessages(RequestPaused pausedReq) {
-    if (pausedReq.getResponseStatusCode().isPresent()
-        || pausedReq.getResponseErrorReason().isPresent()) {
+    if (pausedReq.getResponseStatusCode().isPresent()) {
       String body;
       boolean bodyIsBase64Encoded;
 
@@ -161,6 +160,11 @@ public Either<HttpRequest, HttpResponse> createSeMessages(RequestPaused pausedRe
     return Either.left(req);
   }
 
+  @Override
+  protected boolean hasErrorResponse(RequestPaused pausedReq) {
+    return pausedReq.getResponseErrorReason().isPresent();
+  }
+
   @Override
   protected String getRequestId(RequestPaused pausedReq) {
     return pausedReq.getRequestId().toString();

diff --git a/java/src/org/openqa/selenium/devtools/v123/v123Network.java b/java/src/org/openqa/selenium/devtools/v123/v123Network.java
@@ -114,8 +114,7 @@ public Event<RequestPaused> requestPausedEvent() {
 
   @Override
   public Either<HttpRequest, HttpResponse> createSeMessages(RequestPaused pausedReq) {
-    if (pausedReq.getResponseStatusCode().isPresent()
-        || pausedReq.getResponseErrorReason().isPresent()) {
+    if (pausedReq.getResponseStatusCode().isPresent()) {
       String body;
       boolean bodyIsBase64Encoded;
 
@@ -161,6 +160,11 @@ public Either<HttpRequest, HttpResponse> createSeMessages(RequestPaused pausedRe
     return Either.left(req);
   }
 
+  @Override
+  protected boolean hasErrorResponse(RequestPaused pausedReq) {
+    return pausedReq.getResponseErrorReason().isPresent();
+  }
+
   @Override
   protected String getRequestId(RequestPaused pausedReq) {
     return pausedReq.getRequestId().toString();

diff --git a/java/src/org/openqa/selenium/devtools/v85/V85Network.java b/java/src/org/openqa/selenium/devtools/v85/V85Network.java
@@ -124,8 +124,7 @@ public Event<RequestPaused> requestPausedEvent() {
 
   @Override
   public Either<HttpRequest, HttpResponse> createSeMessages(RequestPaused pausedReq) {
-    if (pausedReq.getResponseStatusCode().isPresent()
-        || pausedReq.getResponseErrorReason().isPresent()) {
+    if (pausedReq.getResponseStatusCode().isPresent()) {
       String body;
       boolean bodyIsBase64Encoded;
 
@@ -171,6 +170,11 @@ public Either<HttpRequest, HttpResponse> createSeMessages(RequestPaused pausedRe
     return Either.left(req);
   }
 
+  @Override
+  protected boolean hasErrorResponse(RequestPaused pausedReq) {
+    return pausedReq.getResponseErrorReason().isPresent();
+  }
+
   @Override
   protected String getRequestId(RequestPaused pausedReq) {
     return pausedReq.getRequestId().toString();

diff --git a/java/test/org/openqa/selenium/devtools/NetworkInterceptorTest.java b/java/test/org/openqa/selenium/devtools/NetworkInterceptorTest.java
@@ -20,7 +20,8 @@
 import static com.google.common.net.MediaType.XHTML_UTF_8;
 import static java.net.HttpURLConnection.HTTP_MOVED_TEMP;
 import static java.net.HttpURLConnection.HTTP_NOT_FOUND;
-import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.AssertionsForClassTypes.assertThatExceptionOfType;
+import static org.assertj.core.api.AssertionsForInterfaceTypes.assertThat;
 import static org.assertj.core.api.Assumptions.assumeThat;
 import static org.openqa.selenium.remote.http.Contents.utf8String;
 import static org.openqa.selenium.testing.Safely.safelyCall;
@@ -35,7 +36,9 @@
 import org.junit.jupiter.api.Test;
 import org.openqa.selenium.By;
 import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.WebDriverException;
 import org.openqa.selenium.environment.webserver.NettyAppServer;
+import org.openqa.selenium.net.PortProber;
 import org.openqa.selenium.remote.http.Contents;
 import org.openqa.selenium.remote.http.Filter;
 import org.openqa.selenium.remote.http.HttpMethod;
@@ -248,4 +251,14 @@ void shouldHandleRedirects() {
       assertThat(body).contains("Hello, World!");
     }
   }
+
+  @Test
+  @NoDriverBeforeTest
+  void shouldProceedAsNormalIfRequestResultInAnKnownError() {
+    Filter filter = next -> next;
+    try (NetworkInterceptor ignored = new NetworkInterceptor(driver, filter)) {
+      assertThatExceptionOfType(WebDriverException.class)
+          .isThrownBy(() -> driver.get("http://localhost:" + PortProber.findFreePort()));
+    }
+  }
 }