1 files changed, 27 insertions, 0 deletions
diff --git a/container-compiler-plugin/src/test/java/io/trygvis/container/compiler/EntityHandlerTest.java b/container-compiler-plugin/src/test/java/io/trygvis/container/compiler/EntityHandlerTest.java
new file mode 100644
index 0000000..98fcaea
--- /dev/null
+++ b/container-compiler-plugin/src/test/java/io/trygvis/container/compiler/EntityHandlerTest.java
@@ -0,0 +1,27 @@
+package io.trygvis.container.compiler;
+
+import org.testng.annotations.DataProvider;
+import org.testng.annotations.Test;
+
+import static io.trygvis.container.compiler.EntityHandler.sqlName;
+import static org.fest.assertions.Assertions.assertThat;
+
+@Test(singleThreaded = false)
+public class EntityHandlerTest {
+
+    @DataProvider(name = "sqlName", parallel = true)
+    public static Object[][] SqlNameDataProvider() {
+        return new Object[][] {
+                new Object[]{"MyClass", "my_class"},
+                new Object[]{"myField", "my_field"},
+                new Object[]{"name", "name"},
+                new Object[]{"first_name", "first_name"},
+                new Object[]{"first_name_", "first_name_"},
+        };
+    }
+
+    @Test(dataProvider = "sqlName")
+    public void testSqlName(String input, String output) throws Exception {
+        assertThat(sqlName(input)).isEqualTo(output);
+    }
+}