BigQueryTableExistenceSensor needs to specify keyword arguments (#9832)

nathadfield · web-flow · commit 770de53eb57b · 2020-07-15T20:49:50.000+02:00
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
         )

Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,7 @@ def test_passing_arguments_to_hook(self, mock_hook):`
`47`	`47`	`delegate_to=TEST_DELEGATE_TO`
`48`	`48`	`)`
`49`	`49`	`mock_hook.return_value.table_exists.assert_called_once_with(`
`50`		`- TEST_PROJECT_ID,`
`51`		`- TEST_DATASET_ID,`
`52`		`- TEST_TABLE_ID`
	`50`	`+ project_id=TEST_PROJECT_ID,`
	`51`	`+ dataset_id=TEST_DATASET_ID,`
	`52`	`+ table_id=TEST_TABLE_ID`
`53`	`53`	`)`