快捷導(dǎo)航

Python 中的Selenium異常處理實(shí)例代碼

更新時間：2018年05月03日 16:36:01 作者：菜鳥可米

本文通過實(shí)例代碼給大家介紹了Python 中的Selenium異常處理的相關(guān)知識，非常不錯，具有參考借鑒價值，需要的朋友參考下吧

自動化測試執(zhí)行過程中，難免會有錯誤/異常出現(xiàn)，比如測試腳本沒有發(fā)現(xiàn)對應(yīng)元素，則會立刻拋出NoSuchElementException異常。這時不要怕，肯定是測試腳本或者測試環(huán)境哪里出錯了！那如何處理才是關(guān)鍵？因?yàn)橐话阒皇蔷植坑袉栴}，為了讓腳本繼續(xù)執(zhí)行，so我們可以用try...except...raise捕獲異常。該捕獲異常后可以打印出相應(yīng)的異常原因，這樣以便于分析異常原因。

　　下面將舉例說明，當(dāng)異常拋出后將信息打印在控制臺，同時截取當(dāng)前瀏覽器窗口，作為后續(xù)bug的依據(jù)給相應(yīng)開發(fā)人員更好下定位問題。代碼如下：

import unittest
from selenium import webdriver
from selenium.common.exceptions import NoSuchElementException  #導(dǎo)入NoSuchElementException
class ExceptionTest(unittest.TestCase):
  def setUp(self):
    self.driver = webdriver.Chrome()
    self.driver.get("https://www.baidu.com")
  def test_exception(self):
    driver = self.driver
    try:
      search_text = driver.find_element_by_id("ss")
      self.assertEqual('百度一下', search_text.get_attribute("value"))
    except NoSuchElementException:
      file_name = "no_such_element.png"
      #driver.save_screenshot(file_name)
      driver.get_screenshot_as_file(file_name) 
      raise  #拋出異常，注釋后則不拋出異常
  def tearDown(self):
    self.driver.quit()
if __name__ == '__main__':
  unittest.main(verbosity=2)

　　運(yùn)行有異常，結(jié)果如下：

　　上面代碼中用到WebDriver內(nèi)置的捕獲屏幕并保存的方法，如這里的save_screenshot(filename)方法和save_screenshot_as_file(filename)方法，在測試異常拋出時，同時截取瀏覽器屏幕并以自定義的圖片文件名保存在指定路徑（上面代碼為當(dāng)前路徑）。

　　又如當(dāng)一個元素呈現(xiàn)在DOM，但它是不可見的，不能與之進(jìn)行交互，異常將拋出，以百度首頁的登錄為例，當(dāng)元素不能不可見時，拋出ElementNotVisibleException的異常，代碼如下：

import unittest
from selenium import webdriver
from selenium.common.exceptions import ElementNotVisibleException  #導(dǎo)入ElementNotVisibleException
class ExceptionTest(unittest.TestCase):
  def setUp(self):
    self.driver = webdriver.Chrome()
    self.driver.get("https://www.baidu.com")
  def test_exception(self):
    driver = self.driver
    try:
      login = driver.find_element_by_name("tj_login")
      login.click()
    except ElementNotVisibleException:
      raise  
  def tearDown(self):
    self.driver.quit()
if __name__ == '__main__':
  unittest.main(verbosity=2)

　　運(yùn)行有異常，結(jié)果如下：