Load testing with Locust
http://locust.io
@locustio
Jonatan Heyman
@jonatanheyman
What is Locust?
- An open source load testing tool
- Write test scenarios in python code
- Uses gevent
- Distributed, can simulate millions
of simultaneous users
What Locust provides
to the developer
- A framework for simulating user behaviour
- An HTTP client (python-requests)
- A web interface that show statistics
- Ability to run your code distributed
across multiple machines
Locust's web interface
Writing Locust test scripts
from locust import HttpLocust, TaskSet, task
class UserBehaviour(TaskSet):
@task(30)
def index_page(self):
self.client.get("/")
@task(10)
def press_page(self):
self.client.get("/press")
class WebUser(HttpLocust):
task_set = UserBehaviour
min_wait = 5 * 1000
max_wait = 60 * 1000 $ locust -f locustfile.py --host http://instagram.comTaskSets can be nested
class ApiDocs(TaskSet):
@task
def overview(self):
self.client.get("/developer/")
@task
def authentication(self):
self.client.get("/developer/authentication")
@task
def endpoints(self):
self.client.get("/developer/endpoints")
@task
def stop(self):
self.interrupt()
class UserBehaviour(TaskSet):
tasks = {ApiDocs: 10}
@task(30)
def index_page(self):
self.client.get("/")
@task(10)
def press_page(self):
self.client.get("/press")
class WebUser(HttpLocust):
task_set = UserBehaviour
min_wait = 5 * 1000
max_wait = 60 * 1000
Parsing response data
import random
from locust import HttpLocust, TaskSet, task
from pyquery import PyQuery
class ApiDocs(TaskSet):
def on_start(self):
self.overview()
@task
def overview(self):
response = self.client.get("/developer/")
pq = PyQuery(response.content)
self.locust.doc_pages = []
for link in pq(".sidebar-content li a"):
self.locust.doc_pages.append(link.attrib["href"])
@task
def sub_page(self):
url = random.choice(self.locust_doc_pages)
self.client.get(url) Finally
- Defining your tests in code is great
-
Create realistic load tests by simulating user behaviour
Do not just test URL endpoints
- Locust can test any request/response based system
- Check out the docs
- $ pip install locustio