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_pool(self):
x = TaskPool()
x.on_start()
x.on_stop()
x.on_apply(Mock())
x._pool = None
x.on_stop()
assert 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 len(x._pool_map.keys()) == 1
assert x.getpid()
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_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_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)
