def test_get_conn(self): with mock.patch('airflow.providers.apache.hive.hooks.hive.HiveMetastoreHook._find_valid_server' ) as find_valid_server: find_valid_server.return_value = mock.MagicMock(return_value={}) metastore_hook = HiveMetastoreHook() self.assertIsInstance(metastore_hook.get_conn(), HMSClient)
def tearDown(self): hook = HiveMetastoreHook() with hook.get_conn() as metastore: metastore.drop_table(self.database, self.table, deleteData=True)