如何在使用Locust时登录Flask应用

pip install locust flask

from flask import Flask, request, session

app = Flask(__name__)
app.secret_key = 'your_secret_key'

@app.route('/login', methods=['POST'])
def login():
    username = request.form.get('username')
    password = request.form.get('password')

    # 验证用户名和密码
    if username == 'admin' and password == 'password':
        session['logged_in'] = True
        return '登录成功'
    else:
        return '用户名或密码错误'

if __name__ == '__main__':
    app.run()

@app.route('/locust/login', methods=['POST'])
def locust_login():
    username = request.form.get('username')
    password = request.form.get('password')

    # 调用用户登录的功能
    if username == 'admin' and password == 'password':
        session['logged_in'] = True
        return '登录成功'
    else:
        return '用户名或密码错误'

from locust import HttpUser, task, between

class User(HttpUser):
    wait_time = between(1, 3)

    @task
    def login(self):
        username = 'admin'
        password = 'password'

        # 发送POST请求到Flask应用的登录路由
        response = self.client.post('/locust/login', data={'username': username, 'password': password})

        # 检查登录是否成功
        if response.text == '登录成功':
            self.client.get('/')  # 访问其他页面
        else:
            self.client.get('/login')  # 重新登录

locust -f locustfile.py --host=http://localhost:5000 --users=10 --spawn-rate=2