from locust import HttpUser, task, between


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

    @task
    def translate(self):
        self.client.post('/api/translate', None,
                         json={
                            "from": "en",
                            "to": "ml",
                            "source": "One morning, when Gregor Samsa woke from"
                            "troubled dreams, he found himself tranformed in his"
                            "bed into a horrible vermin",
                         },
                         headers={
                             "authority": "opusmt.wmflabs.org",
                             "content-type": "application/json",
                             "origin": "https://opusmt.wmflabs.org",
                         },
        )