po层

class LoginPage:
    def __init__(self, driver):
        self.driver = driver
        self.username_input = (By.CSS_SELECTOR, 'input[name="username"]')
        self.password_input = (By.CSS_SELECTOR, 'input[name="password"]')
        self.login_button = (By.CSS_SELECTOR, 'button[type="submit"]')

    def enter_username(self, username):
        WebDriverWait(self.driver, 10).until(EC.presence_of_element_located(self.username_input)).send_keys(username)

    def enter_password(self, password):
        WebDriverWait(self.driver, 10).until(EC.presence_of_element_located(self.password_input)).send_keys(password)

    def click_login(self):
        WebDriverWait(self.driver, 10).until(EC.element_to_be_clickable(self.login_button)).click()

class TestLogin:
    def setup_method(self):
        self.driver = webdriver.Chrome()
        self.login_page = LoginPage(self.driver)

    def teardown_method(self):
        self.driver.quit()

    def test_login_success(self):
        self.login_page.enter_username('valid_user')
        self.login_page.enter_password('valid_password')
        self.login_page.click_login()
        assert 'Dashboard' in self.driver.title