Stop the existing session if it exists before start a new session and add mirroring scenario for cp…

chrome/test/media_router/telemetry/benchmarks/pagesets/media_router_perf_pages.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 utils
 
 from telemetry import page
 from telemetry import story
 from benchmarks.pagesets import media_router_page
 from telemetry.core import exceptions
 from telemetry.page import shared_page_state
 
+SESSION_TIME = 300  # 5 minutes
 
 class SharedState(shared_page_state.SharedPageState):
   """Shared state that restarts the browser for every single story."""
 
   def __init__(self, test, finder_options, story_set):
     super(SharedState, self).__init__(
         test, finder_options, story_set)
 
   def DidRunStory(self, results):
     super(SharedState, self).DidRunStory(results)
@@ skipping 29 matching lines @@
     super(CastIdlePage, self).__init__(
         page_set=page_set,
         url='file://basic_test.html',
         shared_page_state_class=SharedState)
 
   def RunPageInteractions(self, action_runner):
     # Wait for 5s after Chrome is opened in order to get consistent results.
     action_runner.Wait(5)
     with action_runner.CreateInteraction('Idle'):
       action_runner.ExecuteJavaScript('collectPerfData();')
-      action_runner.Wait(300)
+      action_runner.Wait(SESSION_TIME)
 
 
 class CastFlingingPage(media_router_page.CastPage):
   """Cast page to fling a video to Chromecast device."""
 
   def __init__(self, page_set):
     super(CastFlingingPage, self).__init__(
         page_set=page_set,
         url='file://basic_test.html#flinging',
         shared_page_state_class=SharedState)
 
   def RunPageInteractions(self, action_runner):
-     # Wait for 5s after Chrome is opened in order to get consistent results.
+    sink_name = self._GetDeviceName()
+    # Wait for 5s after Chrome is opened in order to get consistent results.
     action_runner.Wait(5)
     with action_runner.CreateInteraction('flinging'):
+
       self._WaitForResult(
           action_runner,
           lambda: action_runner.EvaluateJavaScript('initialized'),
-          'Failed to initialize')
+          'Failed to initialize',
+          timeout=30)
+      self.CloseExistingRoute(action_runner, sink_name)
 
       # Start session
       action_runner.TapElement(selector='#start_session_button')
       self._WaitForResult(
           action_runner,
           lambda: len(action_runner.tab.browser.tabs) >= 2,
           'MR dialog never showed up.')
 
       # Wait for 2s to make sure the dialog is fully loaded.
       action_runner.Wait(2)
       for tab in action_runner.tab.browser.tabs:
         # Choose sink
         if tab.url == 'chrome://media-router/':
-          self.ChooseSink(tab, self._GetDeviceName())
+          self.ChooseSink(tab, sink_name)
 
       self._WaitForResult(
         action_runner,
         lambda: action_runner.EvaluateJavaScript('currentSession'),
          'Failed to start session',
          timeout=10)
 
       # Load Media
       self.ExecuteAsyncJavaScript(
           action_runner,
           'loadMedia("%s");' % utils.GetInternalVideoURL(),
           lambda: action_runner.EvaluateJavaScript('currentMedia'),
           'Failed to load media',
           timeout=120)
 
       action_runner.Wait(5)
       action_runner.ExecuteJavaScript('collectPerfData();')
-      action_runner.Wait(300)
+      action_runner.Wait(SESSION_TIME)
       # Stop session
       self.ExecuteAsyncJavaScript(
           action_runner,
           'stopSession();',
           lambda: not action_runner.EvaluateJavaScript('currentSession'),
           'Failed to stop session')
 
 
+class CastMirroringPage(media_router_page.CastPage):
+  """Cast page to mirror a tab to Chromecast device."""
+
+  def __init__(self, page_set):
+    super(CastMirroringPage, self).__init__(
+        page_set=page_set,
+        url='file://mirroring.html',
+        shared_page_state_class=SharedState)
+
+  def RunPageInteractions(self, action_runner):
+    sink_name = self._GetDeviceName()
+    # Wait for 5s after Chrome is opened in order to get consistent results.
+    action_runner.Wait(5)
+    with action_runner.CreateInteraction('mirroring'):
+      self.CloseExistingRoute(action_runner, sink_name)
+
+      # Start session
+      action_runner.TapElement(selector='#start_session_button')
+      self._WaitForResult(
+          action_runner,
+          lambda: len(action_runner.tab.browser.tabs) >= 2,
+          'MR dialog never showed up.')
+
+      # Wait for 2s to make sure the dialog is fully loaded.
+      action_runner.Wait(2)
+      for tab in action_runner.tab.browser.tabs:
+        # Choose sink
+        if tab.url == 'chrome://media-router/':
+          self.ChooseSink(tab, sink_name)
+
+      # Wait for 5s to make sure the route is created.
+      action_runner.Wait(5)
+      action_runner.TapElement(selector='#start_session_button')
+      action_runner.Wait(2)
+      for tab in action_runner.tab.browser.tabs:
+        if tab.url == 'chrome://media-router/':
+          if not self.CheckIfExistingRoute(tab, sink_name):
+            raise page.page_test.Failure('Failed to start mirroring session.')
+      action_runner.ExecuteJavaScript('collectPerfData();')
+      action_runner.Wait(SESSION_TIME)
+      self.CloseExistingRoute(action_runner, sink_name)
+
+
 class MediaRouterDialogPageSet(story.StorySet):
   """Pageset for media router dialog latency tests."""
 
   def __init__(self):
     super(MediaRouterDialogPageSet, self).__init__(
         cloud_storage_bucket=story.PARTNER_BUCKET)
     self.AddStory(CastDialogPage(self))
 
 
 class MediaRouterCPUMemoryPageSet(story.StorySet):
   """Pageset for media router CPU/memory usage tests."""
 
   def __init__(self):
     super(MediaRouterCPUMemoryPageSet, self).__init__(
         cloud_storage_bucket=story.PARTNER_BUCKET)
     self.AddStory(CastIdlePage(self))
     self.AddStory(CastFlingingPage(self))
+    self.AddStory(CastMirroringPage(self))