From c912d27d927e1e8e3aa0a0313d9dd3ecf75d7c92 Mon Sep 17 00:00:00 2001
From: Kilian Schuettler <kilian.schuettler@knecon.com>
Date: Wed, 19 Apr 2023 12:22:19 +0200
Subject: [PATCH] RED-6009: add direct json output test

---
 .../graph/DocumentGraphJsonWritingTest.java   | 38 +++++++++++++++++++
 .../graph/DocumentGraphVisualizationTest.java |  2 +-
 2 files changed, 39 insertions(+), 1 deletion(-)
 create mode 100644 layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphJsonWritingTest.java

diff --git a/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphJsonWritingTest.java b/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphJsonWritingTest.java
new file mode 100644
index 0000000..a9d1521
--- /dev/null
+++ b/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphJsonWritingTest.java
@@ -0,0 +1,38 @@
+package com.knecon.fforesight.service.layoutparser.server.graph;
+
+import java.io.File;
+import java.io.FileOutputStream;
+
+import org.junit.jupiter.api.Disabled;
+import org.junit.jupiter.api.Test;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.iqser.red.commons.jackson.ObjectMapperFactory;
+import com.knecon.fforesight.service.layoutparser.internal.api.data.DocumentData;
+import com.knecon.fforesight.service.layoutparser.internal.api.graph.DocumentGraph;
+import com.knecon.fforesight.service.layoutparser.internal.api.mapper.DocumentDataMapper;
+
+import lombok.SneakyThrows;
+
+public class DocumentGraphJsonWritingTest extends BuildDocumentGraphTest {
+
+    @Test
+    @Disabled
+    public void writeJsonForFileTest() {
+
+        writeJsons("files/S-Metolachlor_RAR_01_Volume_1_2018-09-06");
+    }
+
+    @SneakyThrows
+    private void writeJsons(String filename) {
+
+        DocumentGraph documentGraph = buildGraph(filename);
+        DocumentData documentData = DocumentDataMapper.toDocumentData(documentGraph);
+        ObjectMapper mapper = ObjectMapperFactory.create();
+        mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_structure", "json")), documentData.getTableOfContents());
+        mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_text", "json")), documentData.getTableOfContents());
+        mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_positions", "json")), documentData.getTableOfContents());
+        mapper.writeValue(new FileOutputStream(File.createTempFile(filename + "_pages", "json")), documentData.getTableOfContents());
+    }
+
+}
diff --git a/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java b/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java
index 27a825c..b8f5d5c 100644
--- a/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java
+++ b/layoutparser-service/layoutparser-service-server/src/test/java/com/knecon/fforesight/service/layoutparser/server/graph/DocumentGraphVisualizationTest.java
@@ -33,7 +33,7 @@ public class DocumentGraphVisualizationTest extends BuildDocumentGraphTest {
     @Disabled
     public void visualizeRotatedTestDocument() {
 
-        String filename = "files/RotateTestFileWithImages";
+        String filename = "files/211";
         visualizePdf(filename);
     }