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 |