From 6fc5cd2bf4fcf83647398b1d1b8167acebd514af Mon Sep 17 00:00:00 2001
From: Arnaud <arnaud@status.im>
Date: Thu, 2 Oct 2025 11:33:33 +0200
Subject: [PATCH] Fix mock clock

---
 tests/codex/helpers/mockclock.nim | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/tests/codex/helpers/mockclock.nim b/tests/codex/helpers/mockclock.nim
index be1eb4d2..dad63751 100644
--- a/tests/codex/helpers/mockclock.nim
+++ b/tests/codex/helpers/mockclock.nim
@@ -33,11 +33,16 @@ proc advance*(clock: MockClock, seconds: int64) =
 method now*(clock: MockClock): SecondsSince1970 =
   clock.time
 
-method waitUntil*(clock: MockClock, time: SecondsSince1970) {.async.} =
-  if time > clock.now():
-    let future = newFuture[void]()
-    clock.waiting.add(Waiting(until: time, future: future))
-    await future
+method waitUntil*(
+    clock: MockClock, time: SecondsSince1970
+) {.async: (raises: [CancelledError]).} =
+  try:
+    if time > clock.now():
+      let future = newFuture[void]()
+      clock.waiting.add(Waiting(until: time, future: future))
+      await future
+  except Exception as e:
+    discard
 
 proc isWaiting*(clock: MockClock): bool =
   clock.waiting.len > 0