SMART-Lab / smartdispatch

        assert_true(os.path.isdir(self.logs_dir))

        assert_equal(len(os.listdir(self.logs_dir)), 1)

        assert_equal(len(open(running_commands_file).readlines()), 0)

        assert_equal(len(open(pending_commands_file).readlines()), len(commands))

    def test_main_resume_by_expanding_pool_default(self):

        # Create SMART_DISPATCH_LOGS structure.

        call(self.launch_command, shell=True)

        batch_uid = os.listdir(self.logs_dir)[0]

        # Simulate that some commands are in the running state.

        nb_commands_files = 2  # 'commands.txt' and 'running_commands.txt'

        path_job_commands = os.path.join(self.logs_dir, batch_uid, "commands")

        pending_commands_file = pjoin(path_job_commands, "commands.txt")

        running_commands_file = pjoin(path_job_commands, "running_commands.txt")

        commands = open(pending_commands_file).read().strip().split("\n")

        with open(running_commands_file, 'w') as running_commands:

            running_commands.write("\n".join(commands[::2]) + "\n")

        with open(pending_commands_file, 'w') as pending_commands:

            pending_commands.write("\n".join(commands[1::2]) + "\n")

        # Remove PBS files so we can check that new ones are going to be created.

        for f in os.listdir(path_job_commands):

            if f.startswith('job_commands_') and f.endswith('.sh'):

                os.remove(pjoin(path_job_commands, f))

        # Should NOT move running commands back to pending but should add new workers.

        command_line = self.resume_command.format(batch_uid)

        command_line += " --expandPool"

        exit_status = call(command_line, shell=True)

        # Test validation

        assert_equal(exit_status, 0)

        assert_equal(len(open(running_commands_file).readlines()), len(commands[::2]))

        assert_equal(len(open(pending_commands_file).readlines()), len(commands[1::2]))

        nb_job_commands_files = len(os.listdir(path_job_commands))

        assert_equal(nb_job_commands_files-nb_commands_files, len(commands[1::2]))

    def test_main_resume_by_expanding_pool(self):

        with open(running_commands_file, 'w') as running_commands:

            running_commands.write("\n".join(commands[::2]) + "\n")

        with open(pending_commands_file, 'w') as pending_commands:

            pending_commands.write("\n".join(commands[1::2]) + "\n")

        # Actual test (should move running commands back to pending).

        exit_status = call(self.resume_command.format(batch_uid), shell=True)

        # Test validation

        assert_equal(exit_status, 0)

        assert_true(os.path.isdir(self.logs_dir))

        assert_equal(len(os.listdir(self.logs_dir)), 1)

        assert_equal(len(open(running_commands_file).readlines()), 0)

        assert_equal(len(open(pending_commands_file).readlines()), len(commands))

        # Test when batch_uid is a path instead of a jobname.

        # Setup

        batch_uid = os.path.join(self.logs_dir, os.listdir(self.logs_dir)[0])

        # Simulate that some commands are in the running state.

        path_job_commands = os.path.join(self.logs_dir, batch_uid, "commands")

        pending_commands_file = pjoin(path_job_commands, "commands.txt")

        running_commands_file = pjoin(path_job_commands, "running_commands.txt")

        commands = open(pending_commands_file).read().strip().split("\n")

        with open(running_commands_file, 'w') as running_commands:

            running_commands.write("\n".join(commands[::2]) + "\n")

        with open(pending_commands_file, 'w') as pending_commands:

            pending_commands.write("\n".join(commands[1::2]) + "\n")

        # Actual test (should move running commands back to pending).

        exit_status = call(self.resume_command.format(batch_uid), shell=True)

        # Test validation

        assert_equal(exit_status, 0)

        assert_true(os.path.isdir(self.logs_dir))

        assert_equal(len(os.listdir(self.logs_dir)), 1)

        assert_equal(len(open(running_commands_file).readlines()), 0)

        assert_equal(len(open(pending_commands_file).readlines()), len(commands))