import pytest
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import NoSuchElementException, UnexpectedAlertPresentException

# Credentials for login, make sure to create a user with Admin role for these test to work
username = "" # Admin user email
password = "" # Admin user password

driver = webdriver.Chrome()
driver.get("") # URL of the LRR

def login(driver, username, password):
    try:
        # Fill in the login form
        user_input = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.ID, "user_name"))
        )
        user_input.send_keys(username)

        password_input = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.ID, "user_password"))
        )
        password_input.send_keys(password)

        # Click the login button
        login_button = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.ID, "login_btn"))
        )
        login_button.click()

        # Wait for the admin_tab to become clickable
        admin_tab = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.ID, "admin_tab"))
        )

        return True

    except (NoSuchElementException, UnexpectedAlertPresentException) as e:
        return f"Error: {str(e)}"


@pytest.mark.parametrize("course_id, course_name, ta_name", [(1, "Teecloudy - Ashly Course Testing", "Mark")])
def test_assign_a_new_ta_to_a_course(course_id, course_name, ta_name):
    try:

        login(driver, username, password)

        admin_tab = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.ID, "admin_tab"))
        )
        admin_tab.click()

        # Locate the form and select the TA
        ta_form = WebDriverWait(driver, 15).until(
            EC.presence_of_element_located((By.XPATH, f"//form[@id='drop_menu_form_{course_id}']"))
        )

        ta_dropdown = Select(ta_form.find_element(By.XPATH, ".//select[@name='ta']"))
        ta_dropdown.select_by_visible_text(ta_name)

        # Submit the form using JavaScript
        driver.execute_script("arguments[0].submit();", ta_form)

        # find table courses
        table_courses = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.XPATH, ".//*[@id='tab-existing-courses']/table"))
        )
        # find the row with matching course_name
        course_row = table_courses.find_element(By.XPATH, f".//tr[td='{course_name}']")
        # find the column with TA name
        ta_column = course_row.find_element(By.XPATH, ".//td[4]")

        # assert the TA name in the column
        assert ta_name in ta_column.text, f"Error: TA name {ta_name} not found in the column {ta_column.text}"
        
        driver.quit()
        
        return True
    
    except NoSuchElementException as e:
        return f"Error: {str(e)}"
    except UnexpectedAlertPresentException as e:
        return f"Error: {str(e)}"
    except AssertionError as e:
        return f"Error: {str(e)}"
    except Exception as e:
        return f"Error: {str(e)}"


@pytest.mark.parametrize("course_id, course_name, ta_name", [(1, "Teecloudy - Ashly Course Testing", "Mark")])
def test_assign_the_same_ta_to_the_same_course_twice(course_id, course_name, ta_name, restore_database):
    try:
        
        login(driver, username, password)

        admin_tab = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.ID, "admin_tab"))
        )
        admin_tab.click()

        # find table courses
        table_courses_before = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.XPATH, ".//*[@id='tab-existing-courses']/table"))
        )
        # find the row with matching course_name
        course_row_before = table_courses_before.find_element(By.XPATH, f".//tr[td='{course_name}']")
        # find the column with TA name
        old_cell_content = course_row_before.find_element(By.XPATH, ".//td[4]")

        # Locate the form and select the TA
        ta_form = WebDriverWait(driver, 15).until(
            EC.presence_of_element_located((By.XPATH, f"//form[@id='drop_menu_form_{course_id}']"))
        )

        ta_dropdown = Select(ta_form.find_element(By.XPATH, ".//select[@name='ta']"))
        ta_dropdown.select_by_visible_text(ta_name)

        # Submit the form using JavaScript
        driver.execute_script("arguments[0].submit();", ta_form)

        # Wait for an expected alert and accept it
        WebDriverWait(driver, 10).until(EC.alert_is_present())
        alert = driver.switch_to.alert
        alert_text = alert.text
        alert.accept()

        # find table courses
        table_courses_after = WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.XPATH, ".//*[@id='tab-existing-courses']/table"))
        )
        # find the row with matching course_name
        course_row_after = table_courses_after.find_element(By.XPATH, f".//tr[td='{course_name}']")
        # find the column with TA name
        new_cell_content = course_row_after.find_element(By.XPATH, ".//td[4]")

        # assert the TA name in the column
        assert old_cell_content == new_cell_content, f"Error: TA name in the column has changed from {old_cell_content} to {new_cell_content}"
        assert alert_text == "The selected TA is already assigned to this course.", f"Error: Alert text is not as expected: {alert_text}"
        assert restore_database == None, f"Error: Database was not restored"

        driver.quit()

        return True, alert_text
    
    except NoSuchElementException as e:
        return f"Error: {str(e)}"
    except UnexpectedAlertPresentException as e:
        return f"Error: {str(e)}"
    except AssertionError as e:
        return f"Error: {str(e)}"
    except Exception as e:
        return f"Error: {str(e)}"