1 files changed, 83 insertions, 0 deletions
diff --git a/container-compiler-plugin/src/test/java/io/trygvis/container/compiler/ProcessorTest.java b/container-compiler-plugin/src/test/java/io/trygvis/container/compiler/ProcessorTest.java
new file mode 100644
index 0000000..e070f3c
--- /dev/null
+++ b/container-compiler-plugin/src/test/java/io/trygvis/container/compiler/ProcessorTest.java
@@ -0,0 +1,83 @@
+package io.trygvis.container.compiler;
+
+import org.apache.commons.io.IOUtils;
+import org.testng.annotations.Test;
+
+import javax.tools.Diagnostic;
+import javax.tools.DiagnosticCollector;
+import javax.tools.JavaCompiler;
+import javax.tools.JavaFileObject;
+import javax.tools.SimpleJavaFileObject;
+import javax.tools.StandardJavaFileManager;
+import javax.tools.ToolProvider;
+import java.net.URI;
+import java.nio.charset.Charset;
+import java.util.Arrays;
+import java.util.Locale;
+
+import static java.util.Collections.singletonList;
+import static org.fest.assertions.Assertions.assertThat;
+
+public class ProcessorTest {
+    Charset UTF_8 = Charset.forName("utf-8");
+
+    /**
+     * A file object used to represent source coming from a string.
+     */
+    public class JavaSourceFromString extends SimpleJavaFileObject {
+        /**
+         * The source code of this "file".
+         */
+        final String code;
+
+        /**
+         * Constructs a new JavaSourceFromString.
+         *
+         * @param name the name of the compilation unit represented by this file object
+         * @param code the source code for the compilation unit represented by this file object
+         */
+        JavaSourceFromString(String name, String code) {
+            super(URI.create("string:///" + name.replace('.', '/') + JavaFileObject.Kind.SOURCE.extension),
+                    JavaFileObject.Kind.SOURCE);
+            this.code = code;
+        }
+
+        @Override
+        public CharSequence getCharContent(boolean ignoreEncodingErrors) {
+            return code;
+        }
+    }
+
+    @Test
+    public void testBasic() throws Exception {
+        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
+
+        DiagnosticCollector<JavaFileObject> collector = new DiagnosticCollector<>();
+        StandardJavaFileManager standardFileManager = compiler.getStandardFileManager(collector, Locale.ENGLISH, UTF_8);
+
+        InMemoryJavaFileManager fileManager = new InMemoryJavaFileManager(standardFileManager);
+
+        JavaSourceFromString myEntity = new JavaSourceFromString("Person", IOUtils.toString(getClass().getResource("/Person.java"), UTF_8));
+
+        JavaCompiler.CompilationTask task = compiler.getTask(null, fileManager, collector, null,
+                null, singletonList(myEntity));
+
+        task.setProcessors(Arrays.asList(new MyProcessor()));
+
+        Boolean result = task.call();
+
+        for (Diagnostic<? extends JavaFileObject> diagnostic : collector.getDiagnostics()) {
+//            System.out.println("diagnostic = " + diagnostic);
+            System.out.println("diagnostic.source = ->" + diagnostic.getSource().getName() + "<-");
+            System.out.println("diagnostic.message = " + diagnostic.getMessage(Locale.ENGLISH));
+        }
+
+        assertThat(collector.getDiagnostics()).isEmpty();
+        assertThat(result).isTrue();
+
+        fileManager.close();
+
+        assertThat(fileManager.codes.keySet()).containsOnly("Person_Sql");
+        System.out.println(fileManager.codes.get("Person_Sql"));
+    }
+}