tools/playback_benchmark/run.py

#!/usr/bin/env python
#
# Copyright 2010 Google Inc. All Rights Reserved.

"""Script for test playback.

Prerequisites:
1. OpenSSL library - http://www.openssl.org/
2. Python interface to the OpenSSL library - https://launchpad.net/pyopenssl

Example usage:
python run.py -t <test_dir>
"""

from optparse import OptionParser

import playback_driver
import proxy_handler


def Run(options):
  driver = playback_driver.PlaybackRequestHandler(options.test_dir)
  httpd = proxy_handler.CreateServer(driver, options.port)
  sa = httpd.socket.getsockname()
  print "Serving HTTP on", sa[0], "port", sa[1], "..."
  httpd.serve_forever()


if __name__ == '__main__':
  parser = OptionParser()
  parser.add_option("-t", "--test-dir", dest="test_dir",
                    help="directory containing recorded test data")
  parser.add_option("-p", "--port", dest="port", type="int", default=8000)
  options = parser.parse_args()[0]
  if not options.test_dir:
    raise Exception('please specify test directory')

  Run(options)