在使用 pytest 和 Selenium 进行自动化测试时，通常我们会根据断言的结果来决定测试流程的走向。如果断言失败，测试通常会停止执行后续的步骤，因为失败意味着被测系统没有按照预期工作。然而，有时候我们可能需要在断言失败后执行特定的操作，比如记录错误信息或截图以便调试。

下面是一个示例，展示了如何在断言失败后继续执行下一步操作：

import pytest
from selenium import webdriver
from selenium.webdriver.common.by import Byclass TestWebPage:def setup_method(self):self.driver = webdriver.Chrome()def teardown_method(self):self.driver.quit()def test_button_click(self):self.driver.get('http://example.com')try:# 尝试查找并点击按钮button = self.driver.find_element(By.ID, 'myButton')button.click()# 检查页面上的某个元素是否出现success_message = self.driver.find_element(By.CLASS_NAME, 'success-message').textassert 'Success' in success_message, "Success message not found"except AssertionError as e:# 断言失败时的操作print("Assertion failed:", str(e))# 可以在这里添加更多操作，如截图、记录日志等self.driver.save_screenshot('failed_assertion.png')finally:# 无论断言是否失败都会执行的操作print("Test execution completed.")if __name__ == "__main__":pytest.main(['-v', 'test_example.py'])

在这个例子中，我们使用了 try-except-finally 结构来处理断言失败的情况。在 try 块中，我们执行了点击按钮和断言检查的代码。如果断言失败（即 assert 语句引发了一个 AssertionError），except 块会被执行，我们在这里打印了错误信息，并保存了一个截图。finally 块中的代码无论断言是否失败都会被执行，这里我们简单地打印了一条消息表明测试执行完成。

请注意，虽然在断言失败后继续执行代码是有用的，但在实际的测试脚本中，我们应该确保这种“失败后的操作”不会影响测试的正确性和完整性。通常，这些操作应该是辅助性的，比如记录信息，而不是改变应用程序的状态或测试的结果。

…………………………

当然，让我们来看一个更具体的例子，展示如何在断言失败后执行下一步操作。在这个例子中，我们将使用 pytest 和 Selenium 来编写一个测试脚本，该脚本会测试一个表单提交功能。如果表单提交成功，我们期望看到一个确认消息。如果提交失败，我们将在页面上查找错误消息，并根据情况执行不同的操作。

首先，确保你已经安装了必要的库：

pip install pytest selenium

然后，创建一个名为 test_form_submission.py 的文件，内容如下：

import pytest
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 ECclass TestFormSubmission:def setup_method(self):self.driver = webdriver.Chrome()def teardown_method(self):self.driver.quit()def test_submit_form(self):self.driver.get('http://example.com/form')# 填写表单input_field = self.driver.find_element(By.NAME, 'inputField')input_field.send_keys('Test Input')# 提交表单submit_button = self.driver.find_element(By.NAME, 'submitButton')submit_button.click()# 等待确认或错误消息出现wait = WebDriverWait(self.driver, 10)try:confirmation_message = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, '.confirmation-message')))assert 'Success' in confirmation_message.text, "Form submission was not successful"print("Form submitted successfully.")except AssertionError:# 断言失败，查找错误消息error_message = self.driver.find_element(By.CSS_SELECTOR, '.error-message')print("Form submission failed with error:", error_message.text)# 执行下一步操作，例如截图或记录日志self.driver.save_screenshot('form_submission_error.png')# 发送邮件通知错误（伪代码）self.send_email_notification(error_message.text)except Exception as e:print("An unexpected error occurred:", str(e))def send_email_notification(self, error_message):# 这里可以实现发送邮件的功能passif __name__ == "__main__":pytest.main(['-v', 'test_form_submission.py'])

在这个例子中，我们首先填写了一个表单并提交。然后，我们使用 WebDriverWait 等待确认消息的出现。如果确认消息中包含预期的文本 'Success'，我们就认为表单提交成功。如果断言失败，我们会查找错误消息，并执行一些后续操作，比如保存截图和发送邮件通知。

请注意，这个例子中的 http://example.com/form 和表单字段的名称都是假设的，你需要替换为你实际要测试的网页和元素的属性。此外，发送邮件的函数 send_email_notification 是一个占位符，你需要根据实际情况实现这个功能。

要运行这个测试，你可以直接运行 test_form_submission.py 文件，或者在命令行中输入 pytest -v test_form_submission.py。

……..........