BigQueryTableExistenceSensor needs to specify keyword arguments (#9832)

apache · Jul 15, 2020 · 770de53 · 770de53
1 parent 4aca72e
commit 770de53
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/airflow/providers/google/cloud/sensors/bigquery.py b/airflow/providers/google/cloud/sensors/bigquery.py
@@ -71,4 +71,7 @@ def poke(self, context):
  hook = BigQueryHook(
  bigquery_conn_id=self.bigquery_conn_id,
  delegate_to=self.delegate_to)
- return hook.table_exists(self.project_id, self.dataset_id, self.table_id)
+ return hook.table_exists(
+ project_id=self.project_id,
+ dataset_id=self.dataset_id,
+ table_id=self.table_id)
diff --git a/tests/providers/google/cloud/sensors/test_bigquery.py b/tests/providers/google/cloud/sensors/test_bigquery.py
@@ -47,7 +47,7 @@ def test_passing_arguments_to_hook(self, mock_hook):
  delegate_to=TEST_DELEGATE_TO
  )
  mock_hook.return_value.table_exists.assert_called_once_with(
- TEST_PROJECT_ID,
- TEST_DATASET_ID,
- TEST_TABLE_ID
+ project_id=TEST_PROJECT_ID,
+ dataset_id=TEST_DATASET_ID,
+ table_id=TEST_TABLE_ID
  )