import os
import sys
sys.path.append(os.path.dirname(__file__) + '/..')

from common.context import Context
from common.storefront import Storefront
from locust import FastHttpUser, task, between

# Optional dependency
try:
    import locust_plugins
except ImportError:
    pass


context = Context()


class Visitor(FastHttpUser):
    wait_time = between(2, 5)
    weight = 1

    @task
    def listing(self):
        page = Storefront(self.client, context)

        # search products over listings
        page.go_to_listing()

        # take a look to the first two products
        page.view_products(2)
        page.go_to_next_page()

        # open two different product pages
        page.view_products(2)

        # sort listing and use properties to filter
        page.select_sorting()
        page.add_property_filter()

        page.view_products(1)
        page.go_to_next_page()

        # switch to search to find products
        page.do_search()
        page.view_products(2)

        # use property filter to find products
        page.add_property_filter()

        # take a look to the top three hits
        page.view_products(3)
        page.go_to_next_page()


class Nvidia(FastHttpUser):
    weight = 20

    def on_start(self):
        self.page = Storefront(self.client, context)
        self.page.register()
        self.page.logout()

    @task
    def follow_advertisement(self):
        self.page.login()
        self.page.add_advertisement()
        self.page.instant_order()
        self.page.logout()