from django.contrib.auth import get_user_model
from django.test import TestCase, LiveServerTestCase
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
unit_name = ""
class UnitLiveTests(LiveServerTestCase):
    def setUp(self):
        User = get_user_model()
        self.user_admin = User.objects.create_superuser(username='admin',
                                 email='admin@decent.mark',
                                 password='password')
        self.browser = webdriver.Firefox()

    def tearDown(self):
        self.browser.quit()

    def perform_login(self):
        self.browser.get(self.live_server_url + '/accounts/login/')

        username = self.browser.find_element_by_id("id_username").send_keys("admin")

        password = self.browser.find_element_by_id ("id_password").send_keys("password")

        self.browser.find_element_by_xpath("/html/body/form[1]/input[2]").click()

    def test_login_redirects_to_unit_list(self):
        self.perform_login()
        self.assertEqual(self.live_server_url + "/", self.browser.current_url, "At the Unit List page")

    def add_unit(self):
        self.perform_login()
        self.browser.find_element_by_link_text("Create new unit").click()
        unit_name = "unit 123"
        start_date = "2018-10-09"
        end_date = "2018-10-10"
        description = "description"
        self.browser.find_element_by_id("id_name").send_keys(unit_name)
        self.browser.find_element_by_id("id_start").send_keys(start_date)
        self.browser.find_element_by_id("id_end").send_keys(end_date)
        self.browser.find_element_by_id("id_description").send_keys(description)
        self.browser.find_element_by_xpath("/html/body/form[1]/button").click()

    def test_adding_unit(self):
        self.perform_login()
        self.browser.find_element_by_link_text("Create new unit").click()
        unit_name = "unit 123"
        start_date = "2018-10-09"
        end_date = "2018-10-10"
        description = "description"
        self.browser.find_element_by_id("id_name").send_keys(unit_name)
        self.browser.find_element_by_id("id_start").send_keys(start_date)
        self.browser.find_element_by_id("id_end").send_keys(end_date)
        self.browser.find_element_by_id("id_description").send_keys(description)
        self.browser.find_element_by_xpath("/html/body/form[1]/button").click()
        self.assertEqual(self.live_server_url + "/", self.browser.current_url, "At the Unit List page")

    def test_adding_assignment(self):
        self.perform_login()
        self.add_unit()
        self.browser.find_element_by_xpath("//a[1]").click()
        self.browser.find_element_by_link_text("Assignments").click()
        self.browser.find_element_by_link_text("Create new assignment").click()
        ass_name = "assignment 123"
        start_date = "2018-10-09"
        end_date = "2018-10-10"
        description = "description"
        attempts = "3"
        total = "10"
        test = "testing"
        solution = "solution"
        template = "template here"
        self.browser.find_element_by_id("id_name").send_keys(ass_name)
        self.browser.find_element_by_id("id_start").send_keys(start_date)
        self.browser.find_element_by_id("id_end").send_keys(end_date)
        self.browser.find_element_by_id("id_description").send_keys(description)
        self.browser.find_element_by_id("id_attempts").clear()
        self.browser.find_element_by_id("id_total").clear()
        self.browser.find_element_by_id("id_attempts").send_keys(attempts)
        self.browser.find_element_by_id("id_total").send_keys(total)
        self.browser.find_element_by_id("id_test").send_keys(test)
        self.browser.find_element_by_id("id_solution").send_keys(solution)
        self.browser.find_element_by_id("id_template").send_keys(template)
        self.browser.find_element_by_xpath("/html/body/form[1]/button").click()
        flag = 0
        if (self.browser.title.find("Assignments") != -1):
            flag=1
        self.assertEqual(flag,1, "At the Assignments Page")