from locust import HttpLocust

from tasksets.apollo_api.tasksets.generated_taskset import GeneratedTaskSet as Apollo_API_Tests
from tasksets.apollo_gate_api.tasksets.generated_taskset import GeneratedTaskSet as Apollo_Gate_API_Tests
from tasksets.dsp_api.tasksets.generated_taskset import GeneratedTaskSet as DSP_API_Tests
from tasksets.user_data_api.tasksets.generated_taskset import GeneratedTaskSet as UserData_API_Tests
from tasksets.vendor_api.tasksets.generated_taskset import GeneratedTaskSet as Vendor_API_Tests


class Apollo_API_User(HttpLocust):
    task_set = Apollo_API_Tests
    min_wait = 20 * 1000
    max_wait = 30 * 1000


class Apollo_Gate_API_User(HttpLocust):
    task_set = Apollo_Gate_API_Tests
    min_wait = 20 * 1000
    max_wait = 30 * 1000


class DSP_API_User(HttpLocust):
    task_set = DSP_API_Tests
    min_wait = 45 * 1000
    max_wait = 60 * 1000


class Vendor_API_User(HttpLocust):
    task_set = Vendor_API_Tests
    min_wait = 10 * 1000
    max_wait = 15 * 1000


class User_Data_API_User(HttpLocust):
    task_set = UserData_API_Tests
    min_wait = 20 * 1000
    max_wait = 30 * 1000