test: add retry in pytest fixtures (#1108)

googleapis · Aug 9, 2023 · 7f741b0 · 7f741b0
1 parent d9b2065
commit 7f741b0
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/tests/system/_helpers.py b/tests/system/_helpers.py
@@ -91,7 +91,11 @@ def empty_bucket(bucket):
 
 
 def delete_blob(blob):
-    errors = (exceptions.Conflict, exceptions.TooManyRequests)
+    errors = (
+        exceptions.Conflict,
+        exceptions.TooManyRequests,
+        exceptions.ServiceUnavailable,
+    )
     retry = RetryErrors(errors)
     try:
         retry(blob.delete)(timeout=120)  # seconds
@@ -105,7 +109,11 @@ def delete_blob(blob):
 
 
 def delete_bucket(bucket):
-    errors = (exceptions.Conflict, exceptions.TooManyRequests)
+    errors = (
+        exceptions.Conflict,
+        exceptions.TooManyRequests,
+        exceptions.ServiceUnavailable,
+    )
     retry = RetryErrors(errors, max_tries=15)
     retry(empty_bucket)(bucket)
     retry(bucket.delete)(force=True)

diff --git a/tests/system/test_bucket.py b/tests/system/test_bucket.py
@@ -1059,7 +1059,7 @@ def test_new_bucket_with_autoclass(
 
     # Autoclass can be enabled/disabled via bucket patch
     bucket.autoclass_enabled = False
-    bucket.patch()
+    bucket.patch(if_metageneration_match=bucket.metageneration)
 
     assert bucket.autoclass_enabled is False
     assert bucket.autoclass_toggle_time != previous_toggle_time