test(samples): add backoff to cluster creation sample (#362)

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

Hide file tree

Showing 2 changed files with 7 additions and 2 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,8 +17,9 @@
 import time
 import warnings
 
+import backoff
+from google.api_core import exceptions
 from google.cloud import bigtable
-
 import pytest
 
 import instanceadmin
@@ -128,7 +129,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