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
5from telemetry.page import page
6from telemetry.page import page_set
7from unittest_data.pages.external_page import ExternalPage
8
9
10class InternalPage(page.Page):
11  def __init__(self, ps):
12    super(InternalPage, self).__init__('file://bar.html', page_set=ps)
13
14class 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        super(Google, self).__init__('https://www.google.com', page_set=ps)
28
29      def RunGetActionRunner(self, action_runner):
30        return action_runner
31
32    self.AddPage(Google(self))
33    self.AddPage(InternalPage(self))
34    self.AddPage(ExternalPage(self))
35