Wait insert_job result in normal mode (#29925)

hussein-awala · web-flow · commit 464ab1b7caa7 · 2023-03-05T11:40:38.000+01:00
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")

Original file line number	Diff line number	Diff line change
`@@ -197,7 +197,7 @@ def _submit_job(`
`197`	`197`	`job_id=job_id,`
`198`	`198`	`timeout=self.result_timeout,`
`199`	`199`	`retry=self.result_retry,`
`200`		`- nowait=True,`
	`200`	`+ nowait=self.deferrable,`
`201`	`201`	`)`
`202`	`202`
`203`	`203`	`def execute(self, context: Context):`
Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,7 @@ def test_execute(self, mock_hook):`
`87`	`87`	`location=None,`
`88`	`88`	`timeout=None,`
`89`	`89`	`retry=DEFAULT_RETRY,`
`90`		`- nowait=True,`
	`90`	`+ nowait=False,`
`91`	`91`	`)`
`92`	`92`
`93`	`93`	`@mock.patch("airflow.providers.google.cloud.transfers.bigquery_to_gcs.BigQueryHook")`