selenium 自动化 加入断言完整代码

以下是一个示例代码，展示如何使用Selenium进行自动化测试，并添加断言：

from selenium import webdriver
import unittest

class TestGoogleSearch(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.driver.implicitly_wait(10)

    def test_search(self):
        self.driver.get("https://www.google.com/")
        self.assertIn("Google", self.driver.title)

        search_box = self.driver.find_element_by_name("q")
        search_box.send_keys("Selenium")
        search_box.submit()

        results = self.driver.find_elements_by_xpath("//h3")
        self.assertTrue(len(results) > 0)

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

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

在这个示例中，我们使用了unittest模块来组织测试用例。在setUp()方法中，我们启动了Chrome浏览器，并设置了最大化窗口和隐式等待时间。在test_search()方法中，我们首先打开了Google搜索页面，并使用断言检查了页面标题是否包含了“Google”关键字。然后，我们在搜索框中输入了“Selenium”关键字，并提交了搜索。最后，我们使用XPath表达式找到了搜索结果中的所有标题，并使用断言检查了是否有结果。

在tearDown()方法中，我们关闭了浏览器。如果有多个测试用例，可以使用setUpClass()和tearDownClass()方法在测试类的开始和结束时分别执行一次。