本篇转自博客:上海-悠悠
原文地址:http://www.cnblogs.com/yoyoketang/tag/unittest/
from selenium import webdriverfrom selenium.webdriver.common.by import Byfrom selenium.webdriver.common.keys import Keysfrom selenium.webdriver.support.ui import Selectfrom selenium.common.exceptions import NoSuchElementExceptionfrom selenium.common.exceptions import NoAlertPresentExceptionimport unittest, time, reclass Baidu(unittest.TestCase): def setUp(self): self.driver = webdriver.Firefox() 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").send_keys("selenium webdriver") driver.find_element_by_id("su").click() driver.close() 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() 啪啪啪上来就是一串代码,是不是有点懵!,下面小编来解释一下:class Baidu(unittest.TestCase):Baidu 类继承 unittest.TestCase 类,从 TestCase 类继承是告诉 unittest 模块的方式,这是一个测试案例。 def setUp(self):self.driver = webdriver.Firefox()self.base_url = "http://www.baidu.com/"setUp 用于设置初始化的部分,在测试用例执行前,这个方法中的函数将先被调用。这里将浏览器的调用和 URL 的访问放到初始化部分。 self.verificationErrors = []脚本运行时,错误的信息将被打印到这个列表中 self.accept_next_alert = True是否继续接受下一个警告。 def test_baidu(self):driver = self.driverdriver.get(self.base_url + "/")driver.find_element_by_id("kw" ).send_keys( "selenium webdriver")driver.find_element_by_id("su" ).click()test_baidu 中放置的就是我们的测试脚本了,这部分我们并不陌生;因为我们执行的脚本就在这里 def is_element_present(self, how, what):try : self.driver.find_element(by=how, value= what)except NoSuchElementException, e: return Falsereturn Trueis_element_present 函数用来查找页面元素是否存在,try...except....为 python 语言的异常捕捉。is_element_present 函数在这里用处不大,通常删除,因为判断页面元素是否存在一般都加在 testcase 中。 def is_alert_present(self):try: self.driver.switch_to_alert()except NoAlertPresentException, e: return Falsereturn True对弹窗异常的处理 def close_alert_and_get_its_text(self):try:alert = self.driver.switch_to_alert()alert_text = alert.textif self.accept_next_alert:alert.accept()else:alert.dismiss()return alert_textfinally : self.accept_next_alert = True关闭警告以及对得到文本框的处理,if 判断语句前面已经多次使用,并不陌生;try....finally...为 python的异常处理。 def tearDown(self):self.driver.quit()self.assertEqual([], self.verificationErrors)tearDown 方法在每个测试方法执行后调用,这个地方做所有测试用例执行完成的清理工作,如退出浏览器等。 self.assertEqual([], self.verificationErrors)这个是难点,对前面 verificationErrors 方法获得的列表进行比较;如查 verificationErrors 的列表不为空,输出列表中的报错信息 if __name__ == "__main__":unittest.main()unitest.main()函数用来测试 类中以 test 开头的测试用例
这样一一分析下来,我们对unittest 框架有了初步的了解。运行脚本,因为引入了unittest 框架,
所以控制台输出了用例的执行个数、时间以及是否ok 等信息。
本篇只是初步对 unittest 的框架进行了分析, 还有许细节将放在后面章节进行讨论, 例如 python 的
异常处理机制等。
时间: 2024-11-05 23:36:56