mirror of https://github.com/locustio/locust.git
27 lines
639 B
Python
27 lines
639 B
Python
# locustfile.py
|
|
|
|
from locust import HttpUser, TaskSet, between, task
|
|
|
|
USER_CREDENTIALS = [
|
|
("user1", "password"),
|
|
("user2", "password"),
|
|
("user3", "password"),
|
|
]
|
|
|
|
|
|
class UserBehaviour(TaskSet):
|
|
def on_start(self):
|
|
if len(USER_CREDENTIALS) > 0:
|
|
user, passw = USER_CREDENTIALS.pop()
|
|
self.client.post("/login", {"username": user, "password": passw})
|
|
|
|
@task
|
|
def some_task(self):
|
|
# user should be logged in here (unless the USER_CREDENTIALS ran out)
|
|
self.client.get("/protected/resource")
|
|
|
|
|
|
class User(HttpUser):
|
|
tasks = [UserBehaviour]
|
|
wait_time = between(5, 60)
|