tools/chrome_proxy/common/chrome_proxy_benchmark.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 common import chrome_proxy_measurements as measurements
   6 from telemetry import benchmark
   7
   8
   9 class ChromeProxyBenchmark(benchmark.Benchmark):
  10   @classmethod
  11   def AddCommandLineArgs(cls, parser):
  12     parser.add_option(
  13         '--extra-chrome-proxy-via-header',
  14         type='string', dest="extra_header",
  15         help='Adds an expected Via header for the Chrome-Proxy tests.')
  16
  17   @classmethod
  18   def ProcessCommandLineArgs(cls, parser, args):
  19     if args.extra_header:
  20       measurements.ChromeProxyValidation.extra_via_header = args.extra_header
  21