diff --git a/tests/test_worker.py b/tests/test_worker.py index f0cc6eac52124fa4fee3f9df5381d1fa7c47afe7..2e1542745109288ad9b5e382ad58092db3479ae4 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())