36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import time
|
|
|
|
|
|
def mg_sleep_and_assert(expected_value, function_to_retrieve_data, max_duration=20, time_between_attempt=0.2):
|
|
result = function_to_retrieve_data()
|
|
start_time = time.time()
|
|
while result != expected_value:
|
|
duration = time.time() - start_time
|
|
if duration > max_duration:
|
|
assert (
|
|
False
|
|
), f" mg_sleep_and_assert has tried for too long and did not get the expected result! Last result was: {result}"
|
|
|
|
time.sleep(time_between_attempt)
|
|
result = function_to_retrieve_data()
|
|
|
|
return result
|
|
|
|
|
|
def mg_sleep_and_assert_collection(
|
|
expected_value, function_to_retrieve_data, max_duration=20, time_between_attempt=0.2
|
|
):
|
|
result = function_to_retrieve_data()
|
|
start_time = time.time()
|
|
while len(result) != len(expected_value) or any((x not in result for x in expected_value)):
|
|
duration = time.time() - start_time
|
|
if duration > max_duration:
|
|
assert (
|
|
False
|
|
), f" mg_sleep_and_assert has tried for too long and did not get the expected result! Last result was: {result}"
|
|
|
|
time.sleep(time_between_attempt)
|
|
result = function_to_retrieve_data()
|
|
|
|
return result
|