class ThreadPoolTestCase(unittest.TestCase):
'''This test case tests the ThreadPool class methods.'''
def setUp(self):
self.min_workers = 3
self.logger = None
self.tp = ThreadPool(self.min_workers, self.logger)
self.assertEquals(self.tp.min_workers, 3)
self.assertEquals(self.tp.logger, None)
self.assertEquals(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon)
def test_enqueue_task(self):
self.tp.enqueue_task('foo', 'bar')
task = self.tp.tasks.get()
self.assertEquals(('foo', ('bar', ), {}), task)
test = 'test'
self.tp.enqueue_task('foo', 'bar', test=True)
task = self.tp.tasks.get()
self.assertEquals(('foo', ('bar', ), {'test': True}), task)
def test_worker(self):
pass
def test_run(self):
pass
class ThreadPoolTestCase(unittest.TestCase):
'''This test case tests the ThreadPool class methods.'''
def setUp(self):
self.min_workers = 3
self.logger = None
self.tp = ThreadPool(self.min_workers, self.logger)
self.assertEquals(self.tp.min_workers, 3)
self.assertEquals(self.tp.logger, None)
self.assertEquals(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon)
def test_enqueue_task(self):
self.tp.enqueue_task('foo', 'bar')
task = self.tp.tasks.get()
self.assertEquals(('foo', ('bar',), {}), task)
test = 'test'
self.tp.enqueue_task('foo', 'bar', test=True)
task = self.tp.tasks.get()
self.assertEquals(('foo', ('bar',), {'test': True}), task)
def test_worker(self):
pass
def test_run(self):
pass
def setUp(self):
self.min_workers = 3
self.logger = None
self.tp = ThreadPool(self.min_workers, self.logger)
self.assertEquals(self.tp.min_workers, 3)
self.assertEquals(self.tp.logger, None)
self.assertEquals(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon)
def setUp(self):
IrcTkTestCase.setUp(self)
self.min_workers = 3
self.tp = ThreadPool(self.min_workers)
self.assertEqual(self.tp.min_workers, 3)
self.assertEqual(self.tp.logger, None)
self.assertEqual(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon is not None)
self.worker = Worker(self.tp.tasks, None)
self.assertEqual(self.tp.tasks, self.worker.tasks)
self.assertEqual(None, self.worker.logger)
def setUp(self):
self.min_workers = 3
self.logger = None
self.tp = ThreadPool(self.min_workers, self.logger)
self.assertEquals(self.tp.min_workers, 3)
self.assertEquals(self.tp.logger, None)
self.assertEquals(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon)
def setUp(self):
self.workers = 3
self.logger = None
self.tp = ThreadPool(self.workers, self.logger)
# Give the htread pool time to get up to speed.
time.sleep(0.3)
self.assertEquals(self.tp.workers, 3)
self.assertEquals(self.tp.logger, None)
self.assertEquals(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon)
def setUp(self):
IrcTkTestCase.setUp(self)
self.min_workers = 3
self.tp = ThreadPool(self.min_workers)
self.assertEqual(self.tp.min_workers, 3)
self.assertEqual(self.tp.logger, None)
self.assertEqual(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon is not None)
self.worker = Worker(self.tp.tasks, None)
self.assertEqual(self.tp.tasks, self.worker.tasks)
self.assertEqual(None, self.worker.logger)
class ThreadPoolTestCase(IrcTkTestCase):
def setUp(self):
IrcTkTestCase.setUp(self)
self.min_workers = 3
self.tp = ThreadPool(self.min_workers)
self.assertEqual(self.tp.min_workers, 3)
self.assertEqual(self.tp.logger, None)
self.assertEqual(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon is not None)
self.worker = Worker(self.tp.tasks, None)
self.assertEqual(self.tp.tasks, self.worker.tasks)
self.assertEqual(None, self.worker.logger)
def foo(self, *args, **kwargs):
return args or kwargs or 'bar'
def bad_foo(self):
raise Exception
def test_enqueue_task(self):
self.tp.enqueue_task('foo', 'bar')
task = self.tp.tasks.get()
self.assertEqual(('foo', ('bar',), {}), task)
self.tp.enqueue_task('foo', 'bar', test=True)
task = self.tp.tasks.get()
self.assertEqual(('foo', ('bar',), {'test': True}), task)
def test_spawn_worker(self):
self.tp._spawn_worker()
self.assertTrue(self.tp.too_few_workers)
# spawn enough workers to surpass too_few_workers
for i in range(3):
self.tp._spawn_worker()
self.assertFalse(self.tp.too_few_workers)
class ThreadPoolTestCase(IrcTkTestCase):
def setUp(self):
IrcTkTestCase.setUp(self)
self.min_workers = 3
self.tp = ThreadPool(self.min_workers)
self.assertEqual(self.tp.min_workers, 3)
self.assertEqual(self.tp.logger, None)
self.assertEqual(self.tp.wait, 0.01)
self.assertTrue(self.tp.daemon is not None)
self.worker = Worker(self.tp.tasks, None)
self.assertEqual(self.tp.tasks, self.worker.tasks)
self.assertEqual(None, self.worker.logger)
def foo(self, *args, **kwargs):
return args or kwargs or 'bar'
def bad_foo(self):
raise Exception
def test_enqueue_task(self):
self.tp.enqueue_task('foo', 'bar')
task = self.tp.tasks.get()
self.assertEqual(('foo', ('bar', ), {}), task)
self.tp.enqueue_task('foo', 'bar', test=True)
task = self.tp.tasks.get()
self.assertEqual(('foo', ('bar', ), {'test': True}), task)
def test_spawn_worker(self):
self.tp._spawn_worker()
self.assertTrue(self.tp.too_few_workers)
# spawn enough workers to surpass too_few_workers
for i in range(3):
self.tp._spawn_worker()
self.assertFalse(self.tp.too_few_workers)
def thread_pool(self):
thread_pool = \
ThreadPool(self.config['MIN_WORKERS'], logger=self.logger)
return thread_pool
