class TestBrowserThreadPool(unittest.TestCase):
"""TestBrowserThreadPool class"""
THREADS = 2
def __test_function(self, arg):
pass
def setUp(self):
self._pool = ThreadPool(num_threads=self.THREADS,
total_items=10,
timeout=0)
def tearDown(self):
logger = Logger.log()
for handler in logger.handlers:
logger.removeHandler(handler)
self._pool.join()
def test_size(self):
""" ThreadPool.size test """
self.assertIs(type(self._pool.size), int)
self.assertEqual(self._pool.size, 0)
def test_worker_size(self):
""" ThreadPool.worker_size test """
self.assertIs(type(self._pool.workers_size), int)
self.assertEqual(self._pool.workers_size, self.THREADS)
def test_items_size(self):
""" ThreadPool.items_size test """
self.assertIs(type(self._pool.items_size), int)
self.assertEqual(self._pool.items_size, 0)
def test_add(self):
""" ThreadPool.add() test """
self.assertIs(self._pool.add(self.__test_function, 1), None)
def pause(self):
ans = self._pool.pause()
return ans
def test_pause(self):
""" ThreadPool.pause() test """
with self.assertRaises(KeyboardInterrupt) as context:
with patch('builtins.input', return_value='e') as _raw_input:
self.assertEqual(self.pause(), 'e')
_raw_input.assert_called_once_with('e')
self.assertTrue(KeyboardInterrupt == context.expected)
def test_resume(self):
""" ThreadPool.resume() test """
self._pool.is_started = False
self.assertIs(self._pool.resume(), None)
class TestBrowserThreadPool(unittest.TestCase):
"""TestBrowserThreadPool class"""
THREADS = 2
def __test_function(self,arg):
pass
def setUp(self):
self._pool = ThreadPool(num_threads=self.THREADS, total_items=10, timeout=0)
def tearDown(self):
logger = Logger.log()
for handler in logger.handlers:
logger.removeHandler(handler)
self._pool.join()
def test_size(self):
""" ThreadPool.size test """
self.assertIs(type(self._pool.size), int)
self.assertEqual(self._pool.size, 0)
def test_worker_size(self):
""" ThreadPool.worker_size test """
self.assertIs(type(self._pool.workers_size), int)
self.assertEqual(self._pool.workers_size, self.THREADS)
def test_items_size(self):
""" ThreadPool.items_size test """
self.assertIs(type(self._pool.items_size), int)
self.assertEqual(self._pool.items_size, 0)
def test_add(self):
""" ThreadPool.add() test """
self.assertIs(self._pool.add(self.__test_function, 1), None)
def pause(self):
ans = self._pool.pause()
return ans
def test_pause(self):
""" ThreadPool.pause() test """
with self.assertRaises(KeyboardInterrupt) as context:
with patch('builtins.input', return_value='e') as _raw_input:
self.assertEqual(self.pause(), 'e')
_raw_input.assert_called_once_with('e')
self.assertTrue(KeyboardInterrupt == context.expected)
def test_resume(self):
""" ThreadPool.resume() test """
self._pool.is_started = False
self.assertIs(self._pool.resume(), None)
def setUp(self):
self._pool = ThreadPool(num_threads=self.THREADS, total_items=10, timeout=0)
def setUp(self):
self.__pool = ThreadPool(num_threads=self.THREADS,
total_items=10,
timeout=0)