Safe mode unit test fixes

asciidoctor · Oct 1, 2020 · db8c0a2 · db8c0a2
1 parent 0213206
commit db8c0a2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/asciidoctorj-diagram/src/test/java/org.asciidoctor/WhenDocumentContainsDitaaDiagram.java b/asciidoctorj-diagram/src/test/java/org.asciidoctor/WhenDocumentContainsDitaaDiagram.java
@@ -15,14 +15,16 @@ public class WhenDocumentContainsDitaaDiagram {
     @Test
     public void png_should_be_rendered_for_diagram() {
 
-        File inputFile = new File("build/resources/test/sample.adoc");
+        File buildDir = new File("build/resources/test");
+
+        File inputFile = new File(buildDir, "sample.adoc");
         File outputFile1 = new File(inputFile.getParentFile(), "asciidoctor-diagram-process.png");
         File outputFile2 = new File(inputFile.getParentFile(), ".asciidoctor/diagram/asciidoctor-diagram-process.png.cache");
         asciidoctor.requireLibrary("asciidoctor-diagram");
         asciidoctor.convertFile(inputFile,
             options()
                 .backend("html5")
-                .toDir(new File("build"))
+                .toFile(new File(buildDir, "sample.html"))
                 .get());
         assertThat(outputFile1.exists(), is(true));
         assertThat(outputFile2.exists(), is(true));

diff --git a/asciidoctorj-diagram/src/test/java/org.asciidoctor/WhenDocumentWantsDataUris.java b/asciidoctorj-diagram/src/test/java/org.asciidoctor/WhenDocumentWantsDataUris.java
@@ -6,6 +6,9 @@
 import org.junit.Test;
 
 import java.io.File;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
 
 import static org.asciidoctor.OptionsBuilder.options;
 import static org.hamcrest.CoreMatchers.is;
@@ -18,23 +21,25 @@ public class WhenDocumentWantsDataUris {
     private Asciidoctor asciidoctor = Asciidoctor.Factory.create();
 
     @Test
-    public void png_should_be_rendered_for_diagram() {
+    public void png_should_be_rendered_for_diagram() throws IOException {
+        File buildDir = new File("build/resources/test");
 
-        File inputFile = new File("build/resources/test/data-uri.adoc");
+        File inputFile = new File(buildDir, "data-uri.adoc");
+        File outputFile = new File(buildDir, "data-uri.html");
         File outputFile1 = new File(inputFile.getParentFile(), "data-uri-test.png");
         File outputFile2 = new File(inputFile.getParentFile(), ".asciidoctor/diagram/data-uri-test.png.cache");
         asciidoctor.requireLibrary("asciidoctor-diagram");
-        final String html = asciidoctor.convertFile(inputFile,
+        asciidoctor.convertFile(inputFile,
             options().backend("html5")
-                .toFile(false)
-                .toDir(new File("build"))
+                .toFile(outputFile)
                 .safe(SafeMode.SERVER)
                 .get());
         assertThat(outputFile1.exists(), is(true));
         assertThat(outputFile2.exists(), is(true));
         outputFile1.delete();
         outputFile2.delete();
 
+        String html = new String(Files.readAllBytes(outputFile.toPath()), StandardCharsets.UTF_8);
         final Document doc = Jsoup.parse(html);
         System.out.println(doc);
         Elements images = doc.getElementsByTag("img");