remove SetCacheBehaviorParameters class

Author: navin772

java/src/org/openqa/selenium/bidi/module/Network.java

@@ -19,6 +19,7 @@
 
 import java.util.Collections;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Map;
 import java.util.Set;
 import java.util.function.Consumer;
@@ -30,12 +31,12 @@
 import org.openqa.selenium.bidi.HasBiDi;
 import org.openqa.selenium.bidi.network.AddInterceptParameters;
 import org.openqa.selenium.bidi.network.BeforeRequestSent;
+import org.openqa.selenium.bidi.network.CacheBehavior;
 import org.openqa.selenium.bidi.network.ContinueRequestParameters;
 import org.openqa.selenium.bidi.network.ContinueResponseParameters;
 import org.openqa.selenium.bidi.network.FetchError;
 import org.openqa.selenium.bidi.network.ProvideResponseParameters;
 import org.openqa.selenium.bidi.network.ResponseDetails;
-import org.openqa.selenium.bidi.network.SetCacheBehaviorParameters;
 import org.openqa.selenium.internal.Require;
 
 public class Network implements AutoCloseable {
@@ -138,8 +139,20 @@ public void provideResponse(ProvideResponseParameters parameters) {
     this.bidi.send(new Command<>("network.provideResponse", parameters.toMap()));
   }
 
-  public void setCacheBehavior(SetCacheBehaviorParameters parameters) {
-    this.bidi.send(new Command<>("network.setCacheBehavior", parameters.toMap()));
+  public void setCacheBehavior(CacheBehavior cacheBehavior) {
+    Require.nonNull("Cache behavior", cacheBehavior);
+    this.bidi.send(
+        new Command<>(
+            "network.setCacheBehavior", Map.of("cacheBehavior", cacheBehavior.toString())));
+  }
+
+  public void setCacheBehavior(CacheBehavior cacheBehavior, List<String> contexts) {
+    Require.nonNull("Cache behavior", cacheBehavior);
+    Require.nonNull("Contexts", contexts);
+    this.bidi.send(
+        new Command<>(
+            "network.setCacheBehavior",
+            Map.of("cacheBehavior", cacheBehavior.toString(), "contexts", contexts)));
   }
 
   public void onBeforeRequestSent(Consumer<BeforeRequestSent> consumer) {

java/src/org/openqa/selenium/bidi/network/SetCacheBehaviorParameters.java

@@ -278,9 +278,7 @@ void canSetCacheBehaviorToBypass() {
       BrowsingContext context = new BrowsingContext(driver, WindowType.TAB);
       String contextId = context.getId();
 
-      network.setCacheBehavior(
-          new SetCacheBehaviorParameters(
-              CacheBehavior.BYPASS, Collections.singletonList(contextId)));
+      network.setCacheBehavior(CacheBehavior.BYPASS, Collections.singletonList(contextId));
     }
   }
 
@@ -293,9 +291,7 @@ void canSetCacheBehaviorToDefault() {
       BrowsingContext context = new BrowsingContext(driver, WindowType.TAB);
       String contextId = context.getId();
 
-      network.setCacheBehavior(
-          new SetCacheBehaviorParameters(
-              CacheBehavior.DEFAULT, Collections.singletonList(contextId)));
+      network.setCacheBehavior(CacheBehavior.DEFAULT, Collections.singletonList(contextId));
     }
   }
 
@@ -305,8 +301,8 @@ void canSetCacheBehaviorWithNoContextId() {
     try (Network network = new Network(driver)) {
       page = appServer.whereIs("basicAuth");
 
-      network.setCacheBehavior(new SetCacheBehaviorParameters(CacheBehavior.BYPASS));
-      network.setCacheBehavior(new SetCacheBehaviorParameters(CacheBehavior.DEFAULT));
+      network.setCacheBehavior(CacheBehavior.BYPASS);
+      network.setCacheBehavior(CacheBehavior.DEFAULT);
     }
   }
 
@@ -319,8 +315,7 @@ void throwsExceptionForInvalidContext() {
       assertThatThrownBy(
               () ->
                   network.setCacheBehavior(
-                      new SetCacheBehaviorParameters(
-                          CacheBehavior.DEFAULT, Collections.singletonList("invalid-context"))))
+                      CacheBehavior.DEFAULT, Collections.singletonList("invalid-context")))
           .isInstanceOf(BiDiException.class)
           .hasMessageContaining("no such frame");
     }