Python3+Selenium2完整的自动化测试实现之旅（六）：Python单元测试模块Unittest运用
一、Unittest单元测试框架简介
Unitest是Python下的一个单元测试模块，是Python标准库模块之一，安装完Python后就可以直接import该模块，能在单元测试下编写具体的测试用例脚本，并调用模块封装好的方法，实现测试用例的执行、测试场景的恢复，甚至能批量采集测试用例脚本、批量运行测试脚本用例、控制执行顺序等，依托于Unittest模块，可以高效的组织测试用例编写、测试用例脚本的采集管理以及脚本运行的执行控制等。Unitest单元测试框架主要包含如下几个重要的逻辑单元：
1.测试固件（test fixture）
一个测试固件包括两部分，执行测试代码的前置条件和测试结束之后的场景恢复，这两部分一般用函数setUp()和tearDown()表示。简单说，就是平时手工测试一条具体的测试用例时，测试的前置环境和测试结束后的环境恢复。
2.测试用例（test case）
unittest中管理的最小单元是测试用例，就是一个测试用例，包括具体测试业务的函数或者方法，只是该test case 必须是以test开头的函数。unittest会自动化识别test开头的函数是测试代码，如果你写的函数不是test开头，unittest是不会执行这个函数里面的脚本，这个千万要记住，所有的测试函数都要test开头，记住是小写的哦。
3.测试套件 （test suite）
就是很多测试用例的集合，一个测试套件可以随意管理多个测试用例，该部分用来实现诸多测试用例的采集，形成一套测试用例集。
4.测试执行器 （test runner）
test runner是一个用来执行加载测试用例，并执行用例，且提供测试输出的一个逻辑单元。test runner可以加载test case或者test suite进行执行测试任务，并能控制用例集的执行顺序等。
从Unitest单元测试框架的基本逻辑单元设计来看，很明显可以看到它包含了用例的整个生命周期：用例的前置条件、用例的编写、用例的采集、用例的执行以及测试执行后的场景恢复。
二、首次使用Unittest模块
下面以打开百度，进行搜索，创建一个名为baidu_search.py的脚本文件，编写百度搜索分别python2和python3的测试用例，代码如下：
'''
Code description：
Create time：
Developer：
'''
# -*- coding: utf-8 -*-
import time
import unittest
from selenium import webdriver
class Search(unittest.TestCase):
def setUp(self):
"""
测试前置条件，这里要搜索的话就是先得打开百度网站啦
"""
self.driver = webdriver.Ie()
self.driver.maximize_window()
self.driver.implicitly_wait(5)
self.driver.get("https://www.baidu.com")
def tearDown():
"""
测试结束后环境复原，这里就是浏览器关闭退出
"""
self.driver.quit()
def test_search1(self):
"""
这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
"""
self.driver.find_element_by_id('kw').send_keys('python2')
self.driver.find_element_by_id('su').click()
time.sleep(1)
try:
assert 'python2' in self.driver.title
print('检索python2完成')
except Exception as e:
print('检索失败', format(e))
def test_search2(self):
"""
这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
"""
self.driver.find_element_by_id('kw').send_keys('python3')
self.driver.find_element_by_id('su').click()
time.sleep(1)
try:
assert 'python3' in self.driver.title
print('检索python3完成')
except Exception as e:
print('检索失败', format(e))
if __name__ == '__main__':
unittest.main()
在PyCharm中运行上面代码，我们会发现浏览器打开关闭了两次，分别检索了python2关闭浏览器，然后检索python3关闭浏览器。
这个效果显然不是我们希望的，我们希望检索完python2后，不用关闭浏览器继续检索python3，Unittest有相关的设置吗？答案是肯定的，我们对以上代码做下调整修改，注意对比不同的地方，然后运行就达到我们想要的效果
'''
Code description：
Create time：
Developer：
'''
# -*- coding: utf-8 -*-
import time
import unittest
from selenium import webdriver
class Search(unittest.TestCase):
@classmethod
def setUpClass(cls):
"""
测试前置条件，这里要搜索的话就是先得打开百度网站啦
"""
cls.driver = webdriver.Ie()
cls.driver.maximize_window()
cls.driver.implicitly_wait(5)
cls.driver.get("https://www.baidu.com")
@classmethod
def tearDownClass(cls):
"""
测试结束后环境复原，这里就是浏览器关闭退出
"""
cls.driver.quit()
def test_search1(self):
"""
这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
"""
self.driver.find_element_by_id('kw').send_keys('python2')
self.driver.find_element_by_id('su').click()
time.sleep(1)
try:
assert 'python2' in self.driver.title
print('检索python2完成')
except Exception as e:
print('检索失败', format(e))
def test_search2(self):
"""
这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
"""
self.driver.find_element_by_id('kw').clear() # 清空之前输入的python2
self.driver.find_element_by_id('kw').send_keys('python3')
self.driver.find_element_by_id('su').click()
time.sleep(1)
try:
assert 'python3' in self.driver.title
print('检索python3完成')
except Exception as e:
print('检索失败', format(e))
if __name__ == '__main__':
unittest.main()
三、Unittest模块批量加载和管理用例
以上对于Unittest框架，我们好像只用到了测试固件、测试用例两个逻辑单元的使用，接下来问题又来了：我们日常项目中的测试案例肯定不止一个，当案例越来越多时我们如何管理这些批量案例？如何保证案例不重复？如果案例非常多（成百上千，甚至更多）时如何保证案例执行的效率？
来看一下在unittest框架中如何管理批量用例：
手动添加采集指定的测试用例集用法：先在PyCharm中新建如下项目层级：
其中baidu_search1还是上面调整修改过的代码，然后编辑run_case.py文件，代码如下：
'''
Code description： 执行add 的测试用例集
Create time：
Developer：
'''
# -*- coding: utf-8 -*-
from testcase.sreach.baidu_sreach1 import Search # 将baidu_sreach.py模块导入进来
import unittest
suite = unittest.TestSuite() # 构造测试用例集
# suite.addTest(Search("test_search1"))
suite.addTest(Search("test_search2")) # 分别添加baidu_sreach1.py中的两个检索的测试用例
if __name__ == '__main__':
runner = unittest.TextTestRunner() # 实例化runner
runner.run(suite) #执行测试
这样运行run_case.py，就只执行了在百度中搜索python3这条用例，手动添加指定用例到测试套件的方法是addTest()，但是很多时候我们写了很多测试脚本文件，每个脚本中有多个test，，如果还是使用addTest()方法就不行了，我们希望能获取所有的测试集，并全部执行。然后编辑run_all_case.py文件，编写如下代码：
'''
Code description： TestLoader所有测试case
Create time：
Developer：
'''
# -*- coding: utf-8 -*-
import time
import os.path
import unittest
from selenium import webdriver
class Search(unittest.TestCase):
@classmethod
def setUpClass(cls):
"""
测试前置条件，这里要搜索的话就是先得打开百度网站啦
"""
cls.driver = webdriver.Ie()
cls.driver.maximize_window()
cls.driver.implicitly_wait(5)
cls.driver.get("https://www.baidu.com")
@classmethod
def tearDownClass(cls):
"""
测试结束后环境复原，这里就是浏览器关闭退出
"""
cls.driver.quit()
def test_search1(self):
"""
这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
"""
self.driver.find_element_by_id('kw').send_keys('python2')
self.driver.find_element_by_id('su').click()
time.sleep(1)
try:
assert 'python2' in self.driver.title
print('检索python2完成')
except Exception as e:
print('检索失败', format(e))
def test_search2(self):
"""
这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
"""
self.driver.find_element_by_id('kw').clear() # 清空之前输入的python2
self.driver.find_element_by_id('kw').send_keys('python3')
self.driver.find_element_by_id('su').click()
time.sleep(1)
try:
assert 'python3' in self.driver.title
print('检索python3完成')
except Exception as e:
print('检索失败', format(e))
def test_search3(self):
"""
这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
"""
self.driver.find_element_by_id('kw').clear() # 清空之前输入的python3
self.driver.find_element_by_id('kw').send_keys('hello world')
self.driver.find_element_by_id('su').click()
time.sleep(1)
try:
assert 'hello world' in self.driver.title
print('检索hello world完成')
except Exception as e:
print('检索失败', format(e))
case_path = os.path.join(os.getcwd()) # 在当前目录中采集测试用例
print(case_path)
all_case = unittest.defaultTestLoader.discover(case_path, pattern="test*.py", top_level_dir=None) # 采集所有test开头的测试用例
print(all_case)
if __name__ == '__main__':
runner = unittest.TextTestRunner() # 实例化runner
runner.run(all_case) # 执行测试
获取所有的测试用例集使用的是discover()方法，在PyCharm中运行该脚本就如下：
以上就完成了在Python Unittest单元测试框架下编写测试用例脚本，并使用其提供的多种方法来批量管理测试用例集并并执行。
