【如何学习Python自动化测试】—— Python 的 unittest 框架-编程知识

10 、Python 的 unittest 框架

10.1 Unittest 框架介绍

   Unittest是Python语言中的一种测试框架，是Python标准库中的一个模块。它可以帮助开发者编写自动化测试，可以进行单元测试、集成测试、功能测试等各种类型的测试。 Unittest的特点是简单易学，文档齐全，可以为每个测试用例创建独立的环境，可以根据测试结果输出详细的测试报告。
   Unittest模块中提供了很多方便测试的工具，如assertEqual、assertTrue、assertFalse、assertRaises等。使用Unittest测试框架可以提高开发的质量和效率，有效减少开发期和维护期中出现的问题，对于大型项目特别是团队合作开发中，使用Unittest测试框架可以提高代码的可维护性，降低开发和维护的成本。
   Unittest 是 python 的单元测试框架，原名为 PyUnit，由 java 的 junit 演化而来。
   Unittest 提供了 test case、test suites、test fixtures、test runnet。

Test case（测试用例）：是一个单独的测试单元，用于测试一个或多个特定的代码路径。
Test suite（测试套件）：是一系列测试用例的集合。测试套件是按照测试计划所定义的各个阶段的测试目标决定的，即先有测试计划，后面才有测试套件。
Test fixture（测试固件）：是为测试提供必要的环境和资源（如数据库连接）的代码。在测试之前和之后执行。（setup + test case + teardown 的组合）
Test runner（测试运行器）：是执行多个测试用例或测试套件的程序组件。它收集测试结果并生成测试报告。

在 Python 中，unittest 是一个内置的测试框架，可用于编写单元测试。它提供了类和方法来组织和运行测试用例。使用 unittest，我们可以轻松地编写和运行单元测试并生成详细的测试报告。

Unittest 结构：
在这里插入图片描述

10.2 Unittest 框架详解

10.2.1 测试用例

在 unittest 中没有明确究竟 test 的 class 是测试用例，还是 class 中的 test 开头的方法是测试用例，在下面的案例中，我们将 test 开头的方法作为测试用例来执行。
先看一个简单的案例：

import unittest
在使用 unittest 框架时，需要先导入 unittestclass login_test(unittest.TestCase):
创建一个测试类，该类继承 unittest 的 TestCase。 def setUp(self):Setup 是测试执行之前的操作， 会在用例执行之前先执行这里的内容，通常用来初始化环境self.testa=10self.testb=10def test_login(self):要执行的测试 1self.assertTrue(self.testa == self.testb)def test_login2(self):要执行的测试 2try:self.testc=30self.assertTrue(self.testa == self.testc)except AssertionError:self.fail("test is false")def tearDown(self):执行结束之后的操作， 通常用来执行还原测试环境的操作print('Test Runner End')if __name__ == '__main__':unittest.main()调用 unittest 的开始执行

   在使用 unittest 前，需要先导入 unittest 的包，在这个包中包含了 unittest 提供的各种类，如：TestCase、TestSuite、TextTestRunner、TestLoader、main 等等。
   创建一个 login_test 的 class，用来测试，该类继承 TestCase，从而能够使用 TestCase 中的方法。
   Setup 方法是 TestCase 中的方法，这里是重写 TestCase 的 setup 方法，该方法用来做测试执行前的操作，比如构建测试数据，打开浏览器、链接数据库等操作。
   下面的 tearDown 方法相同，区别在不该方法用于执行测试执行结束后的操作，比如关闭浏览器，释放数据库连接操作。
   中间以 test 开头的方法，则会被当做测试用例来执行，运行该脚本时，unittest 会自动加载这些 test 开头的方法，来执行测试。
   Unitest.main 是调用 unittest 的执行方法，这里是测试执行的入口。

10.2.2 测试套

测试套又称测试集，也叫测试套件，通常是一组测试用例的集合，通过 suite 可以执行一组测试用例。下面的代码是针对 LMD 的登陆测试，如果登陆成功，则获取他的 title，判断 title 是否不预期结果一致，如果一致，则测试成功，否则用例执行失败。这个案例通过 suite 实现的。代码如下：

#coding=utf-8
__author__ = 'Administrator'
import unittest
from selenium import webdriver
import HTMLTestRunner
from selenium.webdriver.support import expected_conditions
class login_test2(unittest.TestCase) :def setUp(self):self.driver = webdriver.Firefox()self.driver.maximize_window()def test_login(self):driver = self.driverdriver.get('http://www.chuangyijia.com/login')driver.find_element_by_id('email').send_keys('810155067@qq.com')driver.find_element_by_id('pwd').send_keys('a654321')driver.find_element_by_css_selector('#submit').click()test=expected_conditions.title_is(u'创意家，登录') self.assertTrue(test(driver))def tearDown(self):self.driver.quit()if __name__ == '__main__':suite=unittest.TestLoader().loadTestsFromTestCase(login_test2)unittest.TextTestRunner().run(suite)

这段代码中，上面的内容和之前没有上面差别，主要在 if name==’ main__’的内容，这句话的意思为“Make a script both importable and executable”，解释过来就是让你写的脚本模块既可以导入到别的模块中用，另外该模块自己也可执行。试代码的时候，在”if name == ‘main’ “中加入一些我们的调试代码，我们可以让外部模块调用的时候不执行我们的调试代码，但是如果我们想排查问题的时候，直接执行该模块文件，调试代码能够正常运行！

Unittest.TestLoader：源代码解释为“This class is responsible for loading tests according to various criteria and returning them wrapped in a TestSuite ”意思可以通过该类完成测试用例的加载，将用例加载到 siuter 中，但是需要调用该类的 loadTestFromTestCase 方法来加载用例，该方法接收参数为测试类，也就是我们上面定义的 login_test2 的类名。这句代码就会将该类中以 test 开头的方法加载到 suiter 中。 Suiter 变量用来接收加载后的结果。
通过 unittest 调用他的 TextTestRunner 类中的 run 方法来执行测试，run 方法需要将 suiter 作为参数传入。TextTestRunner 是以文本形式显示测试结果，如果测试失败，会显示失败的测试名称，并统计测试的结果。run 方法是运行传入的 case 或 suiter。

10.2.3 断言

在编程中，通常会做一些假设，使用断言，去判断一个代码处理的结果是否和自己的预期相符，如果符合则为 True，否则为 false。在测试中，我们也会有这样的需求，执行一个测试，给一个预期结果，来断言实际结果不预期结果是否符合，不符合则测试失败，符合则执行成功。
在 unittest 的 TestCase 中给我们提供了很丰富的断言操作，下面罗列出了常用的一些断言。

assertTrue：检查表达式返回值是否为真 True ，返回值类型为布尔类型。
assertFalse：不上面相反，检查表达式返回是否为假 False ，布尔类型。
assertEqual(arg1, arg2, msg=None)：验证 arg1 和 arg2 是否相等。
assertNotEqual(arg1, arg2, msg=None)：验证 arg1 和 arg2 是否不相等。
assertIs(arg1, arg2, msg=None)：验证 arg1 和 arg2 是否为同一个对象。
assertIsNot(arg1, arg2, msg=None)：验证 arg1 和 arg2 是否不为同一个对象。
assertIsNone(expr, msg=None)：验证 expr 是否为 none，是则返回 True。
ssertIsNotNone(expr, msg=None)：验证 expr 是否不为 none，是则返回 True。
assertIn(arg1, arg2, msg=None)：验证 arg1 是否在 arg2 中，是则 True
assertNotIn(arg1, arg2, msg=None)：验证 arg1 不在 arg2 的中，是则 fail

示例：

#coding=utf-8
__author__ = 'Administrator'
import unittest
class Test_Asser(unittest.TestCase):def test_assertTrue(self):self.assertTrue(5>4)5>4 则返回 True，否则返回 Falsedef test_assertFalse(self) :self.assertFalse(5<4)表达式成立则返回 False，否则返回 Truedef test_assertEqual(self) :self.assertEqual(u"中国",u"中国")两个字符串相等，则返回 True，否则返回 Falsedef test_assertNotEqual(self):self.assertNotEqul(u"中国 1",u"中国 2")两个字符串不相等，返回 True，否则返回 falsedef test_assertIs(self):a=10a=20self.assertIs(a,a)a 是一个变量，下面的是对 a 重新赋值，所以依然是一个对象， 所以此处返回 True，否则返回 false。def test_assertIsNot(self) :a=10b=20self.assertIsNot(a,b)a 和 b 是两个变量，不属于同一个对象 ，所以此处返回 True， 否则返回 Falsedef test_assertIsNone(self):a=Noneself.assertIsNone(a)None 在 python 中是一个特殊的空值，即表示空值，也可以表 示为空对象。在这里给 a 赋值为 None，那么 a 就是一个空对象，此处 返回 True，否则返回 falsedef test_assertIsNotNone(self):c=10self.assertIsNotNone(c)与上面相反， c 是一个有值的对象，这里判断 c 是不是不为 None，是则返回 True，否则返回 False。def test_assertIsIn(self):text="abcdef"t='c'self.assertIn(t,text)判断 t 的值是否在 text 中，如果在则返回 True，否则返回 False。def test_assertIsNotIn(self):text="abcdef"self.assertNotIn('h',text)判断字符 h 是否不在 text 中，不在则返回 True，否则返回 false.if __name__ == '__main__':unittest.main()

当然在 unittest 中不叧是这些断言，还有其他断言，后续会慢慢加入。

10.3 测试批量执行

下面做两个测试的案例，先创建目录结构：
   主目录：Lmd_auto_test
   次目录：Lmd_auto_test\test_case
   在 test_case 目录中有 login_auto.py 和 Release_Creative.py 两个脚本。
   将下面的代码保存为 login_auto.py

#coding=utf-8
__author__ = 'Administrator'
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions
import time
import unittestclass login_test_case(unittest.TestCase):def setUp(self):self.driver = webdriver.Firefox()self.driver.maximize_window()self.driver.get('http://www.chuangyijia.com/login')def tearDown(self):self.driver.quit()def test_login(self):self.driver.find_element_by_id('email').send_keys('810155067@qq.com')self.driver.find_element_by_id('pwd').send_keys('a654321')self.driver.find_element_by_id('submit').click()#self.driver.implicitly wait(5)time.sleep(3)WebDriverWait(self.driver,30).until(expected_conditions.visibility_of_element_located((By.CSS_SELECTOR,'.logo')))print self.driver.titleis_title = expected_conditions.title_is(u'首页-创意家') 	self.assertTrue(is_title(self.driver))if __name__ == '__main__':unittest.main()

下面的文件保存为 Release_Creative.py

#coding=utf-8
__author__ = 'Administrator'
from selenium import webdriver
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.common.action_chains import ActionChains
import unittest
import sys
import os,time
class release_creat(unittest.TestCase) :
def setUp(self):self.driver = webdriver.Firefox()self.driver.maximize_window()self.driver.get('http://www.chuangyijia.com/login')self.driver.find_element_by_id('email').send_keys('810155067@qq.com')self.driver.find_element_by_id('pwd').send_keys('a654321')self.driver.find_element_by_id('submit').click()time.sleep(3)#is title = expected conditions.title is(u'首页-创意家') #self.assertTrue(is title(self.driver))def test_release(self):self.driver.find_element_by_css_selector('.menu >ul:nth-child(1) > li:nth-child(1) > a:nth-child(1)').click()self.driver.find_element_by_css_selector('.c01 > a:nth-child(1)').click()self.driver.find_element_by_name('mobile').send_keys('1943235923')self.driver.find_element_by_css_selector('.idea_name').send_k eys(u"发布中文创意")self.driver.find_element_by_id('province').click()self.driver.find_element_by_xpath('//option[contains(text()," 江苏")]').click()self.driver.find_element_by_id('city').click()self.driver.find_element_by_xpath('//option[contains(text()," 南京")]').click()text=u"预售管理，后台管理员可以修改时间， 并且只能修改为 当月的日期， 不能修改年月"self.driver.find_element_by_css_selector('#pForm >table:nth-child(11) > tbody:nth-child(1) > tr:nth-child(1) > td:nth-child(1) > table:nth-child(1) > tbody:nth-child(1) >  tr:nth-child(6) > td:nth-child(2) > textarea:nth-child(1)').send_keys(text)self.driver.find_element_by_css_selector('#pForm >table:nth-child(11) > tbody:nth-child(1) > tr:nth-child(1) > td:nth-child(1) > table:nth-child(1) > tbody:nth-child(1) >  tr:nth-child(7) > td:nth-child(2) > textarea:nth-child(1)').send_keys(u"没有问题。。。。 ")self.driver.find_element_by_css_selector('#pForm >table:nth-child(11) > tbody:nth-child(1) > tr:nth-child(1) > td:nth-child(1) > table:nth-child(1) > tbody:nth-child(1) >  tr:nth-child(8) > td:nth-child(2) > textarea:nth-child(1)').send_keys(u"不需要解决")self.driver.find_element_by_name('support_end_date').send_keys('2017-8-18')time.sleep(2)self.driver.find_element_by_css_selector('.c_btn').click()self.driver.find_element_by_css_selector('.c_btn').click()os.system("H:\\pydj\\Lmd_auto_test\\testjpg.exe")time.sleep(2)put2 =self.driver.find_element_by_css_selector('#upimg_1')put1 =self.driver.find_element_by_css_selector('.cy_pic > ul:nth-child(1) > li:nth-child(1) > input:nth-child(3)')ActionChains(self.driver).move_to_element(put2).perform()self.driver.find_element_by_css_selector('.cy_pic >ul:nth-child(1) > li:nth-child(1) > input:nth-child(3)').click()os.system("H:\\pydj\\Lmd_auto_test\\testjpg.exe")#self.driver.find element by_css selector('.cy_pic >ul:nth-child(1) > li:nth-child(2) > input:nth-child(3)').click()#os.system("H:\pydj\Lmd auto test\testjpg.exe")self.driver.find_element_by_id('saveBtn').click()time.sleep(2)is_exist = self.driver.find_element_by_css_selector('div.item:nth-child(1) > div:nth-child(2) > a:nth-child(1)')Creat_Name = is_exist.textself.assertEqual(Creat_Name,u"发布中文创意")#self.assertTrue(is exist)if __name__ == '__main__':unittest.main()

上面的两个文件，一个是测试登陆功能的用例，另一个是测试创意发布的用例，这里对成功和失败都做了简单的判断，例如登陆，在判断登陆后，title 是否为登陆成功的title，是则执行通过，否则执行失败。创意发布时，判断提交后，该创意是否在发布的列表中，并在第一个，如果是，则执行成功，否则执行失败。
在 test_case 的目录中创建__init__.py 文件，文件内容如下：

#coding=utf-8
__author__ = 'Administrator'
import login_auto
import Release_Creative

在 Lmd_auto_test 目录下创建 run_test_case.py 文件，文件主要用来实现执行 test_case 目录下的测试脚本。文件内容如下：

#coding=utf-8
__author__ = 'Administrator'
# import test case
import unittest
from test_case import login_auto,Release_Creative
suiter = unittest.TestSuite()
suiter.addTest (unittest.makeSuite(login_auto.login_test_case)) 
suiter.addTest (unittest.makeSuite(Release_Creative.release_creat))
unittest.TextTestRunner().run(suiter)