[bidi][java] Add missing code example for Script module

SeleniumHQ · Feb 23, 2024 · e24abbe · e24abbe
1 parent 4b3e644
commit e24abbe
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/examples/java/src/test/java/dev/selenium/bidirectional/webdriver_bidi/ScriptTest.java b/examples/java/src/test/java/dev/selenium/bidirectional/webdriver_bidi/ScriptTest.java
@@ -1,6 +1,8 @@
 package dev.selenium.bidirectional.webdriver_bidi;
 
 import dev.selenium.BaseTest;
+
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Optional;
 import java.util.concurrent.CompletableFuture;
@@ -24,6 +26,7 @@
 import org.openqa.selenium.bidi.script.EvaluateResultExceptionValue;
 import org.openqa.selenium.bidi.script.EvaluateResultSuccess;
 import org.openqa.selenium.bidi.script.LocalValue;
+import org.openqa.selenium.bidi.script.PrimitiveProtocolValue;
 import org.openqa.selenium.bidi.script.RealmInfo;
 import org.openqa.selenium.bidi.script.RealmType;
 import org.openqa.selenium.bidi.script.RemoteReference;
@@ -52,6 +55,30 @@ void canCallFunctionWithDeclaration() {
         }
     }
 
+    @Test
+    void canCallFunctionWithArguments() {
+        String id = driver.getWindowHandle();
+        try (Script script = new Script(id, driver)) {
+            List<LocalValue> arguments = new ArrayList<>();
+            LocalValue value1 = PrimitiveProtocolValue.stringValue("ARGUMENT_STRING_VALUE");
+            LocalValue value2 = PrimitiveProtocolValue.numberValue(42);
+            arguments.add(value1);
+            arguments.add(value2);
+
+            EvaluateResult result =
+                    script.callFunctionInBrowsingContext(
+                            id,
+                            "(...args)=>{return args}",
+                            false,
+                            Optional.of(arguments),
+                            Optional.empty(),
+                            Optional.empty());
+
+            EvaluateResultSuccess successResult = (EvaluateResultSuccess) result;
+            Assertions.assertEquals(2, ((List<Object>) successResult.getResult().getValue().get()).size());
+        }
+    }
+
     @Test
     void canCallFunctionWithAwaitPromise() {
         String id = driver.getWindowHandle();