add test case for order as column name

bomeng · bomeng · commit 715777adff92 · 2016-04-14T19:57:54.000-07:00
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/jdbc/JDBCWriteSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/jdbc/JDBCWriteSuite.scala
@@ -54,6 +54,9 @@ class JDBCWriteSuite extends SharedSQLContext with BeforeAndAfter {
     conn1.prepareStatement(
       "create table test.people1 (name TEXT(32) NOT NULL, `the id` INTEGER NOT NULL)")
       .executeUpdate()
+    conn1.prepareStatement(
+      "create table test.orders (`order` TEXT(32) NOT NULL, `order id` INTEGER NOT NULL)")
+      .executeUpdate()
     conn1.commit()
 
     sql(
@@ -69,6 +72,13 @@ class JDBCWriteSuite extends SharedSQLContext with BeforeAndAfter {
         |USING org.apache.spark.sql.jdbc
         |OPTIONS (url '$url1', dbtable 'TEST.PEOPLE1', user 'testUser', password 'testPass')
       """.stripMargin.replaceAll("\n", " "))
+
+    sql(
+      s"""
+         |CREATE TEMPORARY TABLE ORDERS
+         |USING org.apache.spark.sql.jdbc
+         |OPTIONS (url '$url1', dbtable 'TEST.ORDERS', user 'testUser', password 'testPass')
+      """.stripMargin.replaceAll("\n", " "))
   }
 
   after {
@@ -157,5 +167,8 @@ class JDBCWriteSuite extends SharedSQLContext with BeforeAndAfter {
     val df = sqlContext.createDataFrame(sparkContext.parallelize(arr2x2), schema2)
     df.write.insertInto("PEOPLE1")
     assert(2 === sqlContext.read.jdbc(url1, "TEST.PEOPLE1", properties).count)
+
+    df.write.insertInto("ORDERS")
+    assert(2 === sqlContext.read.jdbc(url1, "TEST.ORDERS", properties).count)
   }
 }
