add test case for #567

apache · Jan 17, 2018 · 35f73ba · 35f73ba
1 parent 06c1979
commit 35f73ba
Showing 1 changed file with 25 additions and 1 deletion.
diff --git a/sharding-jdbc-core/src/test/java/io/shardingjdbc/core/util/SQLUtilTest.java b/sharding-jdbc-core/src/test/java/io/shardingjdbc/core/util/SQLUtilTest.java
@@ -17,6 +17,7 @@
 
 package io.shardingjdbc.core.util;
 
+import io.shardingjdbc.core.constant.DatabaseType;
 import org.junit.Test;
 
 import static org.hamcrest.core.Is.is;
@@ -25,10 +26,33 @@
 public class SQLUtilTest {
 
     @Test
-    public void assertGetExactlyValue() throws Exception {
+    public void assertGetExactlyValue() {
         assertThat(SQLUtil.getExactlyValue("`xxx`"), is("xxx"));
         assertThat(SQLUtil.getExactlyValue("[xxx]"), is("xxx"));
         assertThat(SQLUtil.getExactlyValue("\"xxx\""), is("xxx"));
         assertThat(SQLUtil.getExactlyValue("'xxx'"), is("xxx"));
     }
+
+    @Test
+    public void assertGetOriginalValueForOtherDatabase() {
+        assertThat(SQLUtil.getOriginalValue("select", DatabaseType.H2), is("select"));
+        assertThat(SQLUtil.getOriginalValue("select", DatabaseType.Oracle), is("select"));
+        assertThat(SQLUtil.getOriginalValue("select", DatabaseType.SQLServer), is("select"));
+        assertThat(SQLUtil.getOriginalValue("select", DatabaseType.PostgreSQL), is("select"));
+    }
+
+    @Test
+    public void assertGetOriginalValueForMySQLWithoutKeyword() {
+        assertThat(SQLUtil.getOriginalValue("test", DatabaseType.MySQL), is("test"));
+    }
+
+    @Test
+    public void assertGetOriginalValueForMySQLWithDefaultKeyword() {
+        assertThat(SQLUtil.getOriginalValue("select", DatabaseType.MySQL), is("`select`"));
+    }
+
+    @Test
+    public void assertGetOriginalValueForMySQLWithMySQLKeyword() {
+        assertThat(SQLUtil.getOriginalValue("show", DatabaseType.MySQL), is("`show`"));
+    }
 }