debugging - 如何在轨迹测试中保留 session 和 CSRF token

Question

我不想用 locust.io 测试我的 django web 应用程序。在 ha 形式中，我遇到了使用 CSRF token 保护它的问题。我执行以下操作:

class WebsiteTasks(TaskSet):
    def on_start(self):
        print("On start")

    @task
    def post_answer(self):
        self.client.get("/polls/2/vote")
        self.client.post("/polls/2/vote/", {"choice": "8"})

为什么我会收到 403错误？该帖子被禁止，蝗虫文档说客户端对象使 session 保持事件状态..

Answer 1

将您的代码更改为:

@task
def post_answer(self):
    response = self.client.get("/polls/2/vote")
    csrftoken = response.cookies['csrftoken']

    self.client.post("/polls/2/vote/", 
                     {"choice": "8"}, 
                     headers={"X-CSRFToken": csrftoken})