summaryrefslogtreecommitdiffstats
path: root/tools/perf/benchmarks/pywebsocket_server.py
blob: 351090048d8b93c937059ad7e6ba22b8d31e4d22 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# Copyright 2014 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import sys

from core import path_util

from telemetry.core import local_server
from telemetry.core import util


# This invokes pywebsocket's standalone.py under third_party/pywebsocket
class PywebsocketServerBackend(local_server.LocalServerBackend):

  def __init__(self):
    super(PywebsocketServerBackend, self).__init__()
    self.port = 8001
    self.base_dir = os.path.relpath(
        os.path.join(path_util.GetChromiumSrcDir(),
                     'third_party', 'pywebsocket', 'src'),
        start=util.GetTelemetryDir())

  def StartAndGetNamedPorts(self, args):
    return [local_server.NamedPort('http', self.port)]

  def ServeForever(self):
    os.chdir(self.base_dir)
    cmd = [
        sys.executable, '-m', 'mod_pywebsocket.standalone',
        '--port', str(self.port),
        '--log-level', 'debug',
        '-d', 'example'
    ]
    os.execv(sys.executable, cmd)


class PywebsocketServer(local_server.LocalServer):

  def __init__(self):
    super(PywebsocketServer, self).__init__(PywebsocketServerBackend)

  def GetBackendStartupArgs(self):
    return {}