doc(samples): wrap backoff around cluster creation sample

Closes #353.
googleapis · Jul 13, 2021 · b0267cd · b0267cd
1 parent bc57c79
commit b0267cd
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/samples/instanceadmin/requirements.txt b/samples/instanceadmin/requirements.txt
@@ -1 +1,2 @@
 google-cloud-bigtable==2.0.0
+backoff==1.11.0
diff --git a/samples/instanceadmin/test_instanceadmin.py b/samples/instanceadmin/test_instanceadmin.py
@@ -17,10 +17,12 @@
 import time
 import warnings
 
-from google.cloud import bigtable
-
+import backoff
 import pytest
 
+from google.api_core import exceptions
+from google.cloud import bigtable
+
 import instanceadmin
 
 
@@ -128,7 +130,10 @@ def test_add_and_delete_cluster(capsys, dispose_of):
     capsys.readouterr()  # throw away output
 
     # Add a cluster to that instance
-    instanceadmin.add_cluster(PROJECT, INSTANCE, CLUSTER2)
+    # Avoid failing for "instance is currently being changed" by
+    # applying an exponential backoff
+    w_backoff = backoff.on_exception(backoff.expo, exceptions.ServiceUnavailable)
+    w_backoff(instanceadmin.add_cluster)(PROJECT, INSTANCE, CLUSTER2)
     out = capsys.readouterr().out
     assert f"Adding cluster to instance {INSTANCE}" in out
     assert "Listing clusters..." in out