# 页面对象类
class LoginPage:
    def __init__(self, driver):
        self.driver = driver
        self.username_input = (By.ID, "username")
        self.password_input = (By.ID, "password")
        self.login_button = (By.ID, "login_btn")
        self.remember_checkbox = (By.ID, "remember")
        self.error_message = (By.CLASS_NAME, "error-msg")
        self.success_message = (By.CLASS_NAME, "success-msg")

    def enter_username(self, username):
        """输入用户名"""
        self.driver.find_element(*self.username_input).send_keys(username)

    def enter_password(self, password):
        """输入密码"""
        self.driver.find_element(*self.password_input).send_keys(password)

    def click_login(self):
        """点击登录按钮"""
        self.driver.find_element(*self.login_button).click()

    def check_remember(self):
        """勾选记住密码"""
        self.driver.find_element(*self.remember_checkbox).click()

    def get_error_message(self):
        """获取错误信息"""
        try:
            return self.driver.find_element(*self.error_message).text
        except:
            return None

    def is_login_success(self):
        """判断登录是否成功"""
        try:
            self.driver.find_element(*self.success_message)
            return True
        except:
            return False

    def login(self, username, password, remember=False):
        """完整登录流程"""
        self.enter_username(username)
        self.enter_password(password)
        if remember:
            self.check_remember()
        self.click_login()

# 测试用例类
class TestLogin:
    def setup_method(self):
        """测试前准备"""
        self.driver = webdriver.Chrome()
        self.driver.get("http://example.com/login")
        self.login_page = LoginPage(self.driver)

    def teardown_method(self):
        """测试后清理"""
        self.driver.quit()

    def test_login_success(self):
        """测试登录成功"""
        self.login_page.login("user001", "password123")
        assert self.login_page.is_login_success()

    def test_login_with_wrong_password(self):
        """测试密码错误"""
        self.login_page.login("user001", "wrongpass")
        error_msg = self.login_page.get_error_message()
        assert "密码错误" in error_msg

    def test_login_with_nonexistent_user(self):
        """测试用户不存在"""
        self.login_page.login("nonexistent", "password123")
        error_msg = self.login_page.get_error_message()
        assert "用户不存在" in error_msg

    def test_login_with_remember(self):
        """测试记住密码"""
        self.login_page.login("user001", "password123", remember=True)
        assert self.login_page.is_login_success()
        # 验证cookie是否保存
        cookies = self.driver.get_cookies()
        assert any(c['name'] == 'remember_token' for c in cookies)

# 配置文件（config.yaml）
test_config:
  base_url: "http://example.com"
  browser: "chrome"
  timeout: 10
  headless: false

# 测试数据文件（test_data.csv）
test_case_id,username,password,expected_result
TC001,user001,password123,success
TC002,user001,wrongpass,error
TC003,nonexistent,password123,error
TC004,user002,password456,success

# 框架核心类
class DataDrivenTestFramework:
    def __init__(self, config_file, data_file):
        self.config = self.load_config(config_file)
        self.test_data = self.load_test_data(data_file)
        self.driver = None

    def load_config(self, config_file):
        """加载配置文件"""
        with open(config_file, 'r') as f:
            return yaml.safe_load(f)

    def load_test_data(self, data_file):
        """加载测试数据"""
        data = []
        with open(data_file, 'r') as f:
            reader = csv.DictReader(f)
            for row in reader:
                data.append(row)
        return data

    def setup(self):
        """初始化测试环境"""
        options = Options()
        if self.config['test_config']['headless']:
            options.add_argument('--headless')
        self.driver = webdriver.Chrome(options=options)
        self.driver.implicitly_wait(self.config['test_config']['timeout'])

    def teardown(self):
        """清理测试环境"""
        if self.driver:
            self.driver.quit()

    def run_tests(self):
        """执行所有测试"""
        results = []
        for test_data in self.test_data:
            result = self.execute_test(test_data)
            results.append(result)
        return results

    def execute_test(self, test_data):
        """执行单个测试"""
        try:
            self.driver.get(self.config['test_config']['base_url'])
            # 执行测试逻辑
            username = test_data['username']
            password = test_data['password']
            expected = test_data['expected_result']
            # 输入用户名和密码
            self.driver.find_element(By.ID, "username").send_keys(username)
            self.driver.find_element(By.ID, "password").send_keys(password)
            self.driver.find_element(By.ID, "login_btn").click()
            # 验证结果
            if expected == "success":
                success = self.driver.find_element(By.CLASS_NAME, "success-msg")
                return {"test_id": test_data['test_case_id'], "status": "PASS"}
            else:
                error = self.driver.find_element(By.CLASS_NAME, "error-msg")
                return {"test_id": test_data['test_case_id'], "status": "PASS"}
        except Exception as e:
            return {"test_id": test_data['test_case_id'], "status": "FAIL", "error": str(e)}

    def generate_report(self, results):
        """生成测试报告"""
        total = len(results)
        passed = len([r for r in results if r['status'] == 'PASS'])
        failed = total - passed
        report = f"""
测试执行报告
============
总用例数：{total}
通过：{passed}
失败：{failed}
通过率：{passed/total*100:.2f}%
详细结果：
        """
        for result in results:
            report += f"\n{result['test_id']}: {result['status']}"
        return report

# 使用示例
if __name__ == "__main__":
    framework = DataDrivenTestFramework('config.yaml', 'test_data.csv')
    framework.setup()
    results = framework.run_tests()
    report = framework.generate_report(results)
    print(report)
    framework.teardown()

[ERROR] 2026-05-06 14:30:45 - NullPointerException at LoginService.java:42
[STACK TRACE]
at com.example.LoginService.validateUser(LoginService.java:42)
at com.example.LoginController.login(LoginController.java:28)
at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
[CONTEXT]
user_input: "user001"
password_input: "password123"
database_status: "connected"
cache_status: "null"