docs: adding backoff to quickstart test (#135)

* adding backoff to quickstart test * fixing lint
googleapis · Mar 2, 2021 · a22df4c · a22df4c
1 parent 6231361
commit a22df4c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/samples/snippets/quickstart/quickstart_test.py b/samples/snippets/quickstart/quickstart_test.py
@@ -15,6 +15,8 @@
 import os
 import uuid
 
+import backoff
+from google.api_core.exceptions import ServiceUnavailable
 from google.cloud import dataproc_v1 as dataproc
 from google.cloud import storage
 import pytest
@@ -39,7 +41,14 @@
 @pytest.fixture(autouse=True)
 def setup_teardown():
     storage_client = storage.Client()
-    bucket = storage_client.create_bucket(STAGING_BUCKET)
+
+    @backoff.on_exception(backoff.expo,
+                          ServiceUnavailable,
+                          max_tries=5)
+    def create_bucket():
+        return storage_client.create_bucket(STAGING_BUCKET)
+
+    bucket = create_bucket()
     blob = bucket.blob(JOB_FILE_NAME)
     blob.upload_from_string(SORT_CODE)
 

diff --git a/samples/snippets/requirements.txt b/samples/snippets/requirements.txt
@@ -1,3 +1,5 @@
+
+backoff==1.10.0
 grpcio==1.36.0
 google-auth==1.27.0
 google-auth-httplib2==0.0.4