from helper import login, logout
import time
import pytest
from selenium.webdriver.common.by import By
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC


def test_admin_can_create_lecturer_account(driver, url, admin_username, admin_password, restore_database):
    # Administrator (admin@qq.com, password 123) logs in
    driver.maximize_window()
    login(driver, url, admin_username, admin_password)

    # Create a Lecturer account for Mr Lan (mrlan@qq.com, password [123Abc!])
    tab = driver.find_element(By.ID, 'tab_ins_accounts')
    tab.click()
    elem = driver.find_element(By.NAME, 'fullname')
    elem.send_keys('Mr Lan')
    elem = driver.find_element(By.NAME, 'email')
    elem.send_keys('mrlan@qq.com')
    elem = driver.find_element(By.NAME, 'password')
    elem.send_keys('123Abc!!')
    radio_button = driver.find_element(By.NAME, 'type')
    radio_button.click()
    button = driver.find_element(By.NAME, 'create_btn')
    button.click()

    # Log out Admin account
    logout(driver)

    # Log in Lecturer account
    login(driver, url, 'mrlan@qq.com', '123Abc!!')
    elems = driver.find_elements(By.CLASS_NAME, 'nav-link')
    assert '(Lecturer)' in elems[0].text
    assert 'Mr Lan' in elems[0].text
    driver.quit()


def test_lecturer_can_create_course(driver, url, restore_database):
    # Lecturer lanhui@qq.com logs in
    driver.maximize_window()
    login(driver, url, 'lanhui@qq.com', '123')

    # Create a course called CSC1001 Advanced Software Engineering, 2024
    elem = driver.find_element(By.NAME, 'name')
    elem.send_keys('Advanced Software Engineering')
    elem = driver.find_element(By.NAME, 'code')
    elem.send_keys('CSC1001')
    elem = driver.find_element(By.NAME, 'academic')
    elem.send_keys('2004')
    elem = driver.find_element(By.NAME, 'faculty')
    elem.send_keys('School of Computer Science and Technology')
    elem = driver.find_element(By.CLASS_NAME, 'btn-primary')
    elem.click()
    elems = driver.find_elements(By.CLASS_NAME, 'btn-default')
    last_elem = elems[-1]
    assert 'Advanced Software Engineering' in last_elem.text
    assert '(CSC1001)' in last_elem.text

    # Logout
    logout(driver)
    driver.quit()


def test_lecturer_can_post_assignment(driver, url, restore_database):
    # Lecturer lanhui@qq.com logs in
    driver.maximize_window()
    login(driver, url, 'lanhui@qq.com', '123')

    # Create an assignment called Take-home quiz 1 for course (CSC1111) - Project Management
    elems = driver.find_elements(By.CLASS_NAME, 'btn-default')
    elems[1].click()
    elem = driver.find_element(By.NAME, 'deadlinedate')
    elem.send_keys('002024/12/30')
    elem = driver.find_element(By.NAME, 'deadlinetime')
    elem.send_keys('23:59')
    elem = driver.find_element(By.NAME, 'title')
    elem.send_keys('Take-home quiz 1')
    elem = driver.find_element(By.NAME, 'instructions')
    elem.send_keys('This is a closed-book quiz.')
    elem = driver.find_element(By.NAME, 'marks')
    elem.send_keys('10')
    radio_button = driver.find_element(By.NAME, 'type')
    radio_button.click()
    elem = driver.find_element(By.CLASS_NAME, 'btn-primary')
    elem.click()

    # Check if the assignment has been successfully posted
    elem = driver.find_element(By.CLASS_NAME, 'card-title')
    assert 'Take-home quiz 1 (10 Marks, Individual)' in elem.text
    elem = driver.find_element(By.CLASS_NAME, 'text-muted')
    assert 'Deadline: 2024-12-30' in elem.text
    driver.quit()


def test_lecturer_can_add_student_numbers(driver, url, restore_database):
    # Lecturer lanhui@qq.com logs in
    driver.maximize_window()
    login(driver, url, 'lanhui@qq.com', '123')

    # Add ASE student numbers
    student_numbers = '''
    202420781739
    202420781740
    202420781741
    202420781742
    202420781743
    202420781745
    202420581366
    202420581368
    202420581369
    202420581370
    202420581372
    202420581373
    202420581374
    202420581376
    202420581378
    202420581381
    '''
    elem = driver.find_element(By.ID, 'admin_tab')
    elem.click()
    elem = driver.find_element(By.NAME, 'users')
    elem.send_keys(student_numbers)
    elem = driver.find_element(By.ID, 'register_btn')
    elem.click()

    elems = driver.find_elements(By.CSS_SELECTOR, 'p')
    added = 0
    student_lst = [number.strip() for number in student_numbers.strip().split('\n')]
    print(student_lst)
    for student_no in student_lst:
        for elem in elems:
            if student_no in elem.text and 'added' in elem.text:
                added += 1
                break
    assert added == len(student_lst)
    driver.quit()


def test_student_with_valid_student_number_can_sign_up(driver, url, restore_database):
    # Student with recognizable student number 202400000001 can sign up an account
    driver.get(url)
    driver.maximize_window()
    elem = driver.find_element(By.ID, 'signup_link')
    elem.click()
    elem = driver.find_element(By.NAME, 'fullname')
    elem.send_keys('Good Student')
    elem = driver.find_element(By.NAME, 'user_student_id')
    elem.send_keys('202400000001')
    elem = driver.find_element(By.NAME, 'email')
    elem.send_keys('goodstudent@qq.com')
    elem = driver.find_element(By.NAME, 'password')
    elem.send_keys('[123Abc]')
    elem = driver.find_element(By.NAME, 'confirmpassword')
    elem.send_keys('[123Abc]')
    elem = driver.find_element(By.ID, 'signup_btn')
    elem.click()
    logout(driver)

    # Log in Student account
    login(driver, url, '202400000001', '[123Abc]')
    elems = driver.find_elements(By.CLASS_NAME, 'nav-link')
    assert 'Student ID' in elems[0].text
    assert 'Good Student' in elems[0].text
    driver.quit()


def test_student_with_invalid_student_number_cannot_sign_up(driver, url, restore_database):
    # Student with unrecognizable student number cannot sign up an account
    driver.get(url)
    driver.maximize_window()
    elem = driver.find_element(By.ID, 'signup_link')
    elem.click()
    elem = driver.find_element(By.NAME, 'fullname')
    elem.send_keys('Good Student')
    elem = driver.find_element(By.NAME, 'user_student_id')
    elem.send_keys('202400000002')
    elem = driver.find_element(By.NAME, 'email')
    elem.send_keys('goodstudent@qq.com')
    elem = driver.find_element(By.NAME, 'password')
    elem.send_keys('[123Abc]')
    elem = driver.find_element(By.NAME, 'confirmpassword')
    elem.send_keys('[123Abc]')
    elem = driver.find_element(By.ID, 'signup_btn')
    elem.click()

    # Log in Student account
    login(driver, url, '202400000002', '[123Abc]')
    elems = driver.find_elements(By.CLASS_NAME, 'nav-link')
    assert not 'Student ID' in elems[0].text
    assert not 'Good Student' in elems[0].text
    driver.quit()


def test_student_can_join_course(driver, url, restore_database):
    # Student can join (CSC1111) - Project Management
    login(driver, url, '201825800050', '123')
    driver.maximize_window()

    # Search for CSC1111
    elem = driver.find_element(By.NAME, 'search')
    elem.send_keys('CSC1111')
    elem = driver.find_element(By.CLASS_NAME, 'btn-primary')
    elem.click()
    elems = driver.find_elements(By.CLASS_NAME, 'btn-default')
    assert 'CSC1111' in elems[0].text

    # Join
    elem = driver.find_element(By.CLASS_NAME, 'btn-success') # find the green Join button
    elem.click()

    # Log out, then log in to check the course-joining status
    logout(driver)
    login(driver, url, '201825800050', '123')
    elems = driver.find_elements(By.CLASS_NAME, 'btn-default')
    assert 'CSC1111' in elems[0].text
    assert 'Project Management' in elems[0].text
    assert 'Joined' in elems[0].text


@pytest.mark.skip()
def test_student_can_submit_assignment():
    # Student can submit Take-home quiz 1 for CSC1001
    assert True