DO NOT SUBMIT: BattOr stress test

common/battor/battor/battor_wrapper.py

 # Copyright 2016 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 datetime
 import os
 import logging
 import platform
 import random
 import subprocess
@@ skipping 183 matching lines @@
   def GetShellReturnCode(self):
     """Gets the return code of the BattOr agent shell."""
     rc = self._battor_shell.poll()
     return rc
 
   def StartShell(self):
     """Start BattOr binary shell."""
     assert not self._battor_shell, 'Attempting to start running BattOr shell.'
 
     battor_cmd = [self._battor_agent_binary]
-    if self._serial_log_bucket:
-      # Create and immediately close a temp file in order to get a filename
-      # for the serial log.
-      self._serial_log_file = tempfile.NamedTemporaryFile(delete=False)
-      self._serial_log_file.close()
-      battor_cmd.append('--battor-serial-log=%s' % self._serial_log_file.name)
+    # if self._serial_log_bucket:
+    # Create and immediately close a temp file in order to get a filename
+    # for the serial log.
+    self._serial_log_file = tempfile.NamedTemporaryFile(delete=False)
+    self._serial_log_file.close()
+    print "Serial log at %s" % (self._serial_log_file.name)
+    battor_cmd.append('--battor-serial-log=%s' % self._serial_log_file.name)
     if self._battor_path:
       battor_cmd.append('--battor-path=%s' % self._battor_path)
     self._battor_shell = self._StartShellImpl(battor_cmd)
     assert self.GetShellReturnCode() is None, 'Shell failed to start.'
 
   def StopShell(self, timeout=None):
     """Stop BattOr binary shell."""
     assert self._battor_shell, 'Attempting to stop a non-running BattOr shell.'
     assert not self._tracing, 'Attempting to stop a BattOr shell while tracing.'
     timeout = timeout if timeout else DEFAULT_SHELL_CLOSE_TIMEOUT_S
@@ skipping 133 matching lines @@
           'Outputted: %s' % (cmd, status))
     return status
 
   def _StartShellImpl(self, battor_cmd):
     return subprocess.Popen(
         battor_cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
         stderr=subprocess.STDOUT, shell=False)
 
   def _UploadSerialLogToCloudStorage(self):
     """Uploads the BattOr serial log to cloud storage."""
-    if not self._serial_log_file or not cloud_storage.IsNetworkIOEnabled():
-      return
+    return False
+    # if not self._serial_log_file
+    #   return
 
-    remote_path = ('battor-serial-log-%s-%d.txt' % (
-        datetime.datetime.now().strftime('%Y-%m-%d_%H-%M.txt'),
-        random.randint(1, 100000)))
+    # remote_path = ('battor-serial-log-%s-%d.txt' % (
+    #     datetime.datetime.now().strftime('%Y-%m-%d_%H-%M.txt'),
+    #     random.randint(1, 100000)))
 
-    try:
-      cloud_url = cloud_storage.Insert(
-          self._serial_log_bucket, remote_path, self._serial_log_file.name)
-      sys.stderr.write('View BattOr serial log at %s\n' % cloud_url)
-    except cloud_storage.PermissionError as e:
-      logging.error('Cannot upload BattOr serial log file to cloud storage due '
-                    'to permission error: %s' % e.message)
+    # try:
+    #   cloud_url = cloud_storage.Insert(
+    #       self._serial_log_bucket, remote_path, self._serial_log_file.name)
+    #   sys.stderr.write('View BattOr serial log at %s\n' % cloud_url)
+    # except cloud_storage.PermissionError as e:
+    #   logging.error('Cannot upload BattOr serial log file to cloud storage due '
+    #                 'to permission error: %s' % e.message)
 
   def GetFirmwareGitHash(self):
     """Gets the git hash for the BattOr firmware.
 
     Returns: Git hash for firmware currently on the BattOr.
         Also sets self._git_hash to this value.
 
     Raises: ValueException if the git hash is not in hex.
     """
     assert self._battor_shell, ('Must start shell before getting firmware git '
@@ skipping 34 matching lines @@
     except subprocess.CalledProcessError as e:
       raise BattOrFlashError('BattOr flash failed with return code %s.'
                              % e.returncode)
 
     self._git_hash = None
     return True
 
 
 class BattOrFlashError(Exception):
   pass