tools/telemetry/unittest_data/test_page_set.py

   1 # Copyright 2014 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
   5 from telemetry.page import page
   6 from telemetry.page import page_set
   7 from unittest_data.pages.external_page import ExternalPage
   8
   9
  10 class InternalPage(page.Page):
  11   def __init__(self, ps):
  12     super(InternalPage, self).__init__('file://bar.html', page_set=ps)
  13
  14 class TestPageSet(page_set.PageSet):
  15   """A pageset for testing purpose"""
  16
  17   def __init__(self):
  18     super(TestPageSet, self).__init__(
  19       archive_data_file='data/test.json',
  20       credentials_path='data/credential',
  21       user_agent_type='desktop',
  22       bucket=page_set.PUBLIC_BUCKET)
  23
  24     #top google property; a google tab is often open
  25     class Google(page.Page):
  26       def __init__(self, ps):
  27         # pylint: disable=bad-super-call
  28         super(Google, self).__init__('https://www.google.com', page_set=ps)
  29
  30       def RunGetActionRunner(self, action_runner):
  31         return action_runner
  32
  33     self.AddPage(Google(self))
  34     self.AddPage(InternalPage(self))
  35     self.AddPage(ExternalPage(self))