Add logs for update_engine and image_to_live to test artifacts.

bin/au_test_harness/au_worker.py

 # Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """Module that contains the interface for au_test_harness workers.
 
 An au test harnss worker is a class that contains the logic for performing
 and validating updates on a target.  This should be subclassed to handle
 various types of target.  Types of targets include VM's, real devices, etc.
 """
@@ skipping 174 matching lines @@
                                                     private_key_path)
     cache_path = self.update_cache[update_id]
     if cache_path:
       update_url = dev_server_wrapper.DevServerWrapper.GetDevServerURL(
           proxy_port, cache_path)
       cmd.append('--update_url=%s' % update_url)
     else:
       cmd.append('--image=%s' % image_path)
       if src_image_path: cmd.append('--src_image=%s' % src_image_path)
 
-  def RunUpdateCmd(self, cmd):
+  def RunUpdateCmd(self, cmd, log_directory=None):
     """Runs the given update cmd given verbose options.
 
     Raises an update_exception.UpdateException if the update fails.
     """
     if self.verbose:
       try:
-        cros_lib.RunCommand(cmd)
+        if log_directory:
+          cros_lib.RunCommand(cmd, log_to_file=os.path.join(log_directory,
+                                                            'update.log'))
+        else:
+          cros_lib.RunCommand(cmd)
       except Exception as e:
         Warning(str(e))
         raise update_exception.UpdateException(1, str(e))
     else:
       (code, stdout, stderr) = cros_lib.RunCommandCaptureOutput(cmd)
       if code != 0:
         Warning(stdout)
         raise update_exception.UpdateException(code, stdout)
 
   def AssertEnoughTestsPassed(self, unittest, output, percent_required_to_pass):
@@ skipping 16 matching lines @@
     return percent_passed
 
   def InitializeResultsDirectory(self):
     """Called by a test to initialize a results directory for this worker."""
     # Use the name of the test.
     test_name = inspect.stack()[1][3]
     self.results_directory = os.path.join(self.test_results_root, test_name)
     self.results_count = 0
 
   def GetNextResultsPath(self, label):
-    """Returns a new results path based for this label.
+    """Returns a path for the results directory for this label.
 
     Prefixes directory returned for worker with time called i.e. 1_label,
-    2_label, etc.
+    2_label, etc.  The directory returned is outside the chroot so if passing
+    to an script that is called with enther_chroot, make sure to use
+    ReinterpretPathForChroot.
     """
     self.results_count += 1
-    return os.path.join(self.results_directory, '%s_%s' % (self.results_count,
-                                                           label))
+    dir = os.path.join(self.results_directory, '%s_%s' % (self.results_count,
+                                                          label))
+    if not os.path.exists(dir):
+      os.makedirs(dir)
+
+    return dir
 
   # --- PRIVATE HELPER FUNCTIONS ---
 
   def _ParseGenerateTestReportOutput(self, output):
     """Returns the percentage of tests that passed based on output."""
     percent_passed = 0
     lines = output.split('\n')
 
     for line in lines:
       if line.startswith("Total PASS:"):
         # FORMAT: ^TOTAL PASS: num_passed/num_total (percent%)$
         percent_passed = line.split()[3].strip('()%')
         cros_lib.Info('Percent of tests passed %s' % percent_passed)
         break
 
     return int(percent_passed)