Chromium Code Reviews| 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..08d8be0b584fd4f818825e9f005ced1003f67b35 |
| --- /dev/null |
| +++ b/mojo/tools/mopy/test_util.py |
| @@ -0,0 +1,79 @@ |
| +# 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 re |
|
msw
2015/03/04 21:04:32
nit: not needed
|
| +import subprocess |
| +import sys |
|
msw
2015/03/04 21:04:32
nit: not needed
|
| + |
| +_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 |