from locust import HttpUser, between, task


class ServicePilotFrontendUser(HttpUser):
    wait_time = between(1, 2)

    pages = [
        ("/dashboard", "页面-工作台"),
        ("/tickets", "页面-工单管理"),
        ("/sla", "页面-SLA管理"),
        ("/knowledge", "页面-知识库"),
        ("/files", "页面-附件管理"),
        ("/reviews", "页面-满意度"),
        ("/test-plan", "页面-测试计划"),
    ]

    @task
    def visit_business_pages(self):
        for path, name in self.pages:
            with self.client.get(path, name=name, catch_response=True) as response:
                if response.status_code == 200:
                    response.success()
                else:
                    response.failure(f"{path} status={response.status_code}")