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