add gdrive bucket into tests

dlt-hub · Feb 7, 2024 · 950cb06 · 950cb06
1 parent f80ef52
commit 950cb06
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/load/utils.py b/tests/load/utils.py
@@ -48,6 +48,7 @@
 AWS_BUCKET = dlt.config.get("tests.bucket_url_s3", str)
 GCS_BUCKET = dlt.config.get("tests.bucket_url_gs", str)
 AZ_BUCKET = dlt.config.get("tests.bucket_url_az", str)
+GDRIVE_BUCKET = dlt.config.get("tests.bucket_url_gdrive", str)
 FILE_BUCKET = dlt.config.get("tests.bucket_url_file", str)
 R2_BUCKET = dlt.config.get("tests.bucket_url_r2", str)
 MEMORY_BUCKET = dlt.config.get("tests.memory", str)
@@ -56,13 +57,14 @@
     "s3",
     "gs",
     "az",
+    "gdrive",
     "file",
     "memory",
     "r2",
 ]
 
 # Filter out buckets not in all filesystem drivers
-DEFAULT_BUCKETS = [GCS_BUCKET, AWS_BUCKET, FILE_BUCKET, MEMORY_BUCKET, AZ_BUCKET]
+DEFAULT_BUCKETS = [GCS_BUCKET, AWS_BUCKET, GDRIVE_BUCKET, FILE_BUCKET, MEMORY_BUCKET, AZ_BUCKET]
 DEFAULT_BUCKETS = [
     bucket for bucket in DEFAULT_BUCKETS if bucket.split(":")[0] in ALL_FILESYSTEM_DRIVERS
 ]