From 879c917242ab2af0a8332669f92987cd1e40370f Mon Sep 17 00:00:00 2001
From: cheatfate <eugene.kabanov@status.im>
Date: Tue, 17 Nov 2020 12:04:19 +0200
Subject: [PATCH] Add cancellation handler for cancelAndWait()

---
 chronos/asyncfutures2.nim | 1 +
 1 file changed, 1 insertion(+)

diff --git a/chronos/asyncfutures2.nim b/chronos/asyncfutures2.nim
index 625be00..231d497 100644
--- a/chronos/asyncfutures2.nim
+++ b/chronos/asyncfutures2.nim
@@ -804,6 +804,7 @@ proc cancelAndWait*[T](fut: Future[T]): Future[void] =
     retFuture.complete()
   else:
     fut.addCallback(continuation)
+    retFuture.cancelCallback = cancellation
     # Initiate cancellation process.
     fut.cancel()
   return retFuture