fix ut

Author: yabola

extensions/spark/kyuubi-spark-connector-hive/src/test/scala/org/apache/kyuubi/spark/connector/hive/command/PartitionManagementSuite.scala

@@ -41,26 +41,6 @@ trait PartitionManagementSuite extends DDLCommandTestUtils {
     }
   }
 
-  test("create partition with location") {
-    withNamespaceAndTable("ns", "tbl") { t =>
-      sql(s"CREATE TABLE $t (id string, year string, month string) PARTITIONED BY (year, month)")
-      val loc = "file:///tmp/kyuubi/hive_catalog_part_loc"
-      sql(s"ALTER TABLE $t ADD PARTITION (year='2023', month='01') LOCATION '$loc'")
-      checkAnswer(
-        sql(s"SHOW PARTITIONS $t"),
-        Row("year=2023/month=01") :: Nil)
-      val catalog = spark.sessionState.catalogManager
-        .catalog(catalogName).asInstanceOf[TableCatalog]
-      val partManagement = catalog.loadTable(Identifier.of(Array("ns"), "tbl"))
-        .asInstanceOf[SupportsPartitionManagement]
-      val partIdent = InternalRow.fromSeq(
-        Seq(UTF8String.fromString("2023"), UTF8String.fromString("01")))
-      val metadata = partManagement.loadPartitionMetadata(partIdent)
-      assert(metadata.containsKey("location"))
-      assert(metadata.get("location").contains("hive_catalog_part_loc"))
-    }
-  }
-
   test("drop partition") {
     withNamespaceAndTable("ns", "tbl") { t =>
       sql(s"CREATE TABLE $t (id string, year string, month string) PARTITIONED BY (year, month)")
@@ -111,6 +91,26 @@ trait PartitionManagementSuite extends DDLCommandTestUtils {
 class PartitionManagementV2Suite extends PartitionManagementSuite {
   override protected def catalogVersion: String = "Hive V2"
   override protected def commandVersion: String = V2_COMMAND_VERSION
+
+  test("create partition with location") {
+    withNamespaceAndTable("ns", "tbl") { t =>
+      sql(s"CREATE TABLE $t (id string, year string, month string) PARTITIONED BY (year, month)")
+      val loc = "file:///tmp/kyuubi/hive_catalog_part_loc"
+      sql(s"ALTER TABLE $t ADD PARTITION (year='2023', month='01') LOCATION '$loc'")
+      checkAnswer(
+        sql(s"SHOW PARTITIONS $t"),
+        Row("year=2023/month=01") :: Nil)
+      val catalog = spark.sessionState.catalogManager
+        .catalog(catalogName).asInstanceOf[TableCatalog]
+      val partManagement = catalog.loadTable(Identifier.of(Array("ns"), "tbl"))
+        .asInstanceOf[SupportsPartitionManagement]
+      val partIdent = InternalRow.fromSeq(
+        Seq(UTF8String.fromString("2023"), UTF8String.fromString("01")))
+      val metadata = partManagement.loadPartitionMetadata(partIdent)
+      assert(metadata.containsKey("location"))
+      assert(metadata.get("location").contains("hive_catalog_part_loc"))
+    }
+  }
 }
 
 class PartitionManagementV1Suite extends PartitionManagementSuite {