[java] Allow setting print options page ranges using list

SeleniumHQ · Nov 21, 2023 · eaaa502 · eaaa502
1 parent c049595
commit eaaa502
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/java/src/org/openqa/selenium/print/PrintOptions.java b/java/src/org/openqa/selenium/print/PrintOptions.java
@@ -17,7 +17,9 @@
 
 package org.openqa.selenium.print;
 
+import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 import org.openqa.selenium.internal.Require;
 
@@ -71,6 +73,11 @@ public void setPageRanges(String firstRange, String... ranges) {
     }
   }
 
+  public void setPageRanges(List<String> ranges) {
+    this.pageRanges = new String[ranges.size()];
+    this.pageRanges = ranges.toArray(this.pageRanges);
+  }
+
   public void setBackground(boolean background) {
     this.background = Require.nonNull("background", background);
   }

diff --git a/java/test/org/openqa/selenium/print/PrintOptionsTest.java b/java/test/org/openqa/selenium/print/PrintOptionsTest.java
@@ -19,6 +19,7 @@
 
 import static org.assertj.core.api.Assertions.assertThat;
 
+import java.util.List;
 import java.util.Map;
 import org.junit.jupiter.api.Tag;
 import org.junit.jupiter.api.Test;
@@ -48,6 +49,14 @@ void setsValuesAsPassed() {
     assertThat(printOptions.getShrinkToFit()).isFalse();
   }
 
+  @Test
+  void setsPageRanges() {
+    PrintOptions printOptions = new PrintOptions();
+
+    printOptions.setPageRanges(List.of("1-2", "6-7"));
+    assertThat(printOptions.getPageRanges().length).isEqualTo(2);
+  }
+
   @Test
   void toMapContainsProperKey() {
     PrintOptions printOptions = new PrintOptions();