web自动化测试进阶篇05 ——— 界面交互场景测试

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

class BasePage:
    def __init__(self, driver):
        self.driver = driver

    def wait_for_element_to_be_visible(self, by, locator, timeout=10):
        wait = WebDriverWait(self.driver, timeout)
        return wait.until(EC.visibility_of_element_located((by, locator)))

class BaseTest:
    @classmethod
    def setUpClass(cls):
        cls.driver = webdriver.Chrome()
        cls.driver.implicitly_wait(5)
        cls.driver.maximize_window()

    @classmethod
    def tearDownClass(cls):
        cls.driver.quit()

from selenium.webdriver.common.by import By
from base import BasePage

class LoginPage(BasePage):
    def __init__(self, driver):
        super().__init__(driver)
        self.username_input = (By.ID, "username")
        self.password_input = (By.ID, "password")
        self.login_button = (By.ID, "login_button")

    def login(self, username, password):
        self.driver.get("https://example-bank.com/login")
        self.driver.find_element(*self.username_input).send_keys(username)
        self.driver.find_element(*self.password_input).send_keys(password)
        self.driver.find_element(*self.login_button).click()

class LoanApplicationPage(BasePage):
    def __init__(self, driver):
        super().__init__(driver)
        self.product_type_dropdown = (By.ID, "product_type")
        self.loan_amount_input = (By.ID, "loan_amount")
        self.repayment_period_dropdown = (By.ID, "repayment_period")
        self.upload_file_input = (By.ID, "upload_file")
        self.submit_button = (By.ID, "submit_button")

    def apply_for_loan(self, product_type, loan_amount, repayment_period, file_path):
        self.driver.get("https://example-bank.com/loan_application")
        self.driver.find_element(*self.product_type_dropdown).send_keys(product_type)
        self.driver.find_element(*self.loan_amount_input).send_keys(loan_amount)
        self.driver.find_element(*self.repayment_period_dropdown).send_keys(repayment_period)
        self.driver.find_element(*self.upload_file_input).send_keys(file_path)
        self.driver.find_element(*self.submit_button).click()

    def click_calculate_button(self):
        calculate_button = self.wait_for_element_to_be_visible(By.ID, "calculate_button")
        calculate_button.click()

    def select_repayment_period(self, period):
        repayment_period_dropdown = self.wait_for_element_to_be_visible(By.ID, "repayment_period")
        select = Select(repayment_period_dropdown)
        select.select_by_visible_text(period)

    def accept_risk_warning(self):
        accept_button = self.wait_for_element_to_be_visible(By.ID, "accept_button")
        accept_button.click()

	def risk_assessment(self):
		assessment_button = self.wait_for_element_to_be_visible(By.ID, "assessment_button")
		assessment_button.click()

import unittest
from base import BaseTest
from page_objects import LoginPage, LoanApplicationPage

class TestLoanApplication(BaseTest):
    def test_loan_application(self):
        login_page = LoginPage(self.driver)
        login_page.login("testuser", "testpass")

        loan_application_page = LoanApplicationPage(self.driver)
        loan_application_page.apply_for_loan("Personal Loan", "5000", "12 months", "path/to/uploaded/file.pdf")

        # 添加异步通信和显式等待，等待贷款申请状态显示并验证
        loan_application_page.wait_for_element_to_be_visible(By.ID, "loan_application_status", timeout=20)
        status = self.driver.find_element(By.ID, "loan_application_status").text
        assert "Successful" in status, "贷款申请状态错误"

        # 测试额度计算和还款计划展示
        loan_amount = "5000"
        repayment_period = "12 months"
        expected_loan_amount = "5000"
        expected_repayment_plan = "Monthly Installments: 12\nAmount per Installment: 416.67"
        loan_application_page.fill_loan_amount_and_repayment_period(loan_amount, repayment_period)
        loan_application_page.click_calculate_button()
        loan_application_page.wait_for_element_to_be_visible(By.ID, "calculated_loan_amount", timeout=10)
        calculated_loan_amount = self.driver.find_element(By.ID, "calculated_loan_amount").text
        assert calculated_loan_amount == expected_loan_amount, "额度计算错误"
        repayment_plan = self.driver.find_element(By.ID, "repayment_plan").text
        assert repayment_plan == expected_repayment_plan, "还款计划展示错误"

        # 处理风险提示
        risk_warning = loan_application_page.get_risk_warning_text()
        if risk_warning:
            loan_application_page.accept_risk_warning()
        else:
            # 如果没有风险提示，进行投资风险评估
            loan_application_page.risk_assessment()

        # 选择还款期限
        new_repayment_period = "24 months"
        loan_application_page.select_repayment_period(new_repayment_period)

        # 提交贷款申请
        loan_application_page.submit_loan_application()
        loan_application_page.wait_for_element_to_be_visible(By.ID, "loan_application_status", timeout=20)
        status = self.driver.find_element(By.ID, "loan_application_status").text
        assert "Successful" in status, "贷款申请状态错误"

if __name__ == "__main__":
    unittest.main()