screenshot disabling policy tests

chrome/test/pyautolib/pyauto.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 """PyAuto: Python Interface to Chromium's Automation Proxy.
 
 PyAuto uses swig to expose Automation Proxy interfaces to Python.
 For complete documentation on the functionality available,
 run pydoc on this file.
@@ skipping 5770 matching lines @@
     for measurement_type in stats:
       values = stats[measurement_type]
       result[measurement_type] = {
         'min': min(values),
         'max': max(values),
         'end': values[-1],
       }
 
     return result
 
+  def RunAshCommand(self, action):
+    """Applies accelerator to ASH.
+
+    Apply the accelerator with given id to ASH (ChromeOS only). Action list can
+    be found in ash/accelerators/accelerator_table.h, AcceleratorAction enum.
+
+    Args:
+      action: accelerator code (eg. pyauto.TAKE_SCREENSHOT).
+
+    Returns:
+      A boolean indicating whether the accelerator was handled.
+    """
+    cmd_dict = {
+      'command': 'RunAshCommand',
+      'action': action
+    }
+    return self._GetResultFromJSONRequest(cmd_dict, windex=None)
+
   ## ChromeOS section -- end
 
 
 class ExtraBrowser(PyUITest):
   """Launches a new browser with some extra flags.
 
   The new browser is launched with its own fresh profile.
   This class does not apply to ChromeOS.
   """
   def __init__(self, chrome_flags=[], methodName='runTest', **kwargs):
@@ skipping 435 matching lines @@
     """Returns a list of tests loaded from the given args.
 
     The given args can be either a module (ex: module1) or a testcase
     (ex: module2.MyTestCase) or a test (ex: module1.MyTestCase.testX)
     If empty, the tests in the already imported modules are loaded.
 
     Args:
       args: [module1, module2, module3.testcase, module4.testcase.testX]
             These modules or test cases or tests should be importable
 
-      Returns:
-        a list of expanded test names.  Example:
-          [
-            'module1.TestCase1.testA',
-            'module1.TestCase1.testB',
-            'module2.TestCase2.testX',
-            'module3.testcase.testY',
-            'module4.testcase.testX'
-          ]
+    Returns:
+      a list of expanded test names.  Example:
+        [
+          'module1.TestCase1.testA',
+          'module1.TestCase1.testB',
+          'module2.TestCase2.testX',
+          'module3.testcase.testY',
+          'module4.testcase.testX'
+        ]
     """
     if not args:  # Load tests ourselves
       if self._HasTestCases('__main__'):    # we are running a test script
         module_name = os.path.splitext(os.path.basename(sys.argv[0]))[0]
         args.append(module_name)   # run the test cases found in it
       else:  # run tests from the test description file
         pyauto_tests_file = os.path.join(self.TestsDir(), self._tests_filename)
         logging.debug("Reading %s", pyauto_tests_file)
         if not os.path.exists(pyauto_tests_file):
           logging.warn("%s missing. Cannot load tests.", pyauto_tests_file)
@@ skipping 94 matching lines @@
     successful = result.wasSuccessful()
     if not successful:
       pyauto_tests_file = os.path.join(self.TestsDir(), self._tests_filename)
       print >>sys.stderr, 'Tests can be disabled by editing %s. ' \
                           'Ref: %s' % (pyauto_tests_file, _PYAUTO_DOC_URL)
     sys.exit(not successful)
 
 
 if __name__ == '__main__':
   Main()