HIVE-29452: Split get table from HMSHandler (#6427)

Author: dengzhhu653

itests/hive-unit/src/test/java/org/apache/hadoop/hive/ql/TestDatabaseTableDefault.java

@@ -98,6 +98,7 @@ public void setUp() throws Exception {
         TestTxnDbUtil.prepDb(hiveConf);
 
         SessionState.start(new SessionState(hiveConf));
+        MetastoreConf.setBoolVar(hiveConf, ConfVars.HIVE_IN_TEST, false);
         d = DriverFactory.newDriver(hiveConf);
 
         wh = new File(System.getProperty("java.io.tmpdir") + File.separator +
@@ -110,7 +111,6 @@ public void setUp() throws Exception {
 
         MetastoreConf.setVar(hiveConf, ConfVars.METASTORE_METADATA_TRANSFORMER_CLASS,
                 "org.apache.hadoop.hive.metastore.MetastoreDefaultTransformer");
-        MetastoreConf.setBoolVar(hiveConf, ConfVars.HIVE_IN_TEST, false);
         MetastoreConf.setVar(hiveConf, ConfVars.WAREHOUSE, wh.getCanonicalPath());
         MetastoreConf.setVar(hiveConf, ConfVars.WAREHOUSE_EXTERNAL, ext_wh.getCanonicalPath());
         MetastoreConf.setBoolVar(hiveConf, MetastoreConf.ConfVars.CREATE_TABLES_AS_ACID, true);

itests/util/src/main/java/org/apache/hadoop/hive/ql/QTestUtil.java

@@ -55,6 +55,10 @@
 import org.apache.hadoop.hive.common.io.QTestFetchConverter;
 import org.apache.hadoop.hive.conf.HiveConf;
 import org.apache.hadoop.hive.conf.HiveConf.ConfVars;
+import org.apache.hadoop.hive.metastore.api.Database;
+import org.apache.hadoop.hive.metastore.api.DatabaseType;
+import org.apache.hadoop.hive.metastore.api.GetDatabaseObjectsRequest;
+import org.apache.hadoop.hive.metastore.api.GetDatabaseObjectsResponse;
 import org.apache.hadoop.hive.ql.metadata.HiveMetaStoreClientWithLocalCache;
 import org.apache.hadoop.hive.metastore.conf.MetastoreConf;
 import org.apache.hadoop.hive.ql.QTestMiniClusters.FsType;
@@ -355,6 +359,31 @@ public void clearUDFsCreatedDuringTests() throws Exception {
     }
   }
 
+  public void clearConnectorsCreatedInTests() throws Exception {
+    if (System.getenv(QTEST_LEAVE_FILES) != null) {
+      return;
+    }
+    conf.set("hive.metastore.filter.hook", "org.apache.hadoop.hive.metastore.DefaultMetaStoreFilterHookImpl");
+    db = Hive.get(conf);
+
+    GetDatabaseObjectsRequest request = new GetDatabaseObjectsRequest();
+    request.setPattern("*");
+    GetDatabaseObjectsResponse response = db.getMSC().get_databases_req(request);
+    if (response != null && response.getDatabasesSize() > 0) {
+      for (Database database : response.getDatabases()) {
+        if (database.getType() == DatabaseType.REMOTE) {
+          db.dropDatabase(database.getName(), true, true);
+        }
+      }
+    }
+    List<String> connectors = db.getAllDataConnectorNames();
+    if (connectors != null) {
+      for (String connectorName : connectors) {
+        db.dropDataConnector(connectorName, true);
+      }
+    }
+  }
+
   /**
    * Clear out any side effects of running tests
    */
@@ -482,6 +511,7 @@ public void clearTestSideEffects() throws Exception {
     Utilities.clearWorkMap(conf);
     NotificationEventPoll.shutdown();
     QueryResultsCache.cleanupInstance();
+    clearConnectorsCreatedInTests();
     clearTablesCreatedDuringTests();
     clearUDFsCreatedDuringTests();
     clearKeysCreatedInTests();