Adjust wait condition in test

enthought · Mar 26, 2024 · cc9127c · cc9127c
1 parent 9148d03
commit cc9127c
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/traits/observation/tests/test_observe.py b/traits/observation/tests/test_observe.py
@@ -690,17 +690,15 @@ async def test_async_dispatch_error(self):
 
         async def handler(event):
             raise Exception("Bad handler")
-            event.set()
 
         def exception_handler(loop, context):
             exceptions.append(context["exception"].args[0])
+            event.set()
 
         with self.asyncio_exception_handler(exception_handler):
-            with self.assertRaises(asyncio.exceptions.TimeoutError):
-                dispatch_same(handler, event)
-                await asyncio.wait_for(event.wait(), timeout=0.1)
+            dispatch_same(handler, event)
+            await asyncio.wait_for(event.wait(), timeout=10.0)
 
-        self.assertFalse(event.is_set())
         self.assertEqual(exceptions, ["Bad handler"])
 
     def test_async_dispatch_no_loop(self):