testing/chromoting/multi_machine_example/example_test_controller.py

   1 #!/usr/bin/env python
   2 # Copyright 2015 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """The test controller for the chromoting localhost browser_tests.
   7
   8 This test uses the legion framework to setup this controller which will run
   9 the chromoting_browser_tests on a task machine. This is intended to
  10 be an example Legion-based test for the chromoting team.
  11
  12 The controller will start a task machine to run browser_tests_launcher on. The
  13 output of these tests are streamed back to the test controller to be output
  14 on the controller's stdout and stderr channels. The final test output is then
  15 read and becomes the final output of the controller, mirroring the test's
  16 pass/fail result.
  17 """
  18
  19 import argparse
  20 import logging
  21 import os
  22 import sys
  23 import time
  24
  25 # Map the legion directory so we can import the host controller.
  26 SRC_DIR = os.path.join('..', '..', '..')
  27 sys.path.append(os.path.join(SRC_DIR, 'testing'))
  28 from legion import test_controller
  29
  30
  31 class ExampleController(test_controller.TestController):
  32   """The test controller for the Chromoting browser_tests."""
  33
  34   def __init__(self):
  35     super(ExampleController, self).__init__()
  36     self.task = None
  37     self.args = None
  38
  39   def RunTest(self):
  40     """Main method to run the test code."""
  41     self.ParseArgs()
  42     self.CreateTask()
  43     self.TestIntegrationTests()
  44
  45   def CreateBrowserTestsLauncherCommand(self):
  46     return [
  47         'python',
  48         self.TaskAbsPath('../browser_tests_launcher.py'),
  49         '--commands_file', self.TaskAbsPath(self.args.commands_file),
  50         '--prod_dir', self.TaskAbsPath(self.args.prod_dir),
  51         '--cfg_file', self.TaskAbsPath(self.args.cfg_file),
  52         '--me2me_manifest_file', self.TaskAbsPath(
  53             self.args.me2me_manifest_file),
  54         '--it2me_manifest_file', self.TaskAbsPath(
  55             self.args.it2me_manifest_file),
  56         '--user_profile_dir', self.args.user_profile_dir,
  57         ]
  58
  59   def TaskAbsPath(self, path):
  60     """Returns the absolute path to the resource on the task machine.
  61
  62     Args:
  63       path: The relative path to the resource.
  64
  65     Since the test controller and the task machines run in different tmp dirs
  66     on different machines the absolute path cannot be calculated correctly on
  67     this machine. This function maps the relative path (from this directory)
  68     to an absolute path on the task machine.
  69     """
  70     return self.task.rpc.AbsPath(path)
  71
  72   def CreateTask(self):
  73     """Creates a task object and sets the proper values."""
  74     self.task = self.CreateNewTask(
  75         isolated_hash=self.args.task_machine,
  76         dimensions={'os': 'Ubuntu-14.04', 'pool': 'Chromoting'})
  77     self.task.Create()
  78     self.task.WaitForConnection()
  79
  80   def ParseArgs(self):
  81     """Gets the command line args."""
  82     parser = argparse.ArgumentParser()
  83     parser.add_argument('--task_machine',
  84                         help='isolated hash of the task machine.')
  85     # The rest of the args are taken from
  86     # testing/chromoting/browser_tests_launcher.py.
  87     parser.add_argument('-f', '--commands_file',
  88                         help='path to file listing commands to be launched.')
  89     parser.add_argument('-p', '--prod_dir',
  90                         help='path to folder having product and test binaries.')
  91     parser.add_argument('-c', '--cfg_file',
  92                         help='path to test host config file.')
  93     parser.add_argument('--me2me_manifest_file',
  94                         help='path to me2me host manifest file.')
  95     parser.add_argument('--it2me_manifest_file',
  96                         help='path to it2me host manifest file.')
  97     parser.add_argument(
  98         '-u', '--user_profile_dir',
  99         help='path to user-profile-dir, used by connect-to-host tests.')
 100     self.args, _ = parser.parse_known_args()
 101
 102   def TestIntegrationTests(self):
 103     """Runs the integration tests via browser_tests_launcher.py."""
 104     # Create a process object, configure it, and start it.
 105     # All interactions with the process are based on this "proc" key.
 106     proc = self.task.rpc.subprocess.Process(
 107         self.CreateBrowserTestsLauncherCommand())
 108     # Set the cwd to browser_tests_launcher relative to this directory.
 109     # This allows browser_test_launcher to use relative paths.
 110     self.task.rpc.subprocess.SetCwd(proc, '../')
 111     # Set the task verbosity to true to allow stdout/stderr to be echo'ed to
 112     # run_task's stdout/stderr on the task machine. This can assist in
 113     # debugging.
 114     self.task.rpc.subprocess.SetVerbose(proc)
 115     # Set the process as detached to create it in a new process group.
 116     self.task.rpc.subprocess.SetDetached(proc)
 117     # Start the actual process on the task machine.
 118     self.task.rpc.subprocess.Start(proc)
 119
 120     # Collect the stdout/stderr and emit it from this controller while the
 121     # process is running.
 122     while self.task.rpc.subprocess.Poll(proc) is None:
 123       # Output the test's stdout and stderr in semi-realtime.
 124       # This is not true realtime due to the RPC calls and the 1s sleep.
 125       stdout, stderr = self.task.rpc.subprocess.ReadOutput(proc)
 126       if stdout:
 127         sys.stdout.write(stdout)
 128       if stderr:
 129         sys.stderr.write(stderr)
 130       time.sleep(1)
 131
 132     # Get the return code, clean up the process object.
 133     returncode = self.task.rpc.subprocess.GetReturncode(proc)
 134     self.task.rpc.subprocess.Delete(proc)
 135
 136     # Pass or fail depending on the return code from the browser_tests_launcher.
 137     if returncode != 0:
 138       raise AssertionError('browser_tests_launcher failed with return code '
 139                            '%i' % returncode)
 140
 141
 142 if __name__ == '__main__':
 143   ExampleController().RunController()