snapshot of https://github.com/zulip/zulip.git @ 48c69911c7dd4e03b20552eb3bd77ad8151c1357 (2026-09-15)

zerver/tests/test_timeout.py

@@ -0,0 +1,56 @@
+import sys
+import time
+import traceback
+from unittest import skipIf
+
+from zerver.lib.test_classes import ZulipTestCase
+from zerver.lib.timeout import TimeoutExpiredError, unsafe_timeout
+
+
+class TimeoutTestCase(ZulipTestCase):
+    # We can't use assertRaises because that doesn't store the
+    # traceback, which we want to verify
+
+    def something_exceptional(self) -> int:
+        raise ValueError("Something went wrong")
+
+    def sleep_x_seconds_y_times(self, x: float, y: int) -> int:
+        for i in range(y):
+            time.sleep(x)
+        return 42  # nocoverage
+
+    def test_timeout_returns(self) -> None:
+        ret = unsafe_timeout(1, lambda: 42)
+        self.assertEqual(ret, 42)
+
+    def test_timeout_exceeded(self) -> None:
+        try:
+            unsafe_timeout(1, lambda: self.sleep_x_seconds_y_times(0.1, 50))
+            raise AssertionError("Failed to raise a timeout")
+        except TimeoutExpiredError as exc:
+            tb = traceback.format_tb(exc.__traceback__)
+            self.assertIn("in sleep_x_seconds_y_times", tb[-1])
+            self.assertIn("time.sleep(x)", tb[-1])
+
+    def test_timeout_raises(self) -> None:
+        try:
+            unsafe_timeout(1, self.something_exceptional)
+            raise AssertionError("Failed to raise an exception")
+        except ValueError as exc:
+            tb = traceback.format_tb(exc.__traceback__)
+            self.assertIn("in something_exceptional", tb[-1])
+            self.assertIn("raise ValueError", tb[-1])
+
+    @skipIf(sys.version_info >= (3, 11), "https://github.com/nedbat/coveragepy/issues/1626")
+    def test_timeout_warn(self) -> None:
+        # If the sleep is long enough, it will outlast the attempts to
+        # kill it
+        with self.assertLogs(level="WARNING") as m:
+            try:
+                unsafe_timeout(1, lambda: self.sleep_x_seconds_y_times(5, 1))
+                raise AssertionError("Failed to raise a timeout")
+            except TimeoutExpiredError as exc:
+                tb = traceback.format_tb(exc.__traceback__)
+                self.assertNotIn("in sleep_x_seconds_y_times", tb[-1])
+                self.assertIn("raise TimeoutExpiredError", tb[-1])
+        self.assertEqual(m.output, ["WARNING:root:Failed to time out backend thread"])