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
   8
   9 class ProfileSafeUrlPage(page_module.Page):
  10   def __init__(self, url, page_set):
  11     super(ProfileSafeUrlPage, self).__init__(
  12         url=url,
  13         page_set = page_set,
  14         credentials_path = 'data/credentials.json')
  15     self.credentials = 'google'
  16
  17
  18 class ProfileSafeUrlsPageSet(page_set_module.PageSet):
  19   """Safe urls used for profile generation."""
  20
  21   def __init__(self):
  22     super(ProfileSafeUrlsPageSet, self).__init__(
  23       archive_data_file='data/profile_safe_urls.json',
  24       user_agent_type='desktop',
  25       bucket=page_set_module.PARTNER_BUCKET)
  26
  27     # Only use the first 500 urls to prevent the .wpr files from getting too
  28     # big.
  29     safe_urls = profile_safe_url_list.GetShuffledSafeUrls()[0:500]
  30     for safe_url in safe_urls:
  31       self.AddUserStory(ProfileSafeUrlPage(safe_url, self))