From 7be0ae2e8a424c277e5da5f0a75c4fe03f72b014 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= <sybren@stuvel.eu>
Date: Fri, 15 Dec 2017 17:17:18 +0100
Subject: [PATCH] Fix unittest

---
 tests/test_worker.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/test_worker.py b/tests/test_worker.py
index f0cc6eac..2e154274 100644
--- a/tests/test_worker.py
+++ b/tests/test_worker.py
@@ -1,3 +1,4 @@
+import concurrent.futures
 import unittest
 import unittest.mock
 from unittest.mock import Mock
@@ -449,7 +450,8 @@ class WorkerSleepingTest(AbstractFWorkerTest):
         }, status_code=423)
 
         self.worker.schedule_fetch_task()
-        self.loop.run_until_complete(self.worker.fetch_task_task)
+        with self.assertRaises(concurrent.futures.CancelledError):
+            self.loop.run_until_complete(self.worker.fetch_task_task)
 
         self.assertIsNotNone(self.worker.sleeping_task)
         self.assertFalse(self.worker.sleeping_task.done())
-- 
GitLab