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