def test_get_info(self): x = TaskPool(10) x._pool = Mock(name='_pool') assert x._get_info() == { 'max-concurrency': 10, 'free-threads': x._pool.free(), 'running-threads': x._pool.running(), }
def test_get_info(self): x = TaskPool(10) x._pool = Mock(name='_pool') self.assertDictEqual(x._get_info(), { 'max-concurrency': 10, 'free-threads': x._pool.free(), 'running-threads': x._pool.running(), })
def test_make_killable_target(self): def valid_target(): return "some result..." def terminating_target(): raise GreenletExit() assert TaskPool._make_killable_target( valid_target)() == "some result..." assert TaskPool._make_killable_target(terminating_target)() == (False, None, None)
def test_terminate_job(self): func = Mock() pool = TaskPool(10) pool.on_start() pool.on_apply(func) assert len(pool._pool_map.keys()) == 1 pid = list(pool._pool_map.keys())[0] greenlet = pool._pool_map[pid] pool.terminate_job(pid) greenlet.link.assert_called_once() greenlet.kill.assert_called_once()
def test_shrink(self): x = TaskPool(10) x._pool = Mock(name='_pool') x.shrink(2) assert x.limit == 8 x._pool.resize.assert_called_with(8)
def test_grow(self): x = TaskPool(10) x._pool = Mock(name='_pool') x.grow(2) assert x.limit == 12 x._pool.resize.assert_called_with(12)
def test_pool(self): with mock_module(*eventlet_modules): with patch_many('eventlet.greenpool.GreenPool', 'eventlet.greenthread') as (GreenPool, greenthread): x = TaskPool() x.on_start() x.on_stop() x.on_apply(Mock()) x._pool = None x.on_stop() self.assertTrue(x.getpid())
def test_shrink(self): x = TaskPool(10) x._pool = Mock(name='_pool') x.shrink(2) self.assertEqual(x.limit, 8) x._pool.resize.assert_called_with(8)
def test_grow(self): x = TaskPool(10) x._pool = Mock(name='_pool') x.grow(2) self.assertEqual(x.limit, 12) x._pool.resize.assert_called_with(12)
def test_pool(self): x = TaskPool() x.on_start() x.on_stop() x.on_apply(Mock()) x._pool = None x.on_stop() assert len(x._pool_map.keys()) == 1 assert x.getpid()
def test_cleanup_after_job_finish(self): testMap = {'1': None} TaskPool._cleanup_after_job_finish(None, testMap, '1') assert len(testMap) == 0
def test_get_info(self): x = TaskPool(10) x._pool = Mock(name="_pool") self.assertDictEqual( x._get_info(), {"max-concurrency": 10, "free-threads": x._pool.free(), "running-threads": x._pool.running()} )
def test_pool(self): x = TaskPool() x.on_start() x.on_stop() x.on_apply(Mock()) x._pool = None x.on_stop() self.assertTrue(x.getpid())
def test_pool(self): x = TaskPool() x.on_start() x.on_stop() x.on_apply(Mock()) x._pool = None x.on_stop() assert x.getpid()