from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from time import sleep

class BasePage:
    def __init__(self, driver):
        """
        Khởi tạo BasePage với một instance WebDriver.
        """
        self.driver = driver
        self.wait = WebDriverWait(driver, 10) # Khởi tạo WebDriverWait với thời gian chờ 10 giây

    def find_element(self, *locator):
        """
        Định vị và trả về một phần tử trên trang.
        Chờ cho đến khi phần tử hiện diện trong DOM.
        """
        return self.wait.until(EC.presence_of_element_located(locator))

    def switch_to_frame(self, *locator):
        """
        Chuyển đổi context của WebDriver sang một iframe cụ thể.
        Đầu tiên tìm iframe bằng locator, sau đó chuyển driver tới iframe đó.
        """
        frame_element = self.wait.until(EC.presence_of_element_located(locator))
        self.driver.switch_to.frame(frame_element)

    def launch_browser_and_navigate(self, url):
        """
        Mở trình duyệt, điều hướng đến URL được cung cấp và tối đa hóa cửa sổ.
        """
        print(f"Mở trình duyệt và điều hướng đến: {url}")
        self.driver.get(url)
        self.driver.maximize_window()
        sleep(2) # Tạm dừng để đảm bảo cửa sổ đã tối đa hóa và ổn định

from pages.BasePage import BasePage
from selenium.webdriver.common.by import By
from time import sleep
from selenium.webdriver.support import expected_conditions as EC

class LoginPage(BasePage):
    # Định nghĩa các locator dưới dạng hằng số để dễ quản lý
    FRAME_LOGIN = (By.XPATH, "//div[@id='loginDiv']/iframe")
    INPUT_USERNAME = (By.NAME, "email")
    INPUT_PASSWORD = (By.NAME, "password")
    BUTTON_LOGIN = (By.ID, "dologin")
    # Locator giả định cho thông báo lỗi sau khi đăng nhập thất bại
    # Cần điều chỉnh theo cấu trúc HTML thực tế của trang web
    ERROR_MESSAGE_LOCATOR = (By.XPATH, "//div[@class='ferrorhead' or contains(@class, 'error-message')]")

    def switch_to_login_frame(self):
        """
        Chuyển đổi context của WebDriver đến iframe chứa form đăng nhập.
        """
        self.switch_to_frame(*self.FRAME_LOGIN)
        print("Đã chuyển đổi sang iframe đăng nhập.")

    def enter_username(self, username):
        """
        Nhập tên người dùng vào trường input tương ứng.
        """
        self.find_element(*self.INPUT_USERNAME).send_keys(username)
        print(f"Đã nhập tên người dùng: {username}")

    def enter_password(self, password):
        """
        Nhập mật khẩu vào trường input tương ứng.
        """
        self.find_element(*self.INPUT_PASSWORD).send_keys(password)
        print(f"Đã nhập mật khẩu (không hiển thị chi tiết vì lý do bảo mật).")

    def click_login_button(self):
        """
        Nhấn vào nút đăng nhập.
        """
        self.find_element(*self.BUTTON_LOGIN).click()
        print("Đã nhấn nút đăng nhập.")

    def perform_login(self, username, password):
        """
        Thực hiện toàn bộ quá trình đăng nhập bao gồm:
        chuyển đổi iframe, nhập thông tin và nhấn nút đăng nhập.
        """
        print("Bắt đầu thực hiện thao tác đăng nhập.")
        self.switch_to_login_frame()
        self.enter_username(username)
        self.enter_password(password)
        self.click_login_button()
        sleep(3) # Chờ một chút để kết quả đăng nhập hiển thị hoặc thông báo lỗi xuất hiện

    def get_error_message(self):
        """
        Lấy văn bản thông báo lỗi nếu có trên trang.
        Trả về None nếu không tìm thấy thông báo lỗi.
        """
        try:
            error_element = self.wait.until(EC.visibility_of_element_located(self.ERROR_MESSAGE_LOCATOR))
            return error_element.text
        except Exception:
            return None # Không tìm thấy thông báo lỗi trong thời gian chờ

from pages.LoginPage import LoginPage
from selenium import webdriver
import unittest
from time import sleep

class LoginTest(unittest.TestCase):
    def setUp(self):
        """
        Thiết lập môi trường kiểm thử trước khi mỗi test case được chạy.
        Khởi tạo WebDriver và đối tượng LoginPage.
        """
        self.driver = webdriver.Chrome()
        self.base_url = "http://mail.163.com/"
        self.login_page = LoginPage(self.driver)
        self.login_page.launch_browser_and_navigate(self.base_url)
        sleep(2) # Đảm bảo trang đã tải xong trước khi bắt đầu test

    def test_invalid_login_credentials(self):
        """
        Kiểm thử trường hợp đăng nhập với thông tin không hợp lệ.
        Mong đợi một thông báo lỗi xuất hiện.
        """
        print("--- Bắt đầu kịch bản kiểm thử: Đăng nhập với thông tin không hợp lệ ---")
        invalid_username = "user_non_existent@163.com"
        invalid_password = "wrongpass123"

        self.login_page.perform_login(invalid_username, invalid_password)

        # Sau khi đăng nhập thất bại, kiểm tra xem thông báo lỗi có hiển thị và nội dung có đúng không.
        # Lưu ý: Cần xác định chính xác văn bản lỗi thực tế trên trang web.
        error_message = self.login_page.get_error_message()
        print(f"Thông báo lỗi nhận được: '{error_message}'")

        self.assertIsNotNone(error_message, "Thông báo lỗi phải hiển thị khi đăng nhập sai.")
        self.assertIn("Tài khoản hoặc mật khẩu không đúng", error_message or "",
                      "Nội dung thông báo lỗi không chính xác với kỳ vọng.")
        print("Kịch bản kiểm thử đăng nhập không hợp lệ hoàn tất.")

    # Bạn có thể thêm các test case khác như:
    # def test_valid_login(self):
    #     """
    #     Kiểm thử trường hợp đăng nhập thành công.
    #     """
    #     print("--- Bắt đầu kịch bản kiểm thử: Đăng nhập thành công ---")
    #     self.login_page.perform_login("valid_username@163.com", "valid_password")
    #     # Ví dụ kiểm tra URL chuyển hướng hoặc sự hiện diện của phần tử sau khi đăng nhập thành công
    #     # self.assertIn("welcome", self.driver.current_url)
    #     print("Kịch bản kiểm thử đăng nhập thành công hoàn tất.")

    def tearDown(self):
        """
        Dọn dẹp môi trường kiểm thử sau khi mỗi test case được chạy, đóng trình duyệt.
        """
        self.driver.quit()
        print("Đã đóng trình duyệt.")