diff options
| -rwxr-xr-x | tools/testcase1.py | 11 |
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 |