Wait insert_job result in normal mode (#29925)

apache · Mar 5, 2023 · 464ab1b · 464ab1b
1 parent 30b2e6c
commit 464ab1b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/airflow/providers/google/cloud/transfers/bigquery_to_gcs.py b/airflow/providers/google/cloud/transfers/bigquery_to_gcs.py
@@ -197,7 +197,7 @@ def _submit_job(
  job_id=job_id,
  timeout=self.result_timeout,
  retry=self.result_retry,
- nowait=True,
+ nowait=self.deferrable,
  )
 
  def execute(self, context: Context):

diff --git a/tests/providers/google/cloud/transfers/test_bigquery_to_gcs.py b/tests/providers/google/cloud/transfers/test_bigquery_to_gcs.py
@@ -87,7 +87,7 @@ def test_execute(self, mock_hook):
  location=None,
  timeout=None,
  retry=DEFAULT_RETRY,
- nowait=True,
+ nowait=False,
  )
 
  @mock.patch("airflow.providers.google.cloud.transfers.bigquery_to_gcs.BigQueryHook")