Python搭配unittest

標籤：firefox   unittest   selenium   工作   測試案例   xpath   pytho   elf   sele   

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開頭的方法

Python搭配unittest