import os
from locust import HttpLocust

# from locustfiles.common.dsslocust import DSSLocust
# from locustfiles import UploadTaskSet, NotifyTaskSet, SearchTaskSet, DownloadTaskSet, CheckoutTaskSet
from locustfiles import SearchTaskSet

HOST = os.getenv('TARGET_URL', "http://localhost")

# class UploadUser(DSSLocust):
#     min_wait = 1000
#     max_wait = 3000
#     task_set = UploadTaskSet
#     weight = 1
#     host = HOST
#
#
# class CheckoutUser(DSSLocust):
#     min_wait = 3000
#     max_wait = 3000
#     task_set = CheckoutTaskSet
#     weight = 3
#     host = HOST
#
#
# class DownloadUser(DSSLocust):
#     min_wait = 500
#     max_wait = 1000
#     task_set = DownloadTaskSet
#     weight = 3
#     host = HOST
#
#
# class NotifiedUser(DSSLocust):
#     min_wait = 1000
#     max_wait = 3000
#     task_set = NotifyTaskSet
#     weight = 2
#     host = HOST


class SearchUser(HttpLocust):
    min_wait = 250
    max_wait = 500
    task_set = SearchTaskSet
    weight = 4
    host = HOST

import os
from locust import HttpLocust

# from locustfiles.common.dsslocust import DSSLocust
# from locustfiles import UploadTaskSet, NotifyTaskSet, SearchTaskSet, DownloadTaskSet, CheckoutTaskSet
from locustfiles import SearchTaskSet

HOST = os.getenv('TARGET_URL', "http://localhost")

# class UploadUser(DSSLocust):
#     min_wait = 1000
#     max_wait = 3000
#     task_set = UploadTaskSet
#     weight = 1
#     host = HOST
#
#
# class CheckoutUser(DSSLocust):
#     min_wait = 3000
#     max_wait = 3000
#     task_set = CheckoutTaskSet
#     weight = 3
#     host = HOST
#
#
# class DownloadUser(DSSLocust):
#     min_wait = 500
#     max_wait = 1000
#     task_set = DownloadTaskSet
#     weight = 3
#     host = HOST
#
#
# class NotifiedUser(DSSLocust):
#     min_wait = 1000
#     max_wait = 3000
#     task_set = NotifyTaskSet
#     weight = 2
#     host = HOST


class SearchUser(HttpLocust):
    min_wait = 250
    max_wait = 500
    task_set = SearchTaskSet
    weight = 4
    host = HOST

# from locustfiles.common.dsslocust import DSSLocust # from locustfiles import UploadTaskSet, NotifyTaskSet, SearchTaskSet, DownloadTaskSet, CheckoutTaskSet # class UploadUser(DSSLocust): #     min_wait = 1000 #     max_wait = 3000 #     task_set = UploadTaskSet #     weight = 1 #     host = HOST # # # class CheckoutUser(DSSLocust): #     min_wait = 3000 #     max_wait = 3000 #     task_set = CheckoutTaskSet #     weight = 3 #     host = HOST # # # class DownloadUser(DSSLocust): #     min_wait = 500 #     max_wait = 1000 #     task_set = DownloadTaskSet #     weight = 3 #     host = HOST # # # class NotifiedUser(DSSLocust): #     min_wait = 1000 #     max_wait = 3000 #     task_set = NotifyTaskSet #     weight = 2 #     host = HOST