tools/perf/benchmarks/pywebsocket_server.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 import os
   6 import sys
   7
   8 from telemetry.core import local_server
   9 from telemetry.core import util
  10
  11
  12 # This invokes pywebsocket's standalone.py under third_party/pywebsocket
  13 class PywebsocketServerBackend(local_server.LocalServerBackend):
  14   def __init__(self):
  15     super(PywebsocketServerBackend, self).__init__()
  16     self.port = 8001
  17     self.base_dir = os.path.relpath(
  18         os.path.join(util.GetChromiumSrcDir(),
  19                      'third_party', 'pywebsocket', 'src'),
  20         start=util.GetTelemetryDir())
  21
  22   def StartAndGetNamedPorts(self, args):
  23     return [local_server.NamedPort('http', self.port)]
  24
  25   def ServeForever(self):
  26     os.chdir(self.base_dir)
  27     cmd = [
  28         sys.executable, '-m', 'mod_pywebsocket.standalone',
  29         '--port', str(self.port),
  30         '--log-level', 'debug',
  31         '-d', 'example'
  32         ]
  33     os.execv(sys.executable, cmd)
  34
  35 class PywebsocketServer(local_server.LocalServer):
  36   def __init__(self):
  37     super(PywebsocketServer, self).__init__(PywebsocketServerBackend)
  38
  39   def GetBackendStartupArgs(self):
  40     return {}