from locust import HttpLocust, TaskSet, task

class GeneralBrowser(TaskSet):
    @task(10)
    def index(self):
        self.client.get("/")

    @task(8)
    def contactLA(self):
        self.client.get("/contact/los-angeles")

    @task(5)
    def corporatePrivateBookings(self):
        self.client.get("/corporate-private-bookings")

    @task(5)
    def filmTV(self):
        self.client.get("/film-and-tv")

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

    @task(5)
    def publishingOurBooks(self):
        self.client.get("/publishing/our-books")

    @task(3)
    def publishingAbout(self):
        self.client.get("/publishing/about")

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

    @task(4)
    def careersNashville(self):
        self.client.get("/careers/?region=nashville")
        


class WebsiteUser(HttpLocust):
    task_set = GeneralBrowser
    min_wait = 5000
    max_wait = 9000