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