Python HiveMetastoreHook.get_conn Beispiele

Beispiel #1

    def test_get_conn(self):
        with patch('airflow.hooks.hive_hooks.HiveMetastoreHook._find_valid_server') \
                as find_valid_server:
            find_valid_server.return_value = MagicMock(return_value={})
            metastore_hook = HiveMetastoreHook()

        self.assertIsInstance(metastore_hook.get_conn(), HMSClient)

Beispiel #2

Datei: test_hive_hook.py Projekt: yanxiaofei395118/airflow

 def tearDown(self):
     hook = HiveMetastoreHook()
     with hook.get_conn() as metastore:
         metastore.drop_table(self.database, self.table, deleteData=True)

Beispiel #3

Datei: test_hive_hook.py Projekt: AdamUnger/incubator-airflow

 def tearDown(self):
     hook = HiveMetastoreHook()
     with hook.get_conn() as metastore:
         metastore.drop_table(self.database, self.table, deleteData=True)