Add exportContext.offload flag to CLOUD_SQL_EXPORT_VALIDATION. (#23614)

Author: ishiis

airflow/providers/google/cloud/example_dags/example_cloud_sql.py

@@ -123,7 +123,12 @@
 # [END howto_operator_cloudsql_patch_body]
 # [START howto_operator_cloudsql_export_body]
 export_body = {
-    "exportContext": {"fileType": "sql", "uri": EXPORT_URI, "sqlExportOptions": {"schemaOnly": False}}
+    "exportContext": {
+        "fileType": "sql",
+        "uri": EXPORT_URI,
+        "sqlExportOptions": {"schemaOnly": False},
+        "offload": True,
+    }
 }
 # [END howto_operator_cloudsql_export_body]
 # [START howto_operator_cloudsql_import_body]

airflow/providers/google/cloud/operators/cloud_sql.py

@@ -154,6 +154,7 @@
                 ],
             ),
             dict(name="csvExportOptions", type="dict", optional=True, fields=[dict(name="selectQuery")]),
+            dict(name="offload", optional=True),
         ],
     )
 ]

tests/providers/google/cloud/operators/test_cloud_sql.py

@@ -123,6 +123,7 @@
         "databases": [],
         "sqlExportOptions": {"tables": ["table1", "table2"], "schemaOnly": False},
         "csvExportOptions": {"selectQuery": "SELECT * FROM TABLE"},
+        "offload": True,
     }
 }
 IMPORT_BODY = {