PyAuto test for GPU memory consumption

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 3812 matching lines @@
         { 'renderer_id': 23567,
           'routing_id': 1,
           'fps': 29.404298782348633 }
     """
     cmd_dict = {  # Prepare command for the json interface.
       'command': 'GetFPS',
       'tab_index': tab_index,
     }
     return self._GetResultFromJSONRequest(cmd_dict, windex=windex)
 
+  def GetGpuMemStats(self, windex=0):
+    """Returns the current FPS associated with the renderer process for a tab.
+
+    FPS is the rendered frames per second.
+
+    Args:
+      window_index: The window index, default is 0.
+
+    Returns:
+      A dictionary containing GPU memory usage info.
+      Example:
+        { 'gpu_active': true,
+          'gpu_memory_bytes_total': 1073741824,
+          'gpu_memory_bytes_used': 16777216 }
+    """
+    cmd_dict = {  # Prepare command for the json interface.
+      'command': 'GetGpuMemStats',
+    }
+    return self._GetResultFromJSONRequest(cmd_dict, windex=windex)
+
   def IsFullscreenForBrowser(self, windex=0):
     """Returns true if the window is currently fullscreen and was initially
     transitioned to fullscreen by a browser (vs tab) mode transition."""
     return self._GetResultFromJSONRequest(
       { 'command': 'IsFullscreenForBrowser' },
       windex=windex).get('result')
 
   def IsFullscreenForTab(self, windex=0):
     """Returns true if fullscreen has been caused by a tab."""
     return self._GetResultFromJSONRequest(
@@ skipping 2159 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()