def setUp(self): with mock.patch( "airflow.hooks.base_hook.BaseHook.get_connection") as conn: conn.return_value.extra_dejson = EXTRA self.hook = dataprep.GoogleDataprepHook( dataprep_conn_id="dataprep_default")
def mock_hook(): with mock.patch("airflow.hooks.base_hook.BaseHook.get_connection") as conn: hook = dataprep.GoogleDataprepHook(dataprep_conn_id="dataprep_conn_id") conn.return_value.extra_dejson = EXTRA yield hook