unittest在Python單元測試中很常見,下面藉助IDE錄製的腳本並導出來分析它的框架。
在百度首頁輸入selenium,空格,刪除空格,然後回車搜索,錄製如下:
將錄製完的腳本導出,另存爲baidu_test.py
打開另存爲的腳本,其內容如下:
# -*- coding: utf-8 -*-
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
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re
class BaiduTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Firefox()#
self.driver.implicitly_wait(30)
self.base_url = "https://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").click()
driver.find_element_by_id("kw").clear()
driver.find_element_by_id("kw").send_keys("Selenium")
driver.find_element_by_id("su").click()
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()
1、setup()裏面大部分熟悉,不過有些還是第一次見:
self.verificationErrors = [] #腳本運行時,錯誤的信息將被打印到這個列表中。
self.accept_next_alert = True #是否繼續接受下一下警告
2、這個測試操作代碼,熟悉:
<span style="font-family: Arial, Helvetica, sans-serif;">def test_baidu(self):</span>
driver = self.driver
driver.get(self.base_url + "/")
driver.find_element_by_id("kw").click()
driver.find_element_by_id("kw").clear()
driver.find_element_by_id("kw").send_keys("Selenium")
driver.find_element_by_id("su").click()
#這個是判斷頁面是否有元素,作用不大
def is_element_present(self, how, what):
try: self.driver.find_element(by=how, value=what)
except NoSuchElementException, e: return False
return True
4、彈窗異常處理:
def is_alert_present(self):
try: self.driver.switch_to_alert()
except NoAlertPresentException, e: return False
return True
5、關閉警告和對得到的文本框的處理:
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
6、teardown()下的代碼:self.assertEqual([], self.verificationErrors) 是個難點,對前面verificationErrors方法獲得的列表進行比較;如查verificationErrors的列表不爲空,輸出列表中的報錯信息。這個東西,也可以將來被你自己更好的調用和使用,根據自己的需要寫入你希望的信息。
def tearDown(self):
self.driver.quit()
self.assertEqual([], self.verificationErrors)