學習unittest 很好的一個切入點就是從selenium IDE 錄制導出腳本。相信不少新手學習selenium 也是從IED 開始的。

IDE學習參考：

菜鳥學自動化測試（一）----selenium IDE

?

借助IED?錄制腳本

?

將腳本導出，保存為baidu.py?，通過python?IDLE編輯器打開。如下：

from selenium import webdriver from selenium.webdriver.common.by import By from selenium.webdriver.common.keys import Keys from selenium.webdriver.support.ui import Select from selenium.common.exceptions import NoSuchElementException import unittest, time, re class Baidu(unittest.TestCase): def setUp(self): self.driver = webdriver.Firefox() self.driver.implicitly_wait(30) self.base_url = "http://www.baidu.com/" self.verificationErrors = [] self.accept_next_alert = True def test_baidu(self): driver = self.driver driver.get(self.base_url + "/") driver.find_element_by_id("kw").send_keys("selenium webdriver") driver.find_element_by_id("su").click() driver.close() def is_element_present(self, how, what): try: self.driver.find_element(by=how, value=what) except NoSuchElementException, e: return False return True def is_alert_present(self): try: self.driver.switch_to_alert() except NoAlertPresentException, e: return False return True def close_alert_and_get_its_text(self): try: alert = self.driver.switch_to_alert() alert_text = alert.text if self.accept_next_alert: alert.accept() else: alert.dismiss() return alert_text finally: self.accept_next_alert = True def tearDown(self): self.driver.quit() self.assertEqual([], self.verificationErrors) if __name__ == "__main__": unittest.main()

加入unittest框架后，看上去比我們之前見的腳本復雜了很多，除了中間操作瀏覽器的幾行，其它都看不懂，不要急，我們來分析一下~！

?

?

框架分析 ? ? ? ? ??

?

?

import unittest

相想使用unittest框架，首先要引入unittest?包，這個不多解釋。

?

class Baidu(unittest.TestCase):

Baidu類繼承unittest.TestCase?類，從TestCase類繼承是告訴unittest模塊的方式，這是一個測試案例。

?

def setUp(self):self.driver = webdriver.Firefox()self.base_url = "http://www.baidu.com/"

setUp?用于設置初始化的部分，在測試用例執行前，這個方法中的函數將先被調用。這里將瀏覽器的調用和URL的訪問放到初始化部分。

?

self.verificationErrors = []

腳本運行時，錯誤的信息將被打印到這個列表中。

?

self.accept_next_alert = True

是否繼續接受下一下警告（字面意思，沒找到解釋！）

?

def test_baidu(self):driver = self.driverdriver.get(self.base_url + "/") driver.find_element_by_id("kw").send_keys("selenium webdriver") driver.find_element_by_id("su").click()

?test_baidu中放置的就是我們的測試腳本了，這部分我們并不陌生；因為我們執行的腳本就在這里。

?

?

def is_element_present(self, how, what):try: self.driver.find_element(by=how, value=what)except NoSuchElementException, e: return False return True

?is_element_present函數用來查找頁面元素是否存在，在這里用處不大，通常刪除。

因為判斷頁面元素是否存在一般都加在testcase中。

?

?

def is_alert_present(self):try: self.driver.switch_to_alert()except NoAlertPresentException, e: return False return True

對彈窗異常的處理

?

def close_alert_and_get_its_text(self):try:alert = self.driver.switch_to_alert() alert_text = alert.text if self.accept_next_alert: alert.accept() else: alert.dismiss() return alert_text finally: self.accept_next_alert = True

關閉警告和對得到文本框的處理，如果不熟悉python的異常處理和if?語句的話，請去補基礎知識，這里不多解釋。

?

def tearDown(self):self.driver.quit()self.assertEqual([], self.verificationErrors)

　　tearDown?方法在每個測試方法執行后調用，這個地方做所有清理工作，如退出瀏覽器等。?

　　self.assertEqual([],?self.verificationErrors)?是個難點，對前面verificationErrors方法獲得的列表進行比較；如查verificationErrors的列表不為空，

輸出列表中的報錯信息。

　　而且，這個東西，也可以將來被你自己更好的調用和使用，根據自己的需要寫入你希望的信息。（rabbit?告訴我的）

?

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

unitest.main()函數用來測試?類中以test開頭的測試用例

?

?

?

執行結果 ? ? ?

?

　　這樣一一分析下來，我們對unittest?框架有了初步的了解。運行腳本，因為引入了unittest?框架，所以控制臺輸出了腳本執行情況的信息。

?

>>> ========================= RESTART ================================ >>> . ---------------------------------------------------------------------- Ran 1 test in 10.656sOK >>>

?

很帥吧！？?后面將以unittest?為基礎，向新的征程進發~！

轉載于:https://www.cnblogs.com/Raul2018/p/9359105.html

總結

以上是生活随笔為你收集整理的selenium-webdriver(python) (十六) --unittest 框架的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。