Retry on duplicate bucket in persistent bucket get_or_create

reef-technologies · Sep 17, 2024 · 2c05f31 · 2c05f31
1 parent 7d18983
commit 2c05f31
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/test/integration/persistent_bucket.py b/test/integration/persistent_bucket.py
@@ -15,7 +15,7 @@
 
 import backoff
 from b2sdk.v2 import Bucket
-from b2sdk.v2.exception import NonExistentBucket
+from b2sdk.v2.exception import DuplicateBucketName, NonExistentBucket
 
 PERSISTENT_BUCKET_NAME_PREFIX = "constst"
 
@@ -65,6 +65,12 @@ def get_persistent_bucket_name(b2_api: Api) -> str:
  return f"{PERSISTENT_BUCKET_NAME_PREFIX}-{bucket_hash}" [:BUCKET_NAME_LENGTH]
 
 
+@backoff.on_exception(
+ backoff.expo,
+ DuplicateBucketName,
+ max_tries=3,
+ jitter=backoff.full_jitter,
+)
 def get_or_create_persistent_bucket(b2_api: Api) -> Bucket:
  bucket_name = get_persistent_bucket_name(b2_api)
  try: