import time

from locust import task
from selenium.common.exceptions import WebDriverException

from wrapper_locust.browser import Browser


class TestClass(Browser):

    browser_type = 'chrome'
    url = 'https://locust.io/'

    def __init__(self, environment):
        super(TestClass, self).__init__(environment)

    def on_start(self):
        self.client.set_number_of_instances(self.environment.runner.user_count)
        self.client.get(self.url)

    def click_documentation(self):
        self.client.find_element_by_xpath("//a[text()='Documentation']").click()

    def click_edit_in_git(self):
        self.client.find_element_by_xpath("//a[contains(text(), ' Edit on GitHub')]").click()

    def close(self):
        try:
            self.client.close()
            self.client.quit()
        except WebDriverException:
            pass

    @task
    def load_testing(self):
        self.client.event_recorder(self.client, 'Clicking', 'Documentation', self.click_documentation)
        self.client.event_recorder(self.client, 'Clicking', 'Edit in git', self.click_edit_in_git)
        self.close()
        self.client.stop_execution()
        time.sleep(30)
        # locust -f test\testing_sample.py TestClass --headless -u 3 -r 3 --only-summary  --> command to run