"""Test module to run benchmarks in a wrapper application. This is typically
used to prefix the benchmark command with simulator commands."""
from litsupport import shellcommand
from litsupport import testplan


def mutateCommandLine(context, commandline):
    cmd = shellcommand.parse(commandline)
    run_under_cmd = shellcommand.parse(context.config.run_under)

    if run_under_cmd.stdin is not None or \
       run_under_cmd.stdout is not None or \
       run_under_cmd.stderr is not None or \
       run_under_cmd.workdir is not None or \
       run_under_cmd.envvars:
        raise Exception("invalid run_under argument!")

    cmd.wrap(run_under_cmd.executable, run_under_cmd.arguments)

    return cmd.toCommandline()


def mutatePlan(context, plan):
    run_under = context.config.run_under
    if run_under:
        plan.runscript = testplan.mutateScript(context, plan.runscript,
                                               mutateCommandLine)