tmp

Author: marchpure

integration/spark/src/test/scala/org/apache/carbondata/spark/testsuite/createTable/TestCreateTableIfNotExists.scala

@@ -17,12 +17,16 @@
 
 package org.apache.carbondata.spark.testsuite.createTable
 
-import java.util.concurrent.{Callable, Executors, ExecutorService, Future, TimeUnit}
+import org.apache.spark.sql.catalyst.TableIdentifier
+import org.apache.spark.sql.hive.CarbonRelation
 
-import org.apache.spark.sql.AnalysisException
+import java.util.concurrent.{Callable, ExecutorService, Executors, Future, TimeUnit}
+import org.apache.spark.sql.{AnalysisException, CarbonEnv}
 import org.apache.spark.sql.test.util.QueryTest
 import org.scalatest.BeforeAndAfterAll
 
+import java.io.File
+
 class TestCreateTableIfNotExists extends QueryTest with BeforeAndAfterAll {
 
   override def beforeAll {
@@ -44,6 +48,25 @@ class TestCreateTableIfNotExists extends QueryTest with BeforeAndAfterAll {
     }
   }
 
+  test("test create table with consistent table location") {
+    val dbName = "testdb"
+    val tblName = "testtbl"
+    sql(s"drop database if exists $dbName cascade")
+    val dblocation = warehouse + File.separator + dbName
+    sql(s"create database $dbName location '$dblocation'")
+    sql(s"use $dbName")
+    new File(warehouse + File.separator + tblName).mkdir()
+    sql(s"create table $tblName(a int, b string) STORED AS carbondata")
+    val tblIdentifier = new TableIdentifier(tblName)
+    val carbonTablePath = CarbonEnv.getInstance(sqlContext.sparkSession).carbonMetaStore
+      .lookupRelation(tblIdentifier)(sqlContext.sparkSession).carbonTable.getTablePath
+
+    val sparkTablePath =
+      sqlContext.sparkSession.sessionState.catalog
+        .getTableMetadata(tblIdentifier).storage.locationUri.get.getPath
+    assert(carbonTablePath.equals(sparkTablePath))
+  }
+
   test("test create table if not exist concurrently") {
 
     val executorService: ExecutorService = Executors.newFixedThreadPool(10)