def test_simple_usage(self):
commands = [
'ls',
]
lil = Pool(workers=1)
lil.run(commands)
lil = Pool(workers=2)
lil.run(commands)
# Should raise an exception.
self.assertRaises(NotEnoughWorkers, Pool, workers=0)
def test_long_wait_time(self):
lil = Pool(workers=1)
start = time.time()
lil.run(['ls'])
end = time.time()
self.assertTrue(end - start < 1.0)
# Now with a longer wait.
lil = Pool(workers=1, wait_time=1.25)
start = time.time()
lil.run(['ls'])
end = time.time()
self.assertTrue(end - start > 1.0)
def test_callback(self):
lil = Pool(workers=1)
commands = [
'sleep 1',
'sleep 1',
'i_am_a_super-jacked-command_you_ought-not_to-have_on-your-1_system',
'sleep 1',
]
# Track exit codes.
codes = []
def track(proc):
codes.append(proc.returncode)
lil.run(commands, callback=track)
self.assertEqual(lil.callback, track)
self.assertEqual(codes, [0, 0, 127, 0])
