Re-introduce time checking in __check_request

sorgerlab · Nov 30, 2023 · 0f2babb · 0f2babb
1 parent b1c7054
commit 0f2babb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/indra/tests/test_db_rest.py b/indra/tests/test_db_rest.py
@@ -20,6 +20,8 @@ def __check_request(seconds, *args, **kwargs):
     time_taken = datetime.now() - now
     if check_stmts:
         assert resp.statements, "Got no statements."
+    assert time_taken.total_seconds() < seconds, \
+        "Request took too long: %s" % time_taken.total_seconds()
     return resp