1# Copyright (c) 2011 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"""Command for measuring how long pages take to load in a browser.
6
7Prerequisites:
8  1. The command_line package from tools/site_compare
9  2. Either the IE BHO or Firefox extension (or both)
10
11Installation:
12  1. Build the IE BHO, or call regsvr32 on a prebuilt binary
13  2. Add a file called "measurepageloadtimeextension@google.com" to
14     the default Firefox profile directory under extensions, containing
15     the path to the Firefox extension root
16
17Invoke with the command line arguments as documented within
18the command line.
19"""
20
21import command_line
22import win32process
23
24from drivers import windowing
25from utils import browser_iterate
26
27def CreateCommand(cmdline):
28  """Inserts the command and arguments into a command line for parsing."""
29  cmd = cmdline.AddCommand(
30    ["measure"],
31    "Measures how long a series of URLs takes to load in one or more browsers.",
32    None,
33    ExecuteMeasure)
34
35  browser_iterate.SetupIterationCommandLine(cmd)
36  cmd.AddArgument(
37    ["-log", "--logfile"], "File to write output", type="string", required=True)
38
39
40def ExecuteMeasure(command):
41  """Executes the Measure command."""
42
43  def LogResult(url, proc, wnd, result):
44    """Write the result of the browse to the log file."""
45    log_file.write(result)
46
47  log_file = open(command["--logfile"], "w")
48
49  browser_iterate.Iterate(command, LogResult)
50
51  # Close the log file and return. We're done.
52  log_file.close()
53