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 as page_module
6from telemetry.page import page_set as page_set_module
7
8
9class SafebrowsingPage(page_module.Page):
10
11  """
12  Why: Expect 'malware ahead' page. Use a short navigation timeout because no
13  response will be received.
14  """
15
16  def __init__(self, url, page_set):
17    super(SafebrowsingPage, self).__init__(url=url, page_set=page_set)
18    self.archive_data_file = '../data/chrome_proxy_safebrowsing.json'
19
20  def RunNavigateSteps(self, action_runner):
21    action_runner.NavigateToPage(self, timeout_in_seconds=5)
22
23
24class SafebrowsingPageSet(page_set_module.PageSet):
25
26  """ Chrome proxy test sites """
27
28  def __init__(self):
29    super(SafebrowsingPageSet, self).__init__(
30      archive_data_file='../data/chrome_proxy_safebrowsing.json')
31
32    self.AddPage(SafebrowsingPage('http://www.ianfette.org/', self))
33