def wait_until(self,
cond,
max_timeout=10,
poll_interval=0.25,
name="cond"):
"""
Like beat.beat.wait_until but catches exceptions
In a ElasticTest `cond` will usually be a query, and we need to keep retrying
eg. on 503 response codes
"""
assert callable(
cond), "First argument of wait_until must be a function"
start = datetime.now()
while datetime.now() - start < timedelta(seconds=max_timeout):
try:
result = cond()
if result:
return result
except AttributeError as ex:
raise ex
except:
pass
time.sleep(poll_interval)
raise TimeoutError(
"Timeout waiting for '{}' to be true. ".format(name) +
"Waited {} seconds.".format(max_timeout))
def wait_until(self, cond, max_timeout=10, poll_interval=0.1, name="cond"):
"""
Like beat.beat.wait_until but catches exceptions
In a ElasticTest `cond` will usually be a query, and we need to keep retrying
eg. on 503 response codes
"""
start = datetime.now()
abort = False
while not abort:
try:
abort = cond()
except:
abort = False
if datetime.now() - start > timedelta(seconds=max_timeout):
raise TimeoutError("Timeout waiting for '{}' to be true. ".format(name) +
"Waited {} seconds.".format(max_timeout))
time.sleep(poll_interval)