daphne/testproject/fabfile.py
Malyshev Artem 335cd1625e Correct benchmark test for recent channels version. (#487)
* Ignore Emacs backups.

* Do not override default websocket.connect handler.

Channels specification 1.0 requires that websocket.connect handler
returns meaningful message with {'accept': True} at least.

* Add rabbitmq channel layer settings.

* Add benchmark requirements in separate file.

* Add RabbitMQ infrastructure part.

* Adapt benchmark README for new docker layout.

* Adapt fabric deploy for new settings module.
2017-01-17 12:25:08 -08:00

68 lines
2.1 KiB
Python

from fabric.api import sudo, task, cd
# CHANNEL TASKS
@task
def setup_redis():
sudo("apt-get update && apt-get install -y redis-server")
sudo("sed -i -e 's/127.0.0.1/0.0.0.0/g' /etc/redis/redis.conf")
sudo("/etc/init.d/redis-server stop")
sudo("/etc/init.d/redis-server start")
@task
def setup_channels():
sudo("apt-get update && apt-get install -y git python-dev python-setuptools python-pip")
sudo("pip install -U pip")
sudo("pip install -U asgi_redis asgi_ipc git+https://github.com/django/daphne.git@#egg=daphne")
sudo("rm -rf /srv/channels")
sudo("git clone https://github.com/django/channels.git /srv/channels/")
with cd("/srv/channels/"):
sudo("python setup.py install")
@task
def run_daphne(redis_ip):
with cd("/srv/channels/testproject/"):
sudo("REDIS_URL=redis://%s:6379 daphne -b 0.0.0.0 -p 80 testproject.asgi.redis:channel_layer" % redis_ip)
@task
def run_worker(redis_ip):
with cd("/srv/channels/testproject/"):
sudo("REDIS_URL=redis://%s:6379 python manage.py runworker" % redis_ip)
# Current loadtesting setup
@task
def setup_load_tester(src="https://github.com/django/channels.git"):
sudo("apt-get update && apt-get install -y git nodejs && apt-get install npm")
sudo("npm install -g loadtest")
sudo("ln -s /usr/bin/nodejs /usr/bin/node")
# Run current loadtesting setup
# example usage: $ fab run_loadtest:http://127.0.0.1,rps=10 -i "id_rsa" -H ubuntu@example.com
@task
def run_loadtest(host, t=90):
sudo("loadtest -c 10 -t {t} {h}".format(h=host, t=t))
# Run current loadtesting setup
# example usage: $ fab run_loadtest:http://127.0.0.1,rps=10 -i "id_rsa" -H ubuntu@example.com
@task
def run_loadtest_rps(host, t=90, rps=200):
sudo("loadtest -c 10 --rps {rps} -t {t} {h}".format(h=host, t=t, rps=rps))
# Task that Andrew used for loadtesting earlier on
@task
def setup_tester():
sudo("apt-get update && apt-get install -y apache2-utils python3-pip")
sudo("pip3 -U pip autobahn twisted")
sudo("rm -rf /srv/channels")
sudo("git clone https://github.com/django/channels.git /srv/channels/")
@task
def shell():
sudo("bash")