tools/perf/page_sets/profile_safe_urls.py

   1 # Copyright 2015 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4 from profile_creators import profile_safe_url_list
   5 from telemetry.page import page as page_module
   6 from telemetry.page import page_set as page_set_module
   7 from telemetry.page import shared_page_state
   8
   9
  10 class ProfileSafeUrlPage(page_module.Page):
  11   def __init__(self, url, page_set):
  12     super(ProfileSafeUrlPage, self).__init__(
  13         url=url,
  14         page_set = page_set,
  15         shared_page_state_class=shared_page_state.SharedDesktopPageState,
  16         credentials_path = 'data/credentials.json')
  17     self.credentials = 'google'
  18
  19
  20 class ProfileSafeUrlsPageSet(page_set_module.PageSet):
  21   """Safe urls used for profile generation."""
  22
  23   def __init__(self):
  24     super(ProfileSafeUrlsPageSet, self).__init__(
  25       archive_data_file='data/profile_safe_urls.json',
  26       bucket=page_set_module.PARTNER_BUCKET)
  27
  28     # Only use the first 500 urls to prevent the .wpr files from getting too
  29     # big.
  30     safe_urls = profile_safe_url_list.GetShuffledSafeUrls()[0:500]
  31     for safe_url in safe_urls:
  32       self.AddUserStory(ProfileSafeUrlPage(safe_url, self))