diff --git a/tests/client/loomenv.py b/tests/client/loomenv.py index 7ba81e3fd9d6b1fb05309907d067696d6f4f5c15..d82e71f5a622fa740c2db97560bb8d5c23dcadc8 100644 --- a/tests/client/loomenv.py +++ b/tests/client/loomenv.py @@ -135,11 +135,13 @@ class LoomEnv(Env): self.check_stats() return self._client - def submit(self, results, report=None): + def submit(self, results, report=None, check_timeout=None): if report: report = os.path.join(LOOM_TEST_BUILD_DIR, report) result = self.client.submit(results, report) self.check_stats() + if check_timeout: + time.sleep(check_timeout) self.check_final_state() return result diff --git a/tests/client/test_client.py b/tests/client/test_client.py index b6fdea0a76bb2f444ac829ef95998d1723556dcf..7fe03bb09810f0cccef9e8e8276bc792db7f04fa 100644 --- a/tests/client/test_client.py +++ b/tests/client/test_client.py @@ -57,7 +57,7 @@ print(client.submit((a, b, c, d))) b = tasks.const("xyz") c = tasks.const("123") d = tasks.merge((a, b, c)) - result = loom_env.submit(d) + result = loom_env.submit(d, check_timeout=0.9) assert result == b"abcxyz123"