python selenium unittest_selenium-webdriver(python) (十六) --unittest 框架

学习unittest 很好的一个切入点就是从selenium IDE 录制导出脚本。相信不少新手学习selenium 也是从IED 开始的。

IDE学习参考：

借助IED录制脚本

将脚本导出，保存为baidu.py，通过python IDLE编辑器打开。如下：

from selenium importwebdriverfrom selenium.webdriver.common.by importByfrom selenium.webdriver.common.keys importKeysfrom selenium.webdriver.support.ui importSelectfrom selenium.common.exceptions importNoSuchElementExceptionimportunittest, time, reclassBaidu(unittest.TestCase):defsetUp(self):

self.driver=webdriver.Firefox()

self.driver.implicitly_wait(30)

self.base_url= "http://www.baidu.com/"self.verificationErrors=[]

self.accept_next_alert=Truedeftest_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()defis_element_present(self, how, what):try: self.driver.find_element(by=how, value=what)except NoSuchElementException, e: returnFalsereturnTruedefis_alert_present(self):try: self.driver.switch_to_alert()except NoAlertPresentException, e: returnFalsereturnTruedefclose_alert_and_get_its_text(self):try:

alert=self.driver.switch_to_alert()

alert_text=alert.textifself.accept_next_alert:

alert.accept()else:

alert.dismiss()returnalert_textfinally: self.accept_next_alert =TruedeftearDown(self):

self.driver.quit()

self.assertEqual([], self.verificationErrors)if __name__ == "__main__":

unittest.main()

加入unittest框架后，看上去比我们之前见的脚本复杂了很多，除了中间操作浏览器的几行，其它都看不懂，不要急，我们来分析一下~！

框架分析

import unittest

相想使用unittest框架，首先要引入unittest包，这个不多解释。

class Baidu(unittest.TestCase):

Baidu类继承unittest.TestCase 类，从TestCase类继承是告诉unittest模块的方式，这是一个测试案例。

defsetUp(self):

self.driver=webdriver.Firefox()

self.base_url= "http://www.baidu.com/"

setUp 用于设置初始化的部分，在测试用例执行前，这个方法中的函数将先被调用。这里将浏览器的调用和URL的访问放到初始化部分。

self.verificationErrors = []

脚本运行时，错误的信息将被打印到这个列表中。

self.accept_next_alert = True

是否继续接受下一下警告(字面意思，没找到解释！)

deftest_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()

test_baidu中放置的就是我们的测试脚本了，这部分我们并不陌生；因为我们执行的脚本就在这里。

defis_element_present(self, how, what):try: self.driver.find_element(by=how, value=what)except NoSuchElementException, e: returnFalsereturn True

is_element_present函数用来查找页面元素是否存在，在这里用处不大，通常删除。

因为判断页面元素是否存在一般都加在testcase中。

defis_alert_present(self):try: self.driver.switch_to_alert()except NoAlertPresentException, e: returnFalsereturn True

对弹窗异常的处理

defclose_alert_and_get_its_text(self):try:

alert=self.driver.switch_to_alert()

alert_text=alert.textifself.accept_next_alert:

alert.accept()else:

alert.dismiss()returnalert_textfinally: self.accept_next_alert = True

关闭警告和对得到文本框的处理，如果不熟悉python的异常处理和if语句的话，请去补基础知识，这里不多解释。

deftearDown(self):

self.driver.quit()

self.assertEqual([], self.verificationErrors)

tearDown 方法在每个测试方法执行后调用，这个地方做所有清理工作，如退出浏览器等。

self.assertEqual([], self.verificationErrors) 是个难点，对前面verificationErrors方法获得的列表进行比较；如查verificationErrors的列表不为空，

输出列表中的报错信息。

而且，这个东西，也可以将来被你自己更好的调用和使用，根据自己的需要写入你希望的信息。(rabbit 告诉我的)

if __name__ == "__main__":

unittest.main()

unitest.main()函数用来测试 类中以test开头的测试用例

执行结果

这样一一分析下来，我们对unittest 框架有了初步的了解。运行脚本，因为引入了unittest框架，所以控制台输出了脚本执行情况的信息。

>>> ========================= RESTART ================================

>>>.----------------------------------------------------------------------Ran1 test in 10.656s

OK>>>

很帅吧！？ 后面将以unittest为基础，向新的征程进发~！