import time
from locust import User, TaskSet, task, web, between
from locust import events


class MyTaskSet(TaskSet):
    @task(1)
    def foo(self):
        start_time = time.time()
        time.sleep(1)
        events.request_success.fire(
            request_type="http",
            name='foo',
            response_time=(time.time()-start_time) * 1000,
            response_length=0,
        )

    @task(1)
    def bar(self):
        start_time = time.time()
        time.sleep(1)
        events.request_failure.fire(
            request_type="http",
            name='bar',
            response_time=(time.time()-start_time) * 1000,
            response_length=0,
            exception='I broke',
        )


class Dummy(User):
    wait_time = between(2, 5)
    tasks = [MyTaskSet]