带unittest的脚本分析
#test.py #coding:utf-8 from selenium import webdriver from selenium.webdriver.common.keys import Keys from selenium.webdriver.common.by import By 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.Chrome() 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").clear() driver.find_element_by_id("kw").send_keys("selenium ide") driver.find_element_by_id("su").click() def is_element_present(self,how,what): try: self.driver.find_element(by=how,vaule=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()
代码分析如下:
Import unittest #首先引入unittest框架
Class BaiduTest(unittest.TestCase): BaiduTest类继承unitt框架的TestCase类成为标砖的测试类。
def setUp(self): self.driver=webdriver.Chrome() self.driver.implicitly_wait(30) self.base_url="http://www.baidu.com" self.verificationErrors=[] self.accept_next_alert=True
1.-setUp用于设置初始化工作,在执行每一个测试用例前先被执行,它与tearDown方法相呼应,后者在每一个测试用例执行后被执行。这里的初始化工作定义了浏览器启动和基础URL地址。
2.implicitly_wait()设置页面上元素的隐性等待时间为30秒。
3.定义空的verificationErrors数组,脚本运行时的错误信息将被记录到这个数组中。
4.定义accept_next_alert变量,表示是否继续接受下一个警告,初始状态为True。
def test_baidu(self): driver=self.driver driver.get(self.base_url +"/") driver.find_element_by_id("kw").clear() driver.find_element_by_id("kw").send_keys("selenium ide") driver.find_element_by_id("su").click()
这里就是测试脚本
def is_element_present(self,how,what): try: self.driver.find_element(by=how,vaule=what) except NoSuchElementException,e: return False return True
is_element_present方法用于查找页面元素是否存在,通过find_element()来接收元素的定位方法(how)和定位值(what),如果定位到元素则返回True,否则抛出异常并返回False。Try…except….为python语言的异常处理。
def is_alert_present(self): try: self.driver.switch_to_alert() except NoAlertPresentException,e: return False return True
is_alert_present()方法用于判断当前页面是否存在警告框,利用webDriver提供的switch_to_alert()方法来捕捉页面上警告框。如果捕捉到警告框则返回True,否则将抛出NoAlertPresentException类型异常,并返回False。
经过验证,不管页面是否出现警告框,返回结果都是True,所有可以调整该方法为driver.switch_to_alert().text,用于获取当前页面上的警告提示信息,可以获取到就返回True,获取不到则返回False。
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
close_alert_and_get_its_text()关闭警告并获得警告信息,首先通过switch_to_alert()获得警告,通过text获得警告框信息。接着通过if语句判断accept_next_alert的状态,在setUp()中已经初始状态为True,如果为True,则通过accept()接受警告,否则dissmiss()忽略此警告。
def tearDown(self): self.driver.quit() self.assertEqual([],self.verificationErrors)
tearDown()方法在每个测试方法执行后调用,这个方法用于测试用例执行后的清理工作,如退出浏览器、关闭驱动、恢复用例执行状态等。
在setUp()方法中定义的verificationErrors为空数组,这里通过assertEqual()比较其是否为空,如果为空则说明用例执行的过程中没有出现异常,否则将抛出AssertionError异常。
if __name__=="__main__": unittest.main()
通过unittest.main()方法来运行当前文件中的测试方法,其默认匹配并运行以test开头的方法。
原文地址:https://www.cnblogs.com/Rita-LJ/p/11792975.html