diff options
Diffstat (limited to 'testing/web-platform/tests/tools/pywebsocket/src/example/bench_wsh.py')
-rw-r--r-- | testing/web-platform/tests/tools/pywebsocket/src/example/bench_wsh.py | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/testing/web-platform/tests/tools/pywebsocket/src/example/bench_wsh.py b/testing/web-platform/tests/tools/pywebsocket/src/example/bench_wsh.py new file mode 100644 index 000000000..5067ca7d8 --- /dev/null +++ b/testing/web-platform/tests/tools/pywebsocket/src/example/bench_wsh.py @@ -0,0 +1,60 @@ +# Copyright 2011, Google Inc. +# All rights reserved. +# +# Redistribution and use in source and binary forms, with or without +# modification, are permitted provided that the following conditions are +# met: +# +# * Redistributions of source code must retain the above copyright +# notice, this list of conditions and the following disclaimer. +# * Redistributions in binary form must reproduce the above +# copyright notice, this list of conditions and the following disclaimer +# in the documentation and/or other materials provided with the +# distribution. +# * Neither the name of Google Inc. nor the names of its +# contributors may be used to endorse or promote products derived from +# this software without specific prior written permission. +# +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR +# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, +# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, +# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + + +"""A simple load tester for WebSocket clients. + +A client program sends a message formatted as "<time> <count> <message>" to +this handler. This handler starts sending total <count> WebSocket messages +containing <message> every <time> seconds. <time> can be a floating point +value. <count> must be an integer value. +""" + + +import time + + +def web_socket_do_extra_handshake(request): + pass # Always accept. + + +def web_socket_transfer_data(request): + line = request.ws_stream.receive_message() + parts = line.split(' ') + if len(parts) != 3: + raise ValueError('Bad parameter format') + wait = float(parts[0]) + count = int(parts[1]) + message = parts[2] + for i in xrange(count): + request.ws_stream.send_message(message) + time.sleep(wait) + + +# vi:sts=4 sw=4 et |