Tolerate 500 for a while when waiting for tree size

1 files changed, 10 insertions, 1 deletions

diff --git a/tools/testcase1.py b/tools/testcase1.py
index 12079c8..6a12301 100755
--- a/tools/testcase1.py
+++ b/tools/testcase1.py
@@ -148,9 +148,18 @@ def get_and_check_entry(timestamp, chain, leaf_index, baseurl):
                     len(submittedcertchain),
                     len(submittedcertchain))
 
+def get_sth_retry(baseurl, tries=1):
+    for i in range(tries):
+        try:
+            return get_sth(baseurl)
+        except requests.exceptions.HTTPError, e:
+            if e.response.status_code == 500:
+                sleep(1)
+    raise e
+
 def correct_tree_size(expected):
     for baseurl in baseurls:
-        sth = get_sth(baseurl)
+        sth = get_sth_retry(baseurl, tries=10)
         tree_size = sth["tree_size"]
         if tree_size != expected:
             return False