Add unit tests for new batch operation classes

Author: andreasrosdal

openpdf-core/src/test/java/org/openpdf/text/pdf/PdfBatchUtilsTest.java

@@ -0,0 +1,85 @@
+package org.openpdf.text.pdf;
+
+import org.junit.jupiter.api.Test;
+import org.openpdf.text.Document;
+import org.openpdf.text.Paragraph;
+
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.List;
+import java.util.concurrent.Callable;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class PdfBatchUtilsTest {
+
+    private static Path tinyPdf(String prefix) throws Exception {
+        Path p = Files.createTempFile(prefix, ".pdf");
+        var doc = new Document();
+        try (var out = new FileOutputStream(p.toFile())) {
+            PdfWriter.getInstance(doc, out);
+            doc.open();
+            doc.add(new Paragraph("Hello OpenPDF"));
+            doc.close();
+        }
+        return p;
+    }
+
+
+    @Test
+    void runBatch_usesVirtualThreads() {
+        // We don't create any executors here; runBatch does that internally.
+        var tasks = List.of(
+                (Callable<Integer>) () -> {
+                    assertTrue(Thread.currentThread().isVirtual(), "Task should run on a virtual thread");
+                    return 1;
+                },
+                (Callable<Integer>) () -> {
+                    assertTrue(Thread.currentThread().isVirtual(), "Task should run on a virtual thread");
+                    return 2;
+                },
+                (Callable<Integer>) () -> {
+                    assertTrue(Thread.currentThread().isVirtual(), "Task should run on a virtual thread");
+                    return 3;
+                }
+        );
+
+        var result = PdfBatchUtils.runBatch(tasks, v -> {}, t -> fail(t));
+        assertTrue(result.isAllSuccessful(), "All tasks should succeed");
+        assertEquals(3, result.successes.size());
+        assertEquals(0, result.failures.size());
+    }
+
+    @Test
+    void batchMerge_createsOutput_and_runsOnVirtualThreads() throws Exception {
+        // Prepare inputs
+        Path a = tinyPdf("a-");
+        Path b = tinyPdf("b-");
+        Path merged = Files.createTempFile("merged-", ".pdf");
+
+        // Use the batch API which internally uses virtual threads.
+        var jobs = List.of(new PdfBatchUtils.MergeJob(List.of(a, b), merged));
+
+        var result = PdfBatchUtils.batchMerge(jobs,
+                // onSuccess: simple sanity checks per job
+                out -> {
+                    assertTrue(Files.exists(out), "Merged PDF should exist");
+                    try {
+                        assertTrue(Files.size(out) > 0, "Merged PDF should be non-empty");
+                    } catch (IOException e) {
+                        throw new RuntimeException(e);
+                    }
+                },
+                t -> fail(t)
+        );
+
+        assertTrue(result.isAllSuccessful(), "Merge job should succeed");
+
+        // Cleanup
+        Files.deleteIfExists(a);
+        Files.deleteIfExists(b);
+        Files.deleteIfExists(merged);
+    }
+}

openpdf-html/src/main/java/org/openpdf/html/Html2PdfBatchUtils.java

@@ -167,7 +167,7 @@ public static Path renderHtmlString(String html, String baseUri, Path output,
             ITextRenderer renderer = new ITextRenderer();
             SharedContext sc = renderer.getSharedContext();
             // Slightly safer resource loading if you need custom schemes:
-            //sc.setUserAgentCallback(renderer.getOutputDevice().getUserAgentCallback());
+            sc.setUserAgentCallback(renderer.getOutputDevice().getSharedContext().getUserAgentCallback());
 
             if (rendererCustomizer != null) rendererCustomizer.accept(renderer);
 

openpdf-html/src/test/java/org/openpdf/html/Html2PdfBatchUtilsTest.java

@@ -0,0 +1,55 @@
+package org.openpdf.html;
+
+import org.junit.jupiter.api.Test;
+
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.util.List;
+import java.util.Optional;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class Html2PdfBatchUtilsTest {
+
+    @Test
+    void testBatchHtmlStringsRunsOnVirtualThreads() throws Exception {
+        String html = "<html><body><h1>Hello PDF Batch test</h1></body></html>";
+
+        Path out1 = Files.createTempFile("vt-batch-1-", ".pdf");
+        Path out2 = Files.createTempFile("vt-batch-2-", ".pdf");
+        Path out3 = Files.createTempFile("vt-batch-3-", ".pdf");
+
+        // Assert virtual-thread execution from inside each batch task.
+        var vtAssertCustomizer = Html2PdfBatchUtils.setDpi(96).andThen(renderer ->
+                assertTrue(Thread.currentThread().isVirtual(), "Batch task should run on a virtual thread")
+        );
+
+        var jobs = List.of(
+                new Html2PdfBatchUtils.HtmlStringJob(
+                        html, null, out1,
+                        Optional.of(Html2PdfBatchUtils.CSS_A4_20MM),
+                        Optional.of(vtAssertCustomizer)
+                ),
+                new Html2PdfBatchUtils.HtmlStringJob(
+                        html, null, out2,
+                        Optional.of(Html2PdfBatchUtils.CSS_A4_20MM),
+                        Optional.of(vtAssertCustomizer)
+                ),
+                new Html2PdfBatchUtils.HtmlStringJob(
+                        html, null, out3,
+                        Optional.of(Html2PdfBatchUtils.CSS_A4_20MM),
+                        Optional.of(vtAssertCustomizer)
+                )
+        );
+
+        var result = Html2PdfBatchUtils.batchHtmlStrings(jobs, p -> {}, t -> fail(t));
+
+        assertTrue(result.isAllSuccessful(), "All batch jobs should succeed");
+
+        for (Path p : List.of(out1, out2, out3)) {
+            assertTrue(Files.exists(p), "PDF should exist: " + p);
+            assertTrue(Files.size(p) > 0, "PDF should be non-empty: " + p);
+            Files.deleteIfExists(p);
+        }
+    }
+}