test_hive_to_druid.py 文件源码

def setUp(self):
        super(TestDruidHook, self).setUp()

        configuration.load_test_config()
        args = {
            'owner': 'airflow',
            'start_date': '2017-01-01'
        }
        self.dag = DAG('hive_to_druid', default_args=args)

        session = requests.Session()
        adapter = requests_mock.Adapter()
        session.mount('mock', adapter)