[Telemetry] Add trivial pages page set to BattOr benchmarks.

tools/perf/page_sets/mac_gpu_sites.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.
 from telemetry.page import page as page_module
 from telemetry.page import shared_page_state
 from telemetry import story
 
 
 class _NoGpuSharedPageState(shared_page_state.SharedPageState):
   def __init__(self, test, finder_options, story_set):
@@ skipping 12 matching lines @@
 
 
 class _NoWebGLImageChromiumSharedPageState(shared_page_state.SharedPageState):
   def __init__(self, test, finder_options, story_set):
     super(_NoWebGLImageChromiumSharedPageState, self).__init__(
       test, finder_options, story_set)
     finder_options.browser_options.AppendExtraBrowserArgs(
       ['--disable-webgl-image-chromium'])
 
 
-class TrivialScrollingPage(page_module.Page):
+class _BasePage(page_module.Page):
+  def __init__(self, page_set, shared_page_state_class, url, name, wait):
+    super(_BasePage, self).__init__(
+      url=url, page_set=page_set, name=name,
+      shared_page_state_class=shared_page_state_class)
+    self._wait = wait
 
-  def __init__(self, page_set, shared_page_state_class):
+  def RunPageInteractions(self, action_runner):
+    if self._wait:
+      action_runner.Wait(self._wait)
+
+
+class TrivialScrollingPage(_BasePage):
+
+  def __init__(self, page_set, shared_page_state_class, wait):
     super(TrivialScrollingPage, self).__init__(
         url='file://trivial_sites/trivial_scrolling_page.html',
         page_set=page_set,
         name=self.__class__.__name__ + shared_page_state_class.__name__,
-        shared_page_state_class=shared_page_state_class)
+        shared_page_state_class=shared_page_state_class,
+        wait=wait)
 
 
-class TrivialBlinkingCursorPage(page_module.Page):
+class TrivialBlinkingCursorPage(_BasePage):
 
-  def __init__(self, page_set, shared_page_state_class):
+  def __init__(self, page_set, shared_page_state_class, wait):
     super(TrivialBlinkingCursorPage, self).__init__(
         url='file://trivial_sites/trivial_blinking_cursor.html',
         page_set=page_set,
         name=self.__class__.__name__ + shared_page_state_class.__name__,
-        shared_page_state_class=shared_page_state_class)
+        shared_page_state_class=shared_page_state_class,
+        wait=wait)
 
 
-class TrivialCanvasPage(page_module.Page):
+class TrivialCanvasPage(_BasePage):
 
-  def __init__(self, page_set, shared_page_state_class):
+  def __init__(self, page_set, shared_page_state_class, wait):
     super(TrivialCanvasPage, self).__init__(
         url='file://trivial_sites/trivial_canvas.html',
         page_set=page_set,
         name=self.__class__.__name__ + shared_page_state_class.__name__,
-        shared_page_state_class=shared_page_state_class)
+        shared_page_state_class=shared_page_state_class,
+        wait=wait)
 
 
-class TrivialWebGLPage(page_module.Page):
+class TrivialWebGLPage(_BasePage):
 
-  def __init__(self, page_set, shared_page_state_class):
+  def __init__(self, page_set, shared_page_state_class, wait):
     super(TrivialWebGLPage, self).__init__(
         url='file://trivial_sites/trivial_webgl.html',
         page_set=page_set,
         name=self.__class__.__name__ + shared_page_state_class.__name__,
-        shared_page_state_class=shared_page_state_class)
+        shared_page_state_class=shared_page_state_class,
+        wait=wait)
 
 
-class TrivialBlurAnimationPage(page_module.Page):
+class TrivialBlurAnimationPage(_BasePage):
 
-  def __init__(self, page_set, shared_page_state_class):
+  def __init__(self, page_set, shared_page_state_class, wait):
     super(TrivialBlurAnimationPage, self).__init__(
         url='file://trivial_sites/trivial_blur_animation.html',
         page_set=page_set,
         name=self.__class__.__name__ + shared_page_state_class.__name__,
-        shared_page_state_class=shared_page_state_class)
+        shared_page_state_class=shared_page_state_class,
+        wait=wait)
 
 
-class TrivialFullscreenVideoPage(page_module.Page):
+class TrivialFullscreenVideoPage(_BasePage):
 
-  def __init__(self, page_set, shared_page_state_class):
+  def __init__(self, page_set, shared_page_state_class, wait):
     super(TrivialFullscreenVideoPage, self).__init__(
         url='file://trivial_sites/trivial_fullscreen_video.html',
         page_set=page_set,
         name=self.__class__.__name__ + shared_page_state_class.__name__,
-        shared_page_state_class=shared_page_state_class)
+        shared_page_state_class=shared_page_state_class,
+        wait=wait)
 
   def RunPageInteractions(self, action_runner):
     action_runner.PressKey("Return")
+    super(TrivialFullscreenVideoPage, self).RunPageInteractions(action_runner)
 
 
-class TrivialGifPage(page_module.Page):
+class TrivialGifPage(_BasePage):
 
-  def __init__(self, page_set, shared_page_state_class):
+  def __init__(self, page_set, shared_page_state_class, wait):
     super(TrivialGifPage, self).__init__(
         url='file://trivial_sites/trivial_gif.html',
         page_set=page_set,
         name=self.__class__.__name__ + shared_page_state_class.__name__,
-        shared_page_state_class=shared_page_state_class)
+        shared_page_state_class=shared_page_state_class,
+        wait=wait)
 
 
 class MacGpuTrivialPagesStorySet(story.StorySet):
 
-  def __init__(self):
+  def __init__(self, wait):

[erikchen, 2016/08/19 20:27:55]

Should this be a required argument? Do we need to modify benchmarks/power.py? 

[rnephew (Reviews Here), 2016/08/19 20:52:45]

Woops, should be an optional argument. Fixed.

     super(MacGpuTrivialPagesStorySet, self).__init__(
         cloud_storage_bucket=story.PUBLIC_BUCKET)
-    self.AddStory(TrivialScrollingPage(self, shared_page_state.SharedPageState))
+    self.AddStory(TrivialScrollingPage(
+        self, shared_page_state.SharedPageState, wait))
     self.AddStory(TrivialBlinkingCursorPage(
-        self, shared_page_state.SharedPageState))
-    self.AddStory(TrivialCanvasPage(self, shared_page_state.SharedPageState))
-    self.AddStory(TrivialWebGLPage(self, shared_page_state.SharedPageState))
+        self, shared_page_state.SharedPageState, wait))
+    self.AddStory(TrivialCanvasPage(
+        self, shared_page_state.SharedPageState, wait))
+    self.AddStory(TrivialWebGLPage(
+        self, shared_page_state.SharedPageState, wait))
     self.AddStory(TrivialBlurAnimationPage(
-        self, shared_page_state.SharedPageState))
+        self, shared_page_state.SharedPageState, wait))
     self.AddStory(TrivialFullscreenVideoPage(
-        self, shared_page_state.SharedPageState))
-    self.AddStory(TrivialGifPage(self, shared_page_state.SharedPageState))
+        self, shared_page_state.SharedPageState, wait))
+    self.AddStory(TrivialGifPage(
+        self, shared_page_state.SharedPageState, wait))
 
-    self.AddStory(TrivialScrollingPage(self, _NoOverlaysSharedPageState))
-    self.AddStory(TrivialBlinkingCursorPage(self, _NoOverlaysSharedPageState))
-    self.AddStory(TrivialCanvasPage(self, _NoOverlaysSharedPageState))
-    self.AddStory(TrivialWebGLPage(self, _NoOverlaysSharedPageState))
-    self.AddStory(TrivialBlurAnimationPage(self, _NoOverlaysSharedPageState))
-    self.AddStory(TrivialFullscreenVideoPage(self, _NoOverlaysSharedPageState))
-    self.AddStory(TrivialGifPage(self, _NoOverlaysSharedPageState))
+    self.AddStory(TrivialScrollingPage(
+        self, _NoOverlaysSharedPageState, wait))
+    self.AddStory(TrivialBlinkingCursorPage(
+        self, _NoOverlaysSharedPageState, wait))
+    self.AddStory(TrivialCanvasPage(
+        self, _NoOverlaysSharedPageState, wait))
+    self.AddStory(TrivialWebGLPage(
+        self, _NoOverlaysSharedPageState, wait))
+    self.AddStory(TrivialBlurAnimationPage(
+        self, _NoOverlaysSharedPageState, wait))
+    self.AddStory(TrivialFullscreenVideoPage(
+        self, _NoOverlaysSharedPageState, wait))
+    self.AddStory(TrivialGifPage(
+        self, _NoOverlaysSharedPageState, wait))
 
-    self.AddStory(TrivialScrollingPage(self, _NoGpuSharedPageState))
-    self.AddStory(TrivialBlinkingCursorPage(self, _NoGpuSharedPageState))
-    self.AddStory(TrivialCanvasPage(self, _NoGpuSharedPageState))
-    self.AddStory(TrivialWebGLPage(self, _NoWebGLImageChromiumSharedPageState))
-    self.AddStory(TrivialBlurAnimationPage(self, _NoGpuSharedPageState))
-    self.AddStory(TrivialFullscreenVideoPage(self, _NoGpuSharedPageState))
-    self.AddStory(TrivialGifPage(self, _NoGpuSharedPageState))
+    self.AddStory(TrivialScrollingPage(
+        self, _NoGpuSharedPageState, wait))
+    self.AddStory(TrivialBlinkingCursorPage(
+        self, _NoGpuSharedPageState, wait))
+    self.AddStory(TrivialCanvasPage(
+        self, _NoGpuSharedPageState, wait))
+    self.AddStory(TrivialWebGLPage(
+        self, _NoWebGLImageChromiumSharedPageState, wait))
+    self.AddStory(TrivialBlurAnimationPage(
+        self, _NoGpuSharedPageState, wait))
+    self.AddStory(TrivialFullscreenVideoPage(
+        self, _NoGpuSharedPageState, wait))
+    self.AddStory(TrivialGifPage(
+        self, _NoGpuSharedPageState, wait))
 
   @property
   def allow_mixed_story_states(self):
     # Return True here in order to be able to add the same tests with
     # a different SharedPageState.
     return True