Add coverage for timeout waiting on a call

cjw296 · Apr 20, 2023 · 22c3c60 · 22c3c60
1 parent 46a40d5
commit 22c3c60
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_hook.py b/tests/test_hook.py
@@ -1,4 +1,5 @@
 from testfixtures import ShouldRaise, compare
+from twisted.internet.defer import inlineCallbacks, TimeoutError
 from twisted.trial.unittest import TestCase
 
 from carly import hook, cleanup
@@ -27,3 +28,11 @@ def testUnconsumedCalls(self):
 
     def testUnconsumedstr(self):
         compare(str(UnconsumedCalls({})), expected='\n{}')
+
+    @inlineCallbacks
+    def testTimeout(self):
+        hook(Sample, 'method')
+        self.addCleanup(cleanup)
+        obj = Sample()
+        with ShouldRaise(TimeoutError):
+            yield obj.method.called(timeout=0.01)