From 92145da2592cf08a0719361301d0a28d459251d9 Mon Sep 17 00:00:00 2001
From: Izaac <izaac.carcenac-sautron@iut-tlse3.fr>
Date: Mon, 30 Sep 2024 13:56:22 +0200
Subject: [PATCH] ADD: test to check if the issue
 https://github.com/kestra-io/plugin-serdes/issues/138 is handled correctly

---
 .../plugin/serdes/json/IonToJsonTest.java     | 84 +++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 src/test/java/io/kestra/plugin/serdes/json/IonToJsonTest.java

diff --git a/src/test/java/io/kestra/plugin/serdes/json/IonToJsonTest.java b/src/test/java/io/kestra/plugin/serdes/json/IonToJsonTest.java
new file mode 100644
index 0000000..8c8056f
--- /dev/null
+++ b/src/test/java/io/kestra/plugin/serdes/json/IonToJsonTest.java
@@ -0,0 +1,84 @@
+package io.kestra.plugin.serdes.json;
+
+import io.kestra.core.junit.annotations.KestraTest;
+import io.kestra.core.runners.RunContext;
+import io.kestra.core.runners.RunContextFactory;
+import io.kestra.core.storages.StorageInterface;
+import io.kestra.core.utils.IdUtils;
+import jakarta.inject.Inject;
+import static org.hamcrest.Matchers.is;
+import static org.hamcrest.MatcherAssert.assertThat;
+
+import static org.junit.jupiter.api.Assertions.fail;
+
+import java.io.ByteArrayInputStream;
+import java.io.InputStream;
+import java.net.URI;
+import java.nio.charset.StandardCharsets;
+import java.util.HashMap;
+import java.util.Map;
+
+import org.junit.jupiter.api.Test;
+
+import com.google.common.collect.ImmutableMap;
+
+@KestraTest
+public class IonToJsonTest {
+    @Inject
+    private RunContextFactory runContextFactory;
+
+    @Inject
+    private StorageInterface storageInterface;
+
+    private RunContext getContext(String content) {
+        Map<String, String> kestraPath = new HashMap<>();
+        URI filePath;
+        try {
+            filePath = storageInterface.put(
+                null,
+                URI.create("/" + IdUtils.create() + ".ion"),
+                new ByteArrayInputStream(content.getBytes())
+            );
+            kestraPath.put("file", filePath.toString());
+        } catch (Exception e) {
+            System.err.println(e.getMessage());
+            fail("Unable to load input file.");
+            return null;
+        }
+        return runContextFactory.of(ImmutableMap.copyOf(kestraPath));
+    }
+
+    private void assertEquality(String expected, URI file) {
+        assertThat("Result file should exist", storageInterface.exists(null, file), is(true));
+        try (InputStream streamResult = storageInterface.get(null, file)) {
+            String result = new String(streamResult.readAllBytes(), StandardCharsets.UTF_8).replace("\r\n", "\n");
+            System.out.println("Got :\n" + result);
+            System.out.println("Expecting :\n" + expected);
+            assertThat("Result should match the reference", result.equals(expected));
+        } catch (Exception e) {
+            System.err.println(e.getMessage());
+            fail("Unable to load results files.");
+        }
+    }
+
+    @Test
+    void test_annotation_conversion() throws Exception {
+        String input = """
+            {dn:"cn=tony@orga.com,ou=diffusion_list,dc=orga,dc=com",attributes:{description:["Some description 2",base64::"TGlzdGUgZCfDg8KpY2hhbmdlIHN1ciBsZSBzdWl2aSBkZSBsYSBtYXNzZSBzYWxhcmlhbGUgZGUgbCdJVVQ=","Melusine lover as well"],someOtherAttribute:["perhaps 2","perhapsAgain 2"]}}
+            """;
+        // Expectated result when should_keep_annotations==False | or not specified
+        String expectation_removed_annot = """
+            {"dn":"cn=tony@orga.com,ou=diffusion_list,dc=orga,dc=com","attributes":{"description":["Some description 2","TGlzdGUgZCfDg8KpY2hhbmdlIHN1ciBsZSBzdWl2aSBkZSBsYSBtYXNzZSBzYWxhcmlhbGUgZGUgbCdJVVQ=","Melusine lover as well"],"someOtherAttribute":["perhaps 2","perhapsAgain 2"]}}
+            """;
+        // Expectated result when should_keep_annotations==True
+        // String expectation_indicated_annot = """
+        //     {"dn":"cn=tony@orga.com,ou=diffusion_list,dc=orga,dc=com","attributes":{"description":["Some description 2",{"ion_annotations":["base64"], "value":"TGlzdGUgZCfDg8KpY2hhbmdlIHN1ciBsZSBzdWl2aSBkZSBsYSBtYXNzZSBzYWxhcmlhbGUgZGUgbCdJVVQ="},"Melusine lover as well"],"someOtherAttribute":["perhaps 2","perhapsAgain 2"]}}
+        //     """;
+
+        RunContext runContext = getContext(input);
+        IonToJson task = IonToJson.builder().from("{{file}}").build();//TODO: create two executions, with boolean "should_keep_annotations"
+        IonToJson.Output output = task.run(runContext);
+        assertEquality(expectation_removed_annot, output.getUri());//by defaults, annot shouldn't be kept since it's the default behaviour of the ION cookbook https://amazon-ion.github.io/ion-docs/guides/cookbook.html, see "Down-converting to JSON" section
+        //assertEquality(expectation_indicated_annot, output.getUri());//TODO: if should_keep_annotations == true
+    }
+}
\ No newline at end of file