new interrupt flag for cancel_all to prevent _map() from starting new tasks

2022-02-06 16:32:42 +01:00
parent 2d40f5707b
commit e8e13406ea
2 changed files with 25 additions and 16 deletions
--- a/tests/test_pool.py
+++ b/tests/test_pool.py
@ -52,6 +52,8 @@ class BaseTaskPoolTestCase(TestCase):
        self.assertEqual(self.test_pool_name, self.task_pool._name)
        self.assertIsInstance(self.task_pool._all_tasks_known_flag, asyncio.locks.Event)
        self.assertTrue(self.task_pool._all_tasks_known_flag.is_set())
+        self.assertIsInstance(self.task_pool._interrupt_flag, asyncio.locks.Event)
+        self.assertFalse(self.task_pool._interrupt_flag.is_set())
        self.mock__add_pool.assert_called_once_with(self.task_pool)
        self.mock_pool_size.assert_called_once_with(self.test_pool_size)
        self.mock___str__.assert_called_once_with()