diff options
Diffstat (limited to 'waitress/tests/test_task.py')
| -rw-r--r-- | waitress/tests/test_task.py | 26 |
1 files changed, 16 insertions, 10 deletions
diff --git a/waitress/tests/test_task.py b/waitress/tests/test_task.py index 1493b21..3d88cd2 100644 --- a/waitress/tests/test_task.py +++ b/waitress/tests/test_task.py @@ -29,19 +29,19 @@ class TestThreadedTaskDispatcher(unittest.TestCase): def test_set_thread_count_increase(self): inst = self._makeOne() - L = [] - inst.start_new_thread = lambda *x: L.append(x) + mod = DummyThreadModule() + inst.thread_module = mod inst.set_thread_count(1) - self.assertEqual(L, [(inst.handler_thread, (0,))]) + self.assertEqual(mod.threads, [(inst.handler_thread, (0,))]) def test_set_thread_count_increase_with_existing(self): inst = self._makeOne() - L = [] + mod = DummyThreadModule() + inst.thread_module = mod inst.threads = {0:1} - inst.start_new_thread = lambda *x: L.append(x) inst.set_thread_count(2) - self.assertEqual(L, [(inst.handler_thread, (1,))]) - + self.assertEqual(mod.threads, [(inst.handler_thread, (1,))]) + def test_set_thread_count_decrease(self): inst = self._makeOne() inst.threads = {'a':1, 'b':2} @@ -51,11 +51,11 @@ class TestThreadedTaskDispatcher(unittest.TestCase): def test_set_thread_count_same(self): inst = self._makeOne() - L = [] - inst.start_new_thread = lambda *x: L.append(x) + mod = DummyThreadModule() + inst.thread_module = mod inst.threads = {0:1} inst.set_thread_count(1) - self.assertEqual(L, []) + self.assertEqual(mod.threads, []) def test_add_task(self): task = DummyTask() @@ -684,3 +684,9 @@ class DummyLogger(object): def exception(self, msg): self.logged.append(msg) +class DummyThreadModule(object): + def __init__(self): + self.threads = [] + def start_new_thread(self, *x): + self.threads.append(x) + |