Adding CPU performance test for Youtube video

functional/youtube.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.
 
+import re
 import time
 
 import pyauto_functional
 import pyauto
 
 
 class YoutubeTestHelper():
   """Helper functions for Youtube tests.
      
   For sample usage, look at class YoutubeTest.
@@ skipping 22 matching lines @@
     """), expect_retval='player ready')
 
   def GetPlayerState(self):
     """Returns a player state"""
     js = """
         var val = ytplayer.getPlayerState(); 
         window.domAutomationController.send(val + '');
     """
     return self._pyauto.ExecuteJavascript(js)
 
+  def GetVideoInfo(self):
+    """Returns video info"""

[Nirnimesh, 2011/11/15 02:13:50]

What info? Include sample.

[rohitbm, 2011/11/15 03:10:50]

This API is undocumented, so do we want to include sample?

+    youtube_apis = self._pyauto.GetPrivateInfo()['youtube_api']
+    youtube_debug_text = youtube_apis['GetDebugText']
+    video_data = self._pyauto.ExecuteJavascript(
+        'window.domAutomationController.send(%s);' % youtube_debug_text)
+    return video_data

[Nirnimesh, 2011/11/15 02:13:50]

merge with previous line

[rohitbm, 2011/11/15 03:10:50]

Done.

+
+  def GetVideoDroppedFrames(self):
+    """Returns total Youtube video dropped frames.
+  
+    Returns:
+      -1 if failed to get video frames from the video data
+    """
+    video_data = self._pyauto.GetVideoInfo()
+    matched = re.search('droppedFrames=([\d\.]+)', video_data)
+    if matched:
+      return int(matched.group(1))
+    else:
+      return -1
+
+  def GetVideoFrames(self):
+    """Returns Youtube video frames/second

[Nirnimesh, 2011/11/15 02:13:50]

end with . for consistency

[rohitbm, 2011/11/15 03:10:50]

Done.

+
+    Returns:
+      -1 if failed to get droppd frames from the video data.
+    """
+    video_data = self._pyauto.GetVideoInfo()
+    matched = re.search('videoFps=([\d\.]+)', video_data)
+    if matched:
+      return int(matched.group(1))
+    else:
+      return -1
+
   def GetVideoTotalBytes(self):
-    """Returns video size in bytes
+    """Returns video total size in bytes
        
     To call this function, video must be in the paying state,
     or this returns 0.
     """
     total_bytes = 0
     total_bytes = self._pyauto.ExecuteJavascript("""
         bytes = ytplayer.getVideoBytesTotal();
         window.domAutomationController.send(bytes + '');
     """)
     return int(total_bytes)
 
+  def GetVideoLoadedBytes(self):
+    """Returns video size in bytes."""
+    loaded_bytes = 0
+    loaded_bytes = self.ExecuteJavascript("""
+        bytes = ytplayer.getVideoBytesLoaded();
+        window.domAutomationController.send(bytes + '');
+    """)
+    return int(loaded_bytes)
+
   def PlayVideo(self):
     """Plays the loaded video"""
     self._pyauto.ExecuteJavascript("""
         ytplayer.playVideo();
         window.domAutomationController.send('');
     """)
 
+  def PauseVideo(self):
+    """Pause the video"""
+    self.ExecuteJavascript("""
+        ytplayer.pauseVideo();
+        window.domAutomationController.send('');
+    """)
+
   def AssertPlayingState(self):
     """Assert player's playing state"""
     self._pyauto.assertTrue(self._pyauto.WaitUntil(self._pyauto.GetPlayerState,
         expect_retval=self._pyauto.is_playing),
         msg='Player did not enter the playing state')
 
   def PlayVideoAndAssert(self):
     """Start video and assert the playing state"""
     self._pyauto.assertTrue(self._pyauto.IsFlashPluginEnabled(), 
-                    msg='From here Flash plugin is disabled or not available') 
+        msg='From here Flash plugin is disabled or not available')
     url = self._pyauto.GetHttpURLForDataPath('media', 'youtube.html')
     self._pyauto.NavigateToURL(url)
     self._pyauto.assertTrue(self._pyauto.WaitUntilPlayerReady(),
                             msg='Failed to load YouTube player')
     self._pyauto.PlayVideo()
     self._pyauto.AssertPlayingState()
 
 
 class YoutubeTest(pyauto.PyUITest, YoutubeTestHelper):
   """Test case for Youtube videos."""
 
   def __init__(self, methodName='runTest', **kwargs):
     pyauto.PyUITest.__init__(self, methodName, **kwargs)
     YoutubeTestHelper.__init__(self, self)
 
   def testPlayerStatus(self):
     """Test that YouTube loads a player and changes player states 
         
     Test verifies various player states like unstarted, playing, paused
     and ended.
     """
     # Navigating to Youtube video. This video is 122 seconds long.
     # During tests, we are not goinig to play this video full.
     self.PlayVideoAndAssert()
-    # Pause the playing video
-    self.ExecuteJavascript("""
-        ytplayer.pauseVideo();
-        window.domAutomationController.send('');
-    """)
+    self.PauseVideo()
     self.assertEqual(self.GetPlayerState(), self.is_paused,
                      msg='Player did not enter the paused state')
     # Seek to the end of video
     self.ExecuteJavascript("""
         val = ytplayer.getDuration();
         ytplayer.seekTo(val, true);
         window.domAutomationController.send('');
     """)
     self.PlayVideo()
     # We've seeked to almost the end of the video but not quite.
@@ skipping 26 matching lines @@
     self.PlayVideoAndAssert()
     total_bytes = self.GetVideoTotalBytes()
     prev_loaded_bytes = 0
     loaded_bytes = 0
     count = 0
     while loaded_bytes < total_bytes:
       # We want to test bytes loading only twice
       count = count + 1
       if count == 2:
         break
-      loaded_bytes = self.ExecuteJavascript("""
-          bytes = ytplayer.getVideoBytesLoaded();
-          window.domAutomationController.send(bytes + '');
-      """)
-      loaded_bytes = int(loaded_bytes)
+      loaded_bytes = self.GetVideoLoadedBytes()
       self.assertTrue(prev_loaded_bytes <= loaded_bytes)
       prev_loaded_bytes = loaded_bytes
       # Give some time to load a video
       time.sleep(1)
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()