截圖是測(cè)試時(shí)常用的一項(xiàng)手段,webdriver提供了截圖功能簡(jiǎn)單方便 下面的代碼演示了如何使用webdriver進(jìn)行截圖: #-*-coding=utf-8 from selenium import webdriver from selenium.common.exceptions import NoSuchElementException import unittest import time class login(unittest.TestCase): def setUp(self): self.driver = webdriver.Firefox() self.driver.implicitly_wait(10) self.base_url = "http://www.baidu.com" self.verificationErrors = [] def test_loginpass(self): driver = self.driver driver.get(self.base_url) driver.find_element_by_link_text("搜索設(shè)置").click() driver.find_element_by_id("s1_2").click() m=driver.find_element_by_name("NR") try: m.find_element_by_xpath("http://option[@value='1000']").click() except: driver.get_screenshot_as_file("D:/screenshots/dd.png")#如果沒有找到上面的元素就截取當(dāng)前頁(yè)面。 time.sleep(5) driver.find_element_by_xpath("http://input[@value='保存設(shè)置']").click() driver.switch_to_alert().accept() def tearDown(self): #self.driver.quit() self.assertEqual([], self.verificationErrors) if __name__ == "__main__": unittest.main() python webdriver 截圖函數(shù)為get_screenshot_as_file(filename) 當(dāng)然我們也可能靈活使用,事先定義好文件名和路徑
|
|