Python-unittest测试框架

一、unittest是一个Python的单元测试框架，可以帮我们完成接口自动化测试、UI自动化测试

一个使用unittest框架的自动化测试用例代码如下：

test_baidu.py

# coding=utf-8
'''
Created on 2017-6-14
@author: 灵枢
Project:百度测试用例
'''
from selenium import webdriver
import unittest, time

class BaiduTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30) #隐性等待时间为30秒
self.base_url = "https://www.baidu.com"

def test_baidu_search(self):
u"""百度搜索"""
driver = self.driver
driver.get(self.base_url + "/")
driver.find_element_by_id("kw").clear()
driver.find_element_by_id("kw").send_keys("unittest")
driver.find_element_by_id("su").click()
time.sleep(3)
title=driver.title
self.assertEqual(title, u"unittest_百度搜索")

def test_baidu_set(self):
u"""百度设置"""
driver=self.driver
driver.get(self.base_url+"/gaoji/preferences.html")
m=driver.find_element_by_name("NR")
time.sleep(1)
m.find_element_by_xpath("//option[@value='50']").click()
time.sleep(1)

def tearDown(self):
self.driver.quit()

if __name__ == "__main__":
# unittest.main()
# 构建测试集
suite = unittest.TestSuite()
suite.addTest(BaiduTest("test_baidu_search"))
suite.addTest(BaiduTest("test_baidu_set"))
#执行测试
runner = unittest.TextTestRunner()
runner.run(suite)

二、代码解释

 首先，通过import导入unittest测试框架。创建BaiduTest 类继承unittest.TestCase 类，从 TestCase 类继承是告诉 unittest 这是一个测试用例。

 setUp ()方法用于设置初始化的部分，如初始化变量、打开浏览器等，在测试用例执行前，这个方法中的函数将先被调用。

 tearDown()方法用例测试用例执行之后的善后工作，如清除数据、关闭文件、关闭浏览器等。

 unittest要求测试用例必须以“test”开头。如test_baidu_search、test_baidu_set。

 接下来，调用unittest.TestSuite()类的addTest()方法向测试套件中添加测试用例

 最后，通过unittest.TextTestRunner()类的run()方法运行测试套件中的测试用例

 如果想直接运行当前测试文件中的所有测试用例，则可以直接使用unittest提供的main()方法。只需在if name == “main“:下面写unittest.main()就行了。

三、批量执行测试用例

这里还有一个搜狗搜索的测试用例，如下：

test_sogou.py

# coding=utf-8
'''
Created on 2017-6-14
@author: 灵枢
Project:搜狗搜索测试用例
'''
from selenium import webdriver
import unittest, time

class SogouTest(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30) #隐性等待时间为30秒
self.base_url = "http://www.sogou.com"

def test_sogou(self):
u"""搜狗搜索"""
driver = self.driver
driver.get(self.base_url + "/")
driver.find_element_by_id("query").clear()
driver.find_element_by_id("query").send_keys(u"python3")
driver.find_element_by_id("query").submit()
time.sleep(3)
title = driver.title
self.assertEqual(title, u"python2 - 搜狗搜索")

def tearDown(self):
self.driver.quit()

if __name__ == "__main__":
unittest.main()

如果想同时执行test_baidu.py和test_sogou.py，可以新建一个类用于批量执行测试用例：

all_search_tests.py

#coding=utf-8
'''
Created on 2017-6-14
@author: 灵枢
Project:批量执行测试用例
'''
import HTMLTestRunner
import unittest
import os,time

test_case_dir = "E:\\selenium_use_case\\test_case"  #test_baidu.py和test_sogou.py都放在该目录下
def get_test_suite():
testsunit=unittest.TestSuite()
#在listaa目录下查找以test开头的python文件
discover=unittest.defaultTestLoader.discover(test_case_dir,pattern='test_*.py',top_level_dir=None)
#循环添加测试用例
for test_suite in discover:
for test_case in test_suite:
testsunit.addTests(test_case)
print(testsunit)
return testsunit

if __name__ == '__main__':
runner = unittest.TextTestRunner()
runner.run(get_test_suite())