Locust User Simulation with Random User Agents

This code defines a Locust user class to simulate user visits to a website and randomly set user agents. It uses some basic features of the Locust library, including defining user classes, tasks, and HTTP requests.

Technology Stack : Locust, HttpUser, task, client.get, headers, random.choice

Code Type : The type of code

Code Difficulty : Advanced

                
                    
def random_user_agent():
    import random
    import locust
    from locust import HttpUser, task

    class WebsiteUser(HttpUser):
        @task
        def visit_website(self):
            response = self.client.get("/")

        @task
        def random_user_agent(self):
            user_agents = [
                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3",
                "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/14.0.3 Safari/605.1.15",
                "Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:52.0) Gecko/20100101 Firefox/52.0"
            ]
            self.client.headers['User-Agent'] = random.choice(user_agents)

    locust.main(["locustfile.py"])