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

更新時(shí)間：2018年05月03日 16:36:01 作者：菜鳥(niǎo)可米

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

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

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

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)方法，在測(cè)試異常拋出時(shí)，同時(shí)截取瀏覽器屏幕并以自定義的圖片文件名保存在指定路徑（上面代碼為當(dāng)前路徑）。

　　又如當(dāng)一個(gè)元素呈現(xiàn)在DOM，但它是不可見(jiàn)的，不能與之進(jìn)行交互，異常將拋出，以百度首頁(yè)的登錄為例，當(dāng)元素不能不可見(jiàn)時(shí)，拋出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é)果如下：