def test_inspect_pool(self):
lil = Pool(workers=1)
self.assertEqual(lil.inspect_pool(), 0)
ls_command = lil.create_process('sleep 1')
lil.add_to_pool(ls_command)
self.assertEqual(lil.inspect_pool(), 1)
tmp_command = lil.create_process('sleep 1')
lil.add_to_pool(tmp_command)
self.assertEqual(lil.inspect_pool(), 2)
def test_prepare_commands(self):
lil = Pool(workers=2)
commands = []
lil.prepare_commands(commands)
self.assertEqual(len(lil.commands), 0)
commands = [
'ls',
'cd /tmp',
]
lil.prepare_commands(commands)
self.assertEqual(len(lil.commands), 2)
def test_set_callback(self):
lil = Pool(workers=1)
self.assertEqual(lil.callback, None)
lil.set_callback(lambda: 'hello')
self.assertNotEqual(lil.callback, None)
self.assertEqual(lil.callback(), 'hello')
# Test manual clearing.
lil.set_callback(None)
self.assertEqual(lil.callback, None)
# Test without args/kwargs.
lil.set_callback()
self.assertEqual(lil.callback, None)
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])
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])
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_add_to_pool(self):
lil = Pool(workers=2)
ls_command = lil.create_process('ls')
tmp_command = lil.create_process('cd /tmp')
self.assertEqual(len(lil.pool), 0)
lil.add_to_pool(ls_command)
self.assertEqual(len(lil.pool), 1)
lil.add_to_pool(tmp_command)
self.assertEqual(len(lil.pool), 2)
def test_set_callback(self):
lil = Pool(workers=1)
self.assertEqual(lil.callback, None)
lil.set_callback(lambda: 'hello')
self.assertNotEqual(lil.callback, None)
self.assertEqual(lil.callback(), 'hello')
# Test manual clearing.
lil.set_callback(None)
self.assertEqual(lil.callback, None)
# Test without args/kwargs.
lil.set_callback()
self.assertEqual(lil.callback, None)
def test_remove_from_pool(self):
lil = Pool(workers=2)
lil.pool = {
1: FakeProcess('fake'),
2: FakeProcess('fake'),
}
# Fake more data.
lil.pool[1].pid = 1
lil.pool[2].pid = 2
self.assertEqual(len(lil.pool), 2)
lil.remove_from_pool(2)
self.assertEqual(len(lil.pool), 1)
lil.remove_from_pool(1)
self.assertEqual(len(lil.pool), 0)
lil.remove_from_pool(1)
self.assertEqual(len(lil.pool), 0)
def test_add_to_pool(self):
lil = Pool(workers=2)
ls_command = lil.create_process('ls')
tmp_command = lil.create_process('cd /tmp')
self.assertEqual(len(lil.pool), 0)
lil.add_to_pool(ls_command)
self.assertEqual(len(lil.pool), 1)
lil.add_to_pool(tmp_command)
self.assertEqual(len(lil.pool), 2)
def test_prepare_commands(self):
lil = Pool(workers=2)
commands = []
lil.prepare_commands(commands)
self.assertEqual(len(lil.commands), 0)
commands = [
'ls',
'cd /tmp',
]
lil.prepare_commands(commands)
self.assertEqual(len(lil.commands), 2)
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_remove_from_pool(self):
lil = Pool(workers=2)
lil.pool = {
1: FakeProcess('fake'),
2: FakeProcess('fake'),
}
# Fake more data.
lil.pool[1].pid = 1
lil.pool[2].pid = 2
self.assertEqual(len(lil.pool), 2)
lil.remove_from_pool(2)
self.assertEqual(len(lil.pool), 1)
lil.remove_from_pool(1)
self.assertEqual(len(lil.pool), 0)
lil.remove_from_pool(1)
self.assertEqual(len(lil.pool), 0)
def test_next_command(self):
lil = Pool(workers=2)
commands = [
'ls',
'cd /tmp',
]
lil.prepare_commands(commands)
cmd1 = lil.next_command()
self.assertEqual(cmd1, 'ls')
cmd2 = lil.next_command()
self.assertEqual(cmd2, 'cd /tmp')
cmd3 = lil.next_command()
self.assertEqual(cmd3, None)
def test_next_command(self):
lil = Pool(workers=2)
commands = [
'ls',
'cd /tmp',
]
lil.prepare_commands(commands)
cmd1 = lil.next_command()
self.assertEqual(cmd1, 'ls')
cmd2 = lil.next_command()
self.assertEqual(cmd2, 'cd /tmp')
cmd3 = lil.next_command()
self.assertEqual(cmd3, None)
def run():
from littleworkers import Pool
commands = ['fab regenerate', 'fab serve']
pool = Pool(workers=2)
pool.run(commands)
def test_process_kwargs(self):
lil = Pool(workers=1)
self.assertEqual(lil.process_kwargs(None), {'shell': True})
self.assertEqual(lil.process_kwargs('ls -al'), {'shell': True})
def test_process_kwargs(self):
lil = Pool(workers=1)
self.assertEqual(lil.process_kwargs(None), {'shell': True})
self.assertEqual(lil.process_kwargs('ls -al'), {'shell': True})
def test_inspect_pool(self):
lil = Pool(workers=1)
self.assertEqual(lil.inspect_pool(), 0)
ls_command = lil.create_process('sleep 1')
lil.add_to_pool(ls_command)
self.assertEqual(lil.inspect_pool(), 1)
tmp_command = lil.create_process('sleep 1')
lil.add_to_pool(tmp_command)
self.assertEqual(lil.inspect_pool(), 2)
