src/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):
    super(_NoGpuSharedPageState, self).__init__(
      test, finder_options, story_set)
    finder_options.browser_options.AppendExtraBrowserArgs(
      ['--disable-gpu'])


class _NoOverlaysSharedPageState(shared_page_state.SharedPageState):
  def __init__(self, test, finder_options, story_set):
    super(_NoOverlaysSharedPageState, self).__init__(
      test, finder_options, story_set)
    finder_options.browser_options.AppendExtraBrowserArgs(
      ['--disable-mac-overlays'])


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 _BasePage(page_module.Page):
  def __init__(
      self, page_set, shared_page_state_class, url, name, wait_in_seconds):
    super(_BasePage, self).__init__(
      url=url, page_set=page_set, name=name,
      shared_page_state_class=shared_page_state_class)
    self._wait_in_seconds = wait_in_seconds

  def RunPageInteractions(self, action_runner):
    action_runner.Wait(self._wait_in_seconds)


class TrivialScrollingPage(_BasePage):

  def __init__(self, page_set, shared_page_state_class, wait_in_seconds):
    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,
        wait_in_seconds=wait_in_seconds)


class TrivialBlinkingCursorPage(_BasePage):

  def __init__(self, page_set, shared_page_state_class, wait_in_seconds):
    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,
        wait_in_seconds=wait_in_seconds)


class TrivialCanvasPage(_BasePage):

  def __init__(self, page_set, shared_page_state_class, wait_in_seconds):
    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,
        wait_in_seconds=wait_in_seconds)


class TrivialWebGLPage(_BasePage):

  def __init__(self, page_set, shared_page_state_class, wait_in_seconds):
    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,
        wait_in_seconds=wait_in_seconds)


class TrivialBlurAnimationPage(_BasePage):

  def __init__(self, page_set, shared_page_state_class, wait_in_seconds):
    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,
        wait_in_seconds=wait_in_seconds)


class TrivialFullscreenVideoPage(_BasePage):

  def __init__(self, page_set, shared_page_state_class, wait_in_seconds):
    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,
        wait_in_seconds=wait_in_seconds)

  def RunPageInteractions(self, action_runner):
    action_runner.PressKey("Return")
    super(TrivialFullscreenVideoPage, self).RunPageInteractions(action_runner)


class TrivialGifPage(_BasePage):

  def __init__(self, page_set, shared_page_state_class, wait_in_seconds):
    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,
        wait_in_seconds=wait_in_seconds)


class MacGpuTrivialPagesStorySet(story.StorySet):

  def __init__(self, wait_in_seconds=0):
    # Wait is time to wait_in_seconds on page in seconds.
    super(MacGpuTrivialPagesStorySet, self).__init__(
        cloud_storage_bucket=story.PUBLIC_BUCKET)
    self.AddStory(TrivialScrollingPage(
        self, shared_page_state.SharedPageState, wait_in_seconds))
    self.AddStory(TrivialBlinkingCursorPage(
        self, shared_page_state.SharedPageState, wait_in_seconds))
    self.AddStory(TrivialCanvasPage(
        self, shared_page_state.SharedPageState, wait_in_seconds))
    self.AddStory(TrivialWebGLPage(
        self, shared_page_state.SharedPageState, wait_in_seconds))
    self.AddStory(TrivialBlurAnimationPage(
        self, shared_page_state.SharedPageState, wait_in_seconds))
    self.AddStory(TrivialFullscreenVideoPage(
        self, shared_page_state.SharedPageState, wait_in_seconds))
    self.AddStory(TrivialGifPage(
        self, shared_page_state.SharedPageState, wait_in_seconds))

    self.AddStory(TrivialScrollingPage(
        self, _NoOverlaysSharedPageState, wait_in_seconds))
    self.AddStory(TrivialBlinkingCursorPage(
        self, _NoOverlaysSharedPageState, wait_in_seconds))
    self.AddStory(TrivialCanvasPage(
        self, _NoOverlaysSharedPageState, wait_in_seconds))
    self.AddStory(TrivialWebGLPage(
        self, _NoOverlaysSharedPageState, wait_in_seconds))
    self.AddStory(TrivialBlurAnimationPage(
        self, _NoOverlaysSharedPageState, wait_in_seconds))
    self.AddStory(TrivialFullscreenVideoPage(
        self, _NoOverlaysSharedPageState, wait_in_seconds))
    self.AddStory(TrivialGifPage(
        self, _NoOverlaysSharedPageState, wait_in_seconds))

    self.AddStory(TrivialScrollingPage(
        self, _NoGpuSharedPageState, wait_in_seconds))
    self.AddStory(TrivialBlinkingCursorPage(
        self, _NoGpuSharedPageState, wait_in_seconds))
    self.AddStory(TrivialCanvasPage(
        self, _NoGpuSharedPageState, wait_in_seconds))
    self.AddStory(TrivialWebGLPage(
        self, _NoWebGLImageChromiumSharedPageState, wait_in_seconds))
    self.AddStory(TrivialBlurAnimationPage(
        self, _NoGpuSharedPageState, wait_in_seconds))
    self.AddStory(TrivialFullscreenVideoPage(
        self, _NoGpuSharedPageState, wait_in_seconds))
    self.AddStory(TrivialGifPage(
        self, _NoGpuSharedPageState, wait_in_seconds))

  @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