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_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)