locustfile_start.py

from locust import HttpUser, task, between
from random import randint


class AwesomeApplication(HttpUser):
    wait_time = between(1, 5)

    @task
    def hello(self):
        self.client.get("/hello")

    @task(3)
    def world(self):
        self.client.get("/world")

    @task
    def square(self):
        self.client.get(f"/square/{randint(0, 100)}", headers={"Authorization": f'Bearer {self.token}'})

    def on_start(self):
        # Here you can specify your code that you want to run before running the actual
        # test cases
        response = self.client.post("/login", json={"username": "test", "password": "test"})
        self.token = response.json()['access_token']