from django.test import TestCase
from selenium import webdriver
from selenium.webdriver.common.by import By
from pyvirtualdisplay import Display

from project.utils.selenium_utils import SeleniumExtensions as SE
from django.conf import settings


class SeleniumTestCase(TestCase):

    @classmethod
    def setUpClass(cls):
        super().setUpClass()
        cls.display = Display(visible=0, size=(1280, 1024))
        cls.display.start()
        cls.driver = webdriver.Chrome()

    @classmethod
    def tearDownClass(cls):
        cls.driver.close()
        cls.display.stop()
        super().tearDownClass()

    def get_url(self, url):
        return 'http://%s%s' % (settings.MAIN_HOST, url)

    def wait_elem_xpath(self, xpath, inside_el=None, wait_time=10):
        return SE.wait_elem(self.driver, (By.XPATH, xpath), inside_el, wait_time)

    def wait_elems_xpath(self, xpath, inside_el=None, wait_time=10):
        return SE.wait_elems(self.driver, (By.XPATH, xpath), inside_el, wait_time)

    def wait_elem_css(self, selector, inside_el=None, wait_time=10):
        return SE.wait_elem(self.driver, (By.CSS_SELECTOR, selector), inside_el, wait_time)

    def wait_elems_css(self, selector, inside_el=None, wait_time=10):
        return SE.wait_elems(self.driver, (By.CSS_SELECTOR, selector), inside_el, wait_time)

    def wait_elem_name(self, name, inside_el=None, wait_time=10):
        return SE.wait_elem(self.driver, (By.NAME, name), inside_el, wait_time)

    def wait_elems_name(self, name, inside_el=None, wait_time=10):
        return SE.wait_elems(self.driver, (By.NAME, name), inside_el, wait_time)