| Index: mojo/tools/mopy/test_util.py
|
| diff --git a/mojo/tools/mopy/test_util.py b/mojo/tools/mopy/test_util.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..d8f9fedab2d8b82a9447647aeb2dbbc2cf1c6cec
|
| --- /dev/null
|
| +++ b/mojo/tools/mopy/test_util.py
|
| @@ -0,0 +1,77 @@
|
| +# Copyright 2015 The Chromium Authors. All rights reserved.
|
| +# Use of this source code is governed by a BSD-style license that can be
|
| +# found in the LICENSE file.
|
| +
|
| +import logging
|
| +import os
|
| +import subprocess
|
| +
|
| +_logging = logging.getLogger()
|
| +
|
| +from mopy import android
|
| +from mopy.config import Config
|
| +from mopy.paths import Paths
|
| +from mopy.print_process_error import print_process_error
|
| +
|
| +
|
| +def build_shell_arguments(shell_args, apps_and_args=None):
|
| + """Build the list of arguments for the shell. |shell_args| are the base
|
| + arguments, |apps_and_args| is a dictionary that associates each application to
|
| + its specific arguments|. Each app included will be run by the shell.
|
| + """
|
| + result = shell_args[:]
|
| + if apps_and_args:
|
| + for (application, args) in apps_and_args.items():
|
| + result += ["--args-for=%s %s" % (application, " ".join(args))]
|
| + result += apps_and_args.keys()
|
| + return result
|
| +
|
| +
|
| +def get_shell_executable(config, run_launcher):
|
| + paths = Paths(config=config)
|
| + if config.target_os == Config.OS_ANDROID:
|
| + return os.path.join(paths.src_root, "mojo", "tools",
|
| + "android_mojo_shell.py")
|
| + elif run_launcher:
|
| + return paths.mojo_launcher_path
|
| + else:
|
| + return paths.mojo_shell_path
|
| +
|
| +
|
| +def build_command_line(config, shell_args, apps_and_args, run_launcher=False):
|
| + executable = get_shell_executable(config, run_launcher)
|
| + return "%s %s" % (executable, " ".join(["%r" % x for x in
|
| + build_shell_arguments(
|
| + shell_args, apps_and_args)]))
|
| +
|
| +
|
| +def run_test_android(shell_args, apps_and_args):
|
| + """Run the given test on the single/default android device."""
|
| + (r, w) = os.pipe()
|
| + with os.fdopen(r, "r") as rf:
|
| + with os.fdopen(w, "w") as wf:
|
| + arguments = build_shell_arguments(shell_args, apps_and_args)
|
| + android.StartShell(arguments, wf, wf.close)
|
| + return rf.read()
|
| +
|
| +
|
| +def run_test(config, shell_args, apps_and_args, run_launcher=False):
|
| + """Run the given test, using mojo_launcher if |run_launcher| is True."""
|
| + if (config.target_os == Config.OS_ANDROID):
|
| + return run_test_android(shell_args, apps_and_args)
|
| + else:
|
| + executable = get_shell_executable(config, run_launcher)
|
| + command = ([executable] + build_shell_arguments(shell_args, apps_and_args))
|
| + return subprocess.check_output(command, stderr=subprocess.STDOUT)
|
| +
|
| +
|
| +def try_run_test(config, shell_args, apps_and_args, run_launcher):
|
| + """Returns the output of a command line or an empty string on error."""
|
| + command_line = build_command_line(config, shell_args, apps_and_args,
|
| + run_launcher=run_launcher)
|
| + _logging.debug("Running command line: %s" % command_line)
|
| + try:
|
| + return run_test(config, shell_args, apps_and_args, run_launcher)
|
| + except Exception as e:
|
| + print_process_error(command_line, e)
|
| + return None
|
|
|