Support to upload file to Google Shared Drive (#21319)

Author: lihan

airflow/providers/google/suite/hooks/drive.py

@@ -200,7 +200,7 @@ def upload_file(self, local_location: str, remote_location: str) -> str:
         media = MediaFileUpload(local_location)
         file = (
             service.files()
-            .create(body=file_metadata, media_body=media, fields="id")
+            .create(body=file_metadata, media_body=media, fields="id", supportsAllDrives=True)
             .execute(num_retries=self.num_retries)
         )
         self.log.info("File %s uploaded to gdrive://%s.", local_location, remote_location)

tests/providers/google/suite/hooks/test_drive.py

@@ -245,6 +245,7 @@ def test_upload_file_to_root_directory(
                     body={"name": "remote_path", "parents": ["root"]},
                     fields="id",
                     media_body=mock_media_file_upload.return_value,
+                    supportsAllDrives=True,
                 )
             ]
         )
@@ -274,6 +275,7 @@ def test_upload_file_to_subdirectory(
                     body={"name": "remote_path", "parents": ["PARENT_ID"]},
                     fields="id",
                     media_body=mock_media_file_upload.return_value,
+                    supportsAllDrives=True,
                 )
             ]
         )