Skip to content
Snippets Groups Projects
Commit 7be0ae2e authored by Sybren A. Stüvel's avatar Sybren A. Stüvel
Browse files

Fix unittest

parent 3200929b
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 3 additions and 1 deletion
tests/test_worker.py
+ 3
− 1
View file @ 7be0ae2e
import concurrent.futures
import unittest import unittest
import unittest.mock import unittest.mock
from unittest.mock import Mock from unittest.mock import Mock
...@@ -449,6 +450,7 @@ class WorkerSleepingTest(AbstractFWorkerTest): ...@@ -449,6 +450,7 @@ class WorkerSleepingTest(AbstractFWorkerTest):
}, status_code=423) }, status_code=423)
self.worker.schedule_fetch_task() self.worker.schedule_fetch_task()
with self.assertRaises(concurrent.futures.CancelledError):
self.loop.run_until_complete(self.worker.fetch_task_task) self.loop.run_until_complete(self.worker.fetch_task_task)
self.assertIsNotNone(self.worker.sleeping_task) self.assertIsNotNone(self.worker.sleeping_task)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment