Adding pyauto-based perf tests for WebGL demos Aquarium and Field.

chrome/test/functional/perf.py

 #!/usr/bin/python
 # Copyright (c) 2011 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.
 
 """Basic pyauto performance tests.
 
 For tests that need to be run for multiple iterations (e.g., so that average
 and standard deviation values can be reported), the default number of iterations
 run for each of these tests is specified by |_DEFAULT_NUM_ITERATIONS|.
@@ skipping 583 matching lines @@
          cpu_usage2['system'] + cpu_usage2['idle']
     y1 = cpu_usage1['user'] + cpu_usage1['nice'] + \
          cpu_usage1['system'] + cpu_usage1['idle']
     total_frames = total_shown_frames + total_dropped_frames
     # Counting extrapolation for utilization to play the video
     self._PrintSummaryResults('YoutubeCPUExtrapolation',
         [((float(x2) - x1) / (y2 - y1)) * total_frames/total_shown_frames],
         'extrapolation')
 
 
+class WebGLTest(BasePerfTest):
+  """Tests for WebGL performance."""
+
+  def _RunWebGLTest(self, url, description):
+    """Measures FPS using a specified WebGL demo.
+
+    Args:
+      url: The string URL that, once loaded, will run the WebGL demo (default
+           WebGL demo settings are used, since this test does not modify any
+           settings in the demo).
+      description: A string description for this demo, used as a performance
+                   value description.  Should not contain any spaces.
+    """
+    self.assertTrue(self.AppendTab(pyauto.GURL(url)),
+                    msg='Failed to append tab for %s.' % description)
+
+    get_fps_js = """
+      var fps_field = document.getElementById("fps");
+      var result = -1;
+      if (fps_field)
+        result = fps_field.innerHTML;
+      window.domAutomationController.send(JSON.stringify(result));
+    """
+
+    # Wait until we start getting FPS values.
+    self.assertTrue(
+        self.WaitUntil(
+            lambda: self.ExecuteJavascript(get_fps_js, tab_index=1) != '-1',
+            timeout=300, retry_sleep=1),
+        msg='Timed out when waiting for FPS values to be available.')
+
+    # Let the experiment run for 5 seconds before we start collecting perf
+    # measurements.
+    time.sleep(5)
+
+    # Collect the current FPS value each second for the next 30 seconds.  The
+    # final result of this test will be the average of these FPS values.
+    fps_vals = []
+    for _ in xrange(30):
+      fps = self.ExecuteJavascript(get_fps_js, tab_index=1)
+      fps_vals.append(float(fps.replace('"', '')))
+      time.sleep(1)
+    self._PrintSummaryResults(description, fps_vals, 'fps')
+
+  def testWebGLAquarium(self):
+    """Measures performance using the WebGL Aquarium demo."""
+    self._RunWebGLTest(
+        self.GetFileURLForDataPath('pyauto_private', 'webgl', 'aquarium',
+                                   'aquarium.html'),
+        'WebGLAquarium')
+
+  def testWebGLField(self):
+    """Measures performance using the WebGL Field demo."""
+    self._RunWebGLTest(
+        self.GetFileURLForDataPath('pyauto_private', 'webgl', 'field',
+                                   'field.html'),
+        'WebGLField')
+
+
 class FileUploadDownloadTest(BasePerfTest):
   """Tests that involve measuring performance of upload and download."""
 
   def setUp(self):
     """Performs necessary setup work before running each test in this class."""
     self._temp_dir = tempfile.mkdtemp()
     self._test_server = PerfTestServer(self._temp_dir)
     self._test_server_port = self._test_server.GetPort()
     self._test_server.Run()
     self.assertTrue(self.WaitUntil(self._IsTestServerRunning),
@@ skipping 644 matching lines @@
     """Identifies the port number to which the server is currently bound.
 
     Returns:
       The numeric port number to which the server is currently bound.
     """
     return self._server.server_address[1]
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()