Remove deprecated template_fields from GoogleDriveToGCSOperator (#19991)

bradleybonitatibus · web-flow · commit cb082d361a61 · 2021-12-02T22:25:33.000Z
This prevented `op.dry_run()` from working.
diff --git a/airflow/providers/google/cloud/transfers/gdrive_to_gcs.py b/airflow/providers/google/cloud/transfers/gdrive_to_gcs.py
@@ -67,8 +67,6 @@ class GoogleDriveToGCSOperator(BaseOperator):
     template_fields = [
         "bucket_name",
         "object_name",
-        "destination_bucket",
-        "destination_object",
         "folder_id",
         "file_name",
         "drive_id",
diff --git a/tests/providers/google/cloud/transfers/test_gdrive_to_gcs.py b/tests/providers/google/cloud/transfers/test_gdrive_to_gcs.py
@@ -59,3 +59,5 @@ def test_execute(self, mock_gdrive_hook, mock_gcs_hook):
         mock_gcs_hook.return_value.provide_file_and_upload.assert_called_once_with(
             bucket_name=BUCKET, object_name=OBJECT
         )
+
+        assert op.dry_run() is None

Original file line number	Diff line number	Diff line change
`@@ -59,3 +59,5 @@ def test_execute(self, mock_gdrive_hook, mock_gcs_hook):`
`59`	`59`	`mock_gcs_hook.return_value.provide_file_and_upload.assert_called_once_with(`
`60`	`60`	`bucket_name=BUCKET, object_name=OBJECT`
`61`	`61`	`)`
	`62`	`+`
	`63`	`+ assert op.dry_run() is None`