Create an apptesting framework for dart.

mojo/tools/mopy/test_util.py

+# 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