# 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 py_utils

from telemetry.page import page
from telemetry.page import cache_temperature as cache_temperature_module
from telemetry.page import traffic_setting as traffic_setting_module
from telemetry.page import shared_page_state


_NAVIGATION_TIMEOUT = 180
_WEB_CONTENTS_TIMEOUT = 180

class PageCyclerStory(page.Page):

  def __init__(self, url, page_set,
               shared_page_state_class=shared_page_state.SharedDesktopPageState,
               cache_temperature=cache_temperature_module.ANY, name='',
               traffic_setting=traffic_setting_module.NONE, **kwargs):
    super(PageCyclerStory, self).__init__(
        url=url, page_set=page_set,
        shared_page_state_class=shared_page_state_class,
        cache_temperature=cache_temperature, name=name,
        traffic_setting=traffic_setting,
        **kwargs)
    if cache_temperature != cache_temperature_module.ANY:
      self.grouping_keys['cache_temperature'] = cache_temperature
    if traffic_setting != traffic_setting_module.NONE:
      self.grouping_keys['traffic_setting'] = traffic_setting

  def RunNavigateSteps(self, action_runner):
    url = self.file_path_url_with_scheme if self.is_file else self.url
    action_runner.Navigate(url,
                           self.script_to_evaluate_on_commit,
                           timeout_in_seconds=_NAVIGATION_TIMEOUT)

  def RunPageInteractions(self, action_runner):
    py_utils.WaitFor(action_runner.tab.HasReachedQuiescence,
                     _WEB_CONTENTS_TIMEOUT)
    py_utils.WaitFor(action_runner.tab.IsServiceWorkerActivatedOrNotRegistered,
                     _WEB_CONTENTS_TIMEOUT)