Python搭配unittest

unittest是Python的单元测试框架，

类似于Java里面的TestNG。

from time import sleep
from selenium import webdriver
import unittest


class Unittest(unittest.TestCase):
    # Unittest类继承unittest.TestCase类

    def setUp(self):
        # setUp用于设置初始化工作，在每一个测试用例前先被执行
        self.driver = webdriver.Firefox()
        self.base_url = "https://www.baidu.com/"
        self.driver.maximize_window()
        sleep(2)

    def test_baidu(self):
        # 搜索中国的测试用例
        driver = self.driver
        driver.get(self.base_url)
        driver.find_element_by_xpath(".//*[@id=‘kw‘]").send_keys("中国")
        driver.find_element_by_xpath(".//*[@id=‘su‘]").click()
        sleep(2)
        self.assertEqual(driver.find_element_by_xpath(".//*[@id=‘1‘]/h3/a").text, "中国_百度百科")
        # 断言中国_百度百科元素

    def tearDown(self):
        # tearDown方法在每个测试方法执行后调用，用于测试后的清理工作
        self.driver.quit()

    if __name__ == "__main__":
        unittest.main()
        # 整个测试过程集成在unitest.main()模块中，其默认执行以test开头的方法