1. 全栈程序员必看首页

unittest 测试框架

全栈程序员-站长 • • 未分类 • 阅读 5

unittest 测试框架一、unittest测试框架1、测试固件1.1setUp（）1.2tearDown（）2、unittest基本使用3、测试套件（suit）3.1测试用例执行顺序3.2忽略测试用例的执行4、断言5、HTML报告生成5.1测试报告5.2异常捕捉与错误截图5.2.1错误截图：get_scunittest是python的单元测试框架，unittest单元测试提供了创建测试用例，测试套件以及批量执行的方案。作为单元测试的框架，unittest也是可以对程序最小

大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE稳定放心使用

文章目录

一、unittest测试框架

  • unittest 是python 的单元测试框架， unittest 单元测试提供了创建测试用例，测试套件以及批量执行的方案。作为单元测试的框架， unittest 也是可以对程序最小模块的一种敏捷化的测试。
  • unittest 和 Junit 都是单元测试？区别在于： unittest 是基于功能测试的单元测试，是基于 UI 界面的功能性测试，而 Junit 是白盒单元测试框架。
  • 解决了单个脚本重复的操作（导包，获取浏览器驱动，关闭浏览器）。

1、测试固件

1.1 setUp（）

  • 在执行测试用例脚本之前进行初始化环境的方法。

1.2 tearDown（）

  • 测试用例执行后，进行清理环境工作。

2、unittest 基本使用

  • 测试用例的命名： test_ 。
from selenium import webdriver
import time
import unittest

class TestUnit1(unittest.TestCase):
    # 获取浏览器的驱动
    def setUp(self):
        # 1、self 就是类的引用/实例
        # 2、全局变量的定义：self.变量名
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.url = "https://www.baidu.com/"
        self.driver.get(self.url)
        time.sleep(3)

        # 在百度中搜索信息
        # 测试用例的命名： test_
    def test_search1(self):
        self.driver.find_element_by_id("kw").send_keys("顾一野")
        self.driver.find_element_by_id("su").click()
        time.sleep(6)

    def test_search2(self):
        self.driver.find_element_by_id("kw").send_keys("account")
        self.driver.find_element_by_id("su").click()
        time.sleep(6)

    # 关闭浏览器
    def tearDowm(self):
        self.driver.quit()

    # 一个入口
    if __name__ == "__main__":
        unittest.main()

3、测试套件（suit）

  • 把不同文件下的测试用例组织起来形成的测试用例组。
  • 把需要一次性执行的测试用例，放在一个套件中，就可以一次性运行。
import unittest
from Test import testbaidu1
from Test import testbaidu2


def creatSuit():
    # 1、要把不同测试脚本的类中的需要执行的方法放在一个测试套件中
    # suit = unittest.TestSuite()
    # suit.addTest(testbaidu1.Baidu1("test_hao"))
    # suit.addTest(testbaidu2.Baidu2("test_hao"))
    # suit.addTest(testbaidu2.Baidu2("test_baidusearch"))
    # return suit

    # 2、如果需要把一个测试脚本中年所有的测试用例都添加到suit中，怎么做？
    # 第一种※：makeSuit

    # suit = unittest.TestSuite()
    # suit.addTest(unittest.makeSuite(testbaidu1.Baidu1))
    # suit.addTest(unittest.makeSuite(testbaidu2.Baidu2))
    # return suit

    # 第二种：TestLoader

    # suit1 = unittest.TestLoader().loadTestsFromTestCase(testbaidu1.Baidu1)
    # suit2 = unittest.TestLoader().loadTestsFromTestCase(testbaidu2.Baidu2)
    # suit = unittest.TestSuite([suit1, suit2])
    # return suit

    # 3、可以把一个文件夹下面所有的测试脚本中的测试用例放入测试套件
    discover = unittest.defaultTestLoader.discover("../Test", pattern="testbaidu*.py", top_level_dir=None)
    return discover


if __name__ == "__main__":
    suit = creatSuit()
    # verbersity= 0 测试用例成功多少，失败多少, 1 会标注哪个成功/失败, 2 会标注
    runner = unittest.TextTestRunner()
    runner.run(suit)

在这里插入图片描述

3.1 测试用例执行顺序

  • 按照test_后面的名称排序，0 ~ 9， A ~ Z，a ~ z。如果第一个字母相同，就看第二个字母，依次执行。

3.2 忽略测试用例的执行

@unittest.skip("skipping")

4、断言

  • 在自动化脚本中来判断实际结果和预期结果是否一致情况。
  • 就百度为例，如何判断查询成功？网页标题是搜索信息，或者页面中出现某个元素。
  • assertEqual（预期结果，实际结果，msg = “实际结果于预期结果不一致时输出的内容”）。

在这里插入图片描述

    def test_hbaidu(self):
        driver = self.driver
        url = self.url
        driver.get(url)
        # self.assertEqual("突如其来的假期_百度搜索", driver.title, msg="实际结果和预期结果不一致" )
        self.assertTrue("百度一下，你就知道" == driver.title, msg="不一致！！！")
        driver.find_element_by_id("kw").send_keys("突如其来的假期")
        driver.find_element_by_id("su").submit()
        time.sleep(5)
        print(driver.title)
        # self.assertEqual(driver.title, "突如其来的假期_百度搜索", msg="实际结果和预期结果不一致" )
        time.sleep(6)

5、HTML 报告生成

  • 在测试用例较多时，就需要统计测试用例的执行结果。
  • HTML 报告和错误截图都是用来分析测试用例批量执行后的结果的，错误截图可以帮助我们更清晰的看到错误具体出现的页面以及错误状况。

5.1 测试报告

import HTMLTestRunner
import os
import sys
import time
import unittest

def createsuite():
    discovers = unittest.defaultTestLoader.discover("../Test", pattern="testbaidu*.py", top_level_dir=None)
    print(discovers)
    return discovers

if __name__=="__main__":
    # 文件夹要创建地址，脚本所在的目录，
    curpath = sys.path[0]
    # c 盘下所有的路径都被打开，相当于一个数组
    print(sys.path)
    # 只是需要第一个
    print(sys.path[0])
    # 1，创建文件夹，存放 HTML 报告
    if not os.path.exists(curpath+'/resultreport'):
        os.makedirs(curpath+'/resultreport')

    # 2，文件夹的命名：时间 时分秒，名称绝对不会重复 time.time() 获取时间戳，local转换为当地时间。
    # time.strftime() 把本地时间以某一种特定格式展示。
    now = time.strftime("%Y-%m-%d-%H %M %S", time.localtime(time.time()))
    print(now)
    print(time.time())
    print(time.localtime(time.time()))

    # 文件名
    filename = curpath + '/resultreport/'+ now + 'resultreport.html'

    # 打开文件，编写内容
    with open(filename, 'wb') as fp:

        runner = HTMLTestRunner.HTMLTestRunner(stream=fp, title=u"测试报告",
                                               description=u"用例执行情况", verbosity=2)
        suite = createsuite()
        runner.run(suite)

在这里插入图片描述

5.2 异常捕捉与错误截图

  • BUG 的要素：标题、版本号、测试环境、测试步骤（测试数据）、预期结果、实际结果、附件（错误日志、错误截图）
  • BUG 复现：让 BUG 再次出现，使开发人员能更快的定位到 BUG。
  • 可以捕捉到错误，并且把错误截图保存，给我们错误定位带来方便。

5.2.1 错误截图：get_screenshot_as_file()

    def save_errorImage(self, driver,fileName):
        # "./" 表示当前路径
        if not os.path.exists("./image"):
            os.makedirs("./image")
        # 错误截图的名称要不一样
        now = time.strftime("%Y%m%d-%H%M%S", time.localtime(time.time()))
        driver.get_screenshot_as_file("./image/" + now + fileName)

5.2.2 异常捕捉

        try:
            self.assertEqual("肖战_百度搜索", self.driver.title, msg="和预期结果不一致")
        except:
            self.save_errorImage(self.driver, "error.png")

5.2.3 整体代码

from selenium import webdriver
import time
import unittest
import os

class TestUnit1(unittest.TestCase):
    # 获取浏览器的驱动
    def setUp(self):
        # 1、self 就是类的引用/实例
        # 2、全局变量的定义：self.变量名
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.url = "https://www.baidu.com/"
        self.driver.get(self.url)
        time.sleep(3)

        # 在百度中搜索信息
        # 测试用例的命名： test_

    def test_search1(self):
        self.driver.find_element_by_id("kw").send_keys("顾一野")
        self.driver.find_element_by_id("su").click()
        time.sleep(6)
    # 在百度中搜索英文
    def test_search2(self):
        self.driver.find_element_by_id("kw").send_keys("Lisa")
        self.driver.find_element_by_id("su").click()
        time.sleep(6)
        # 异常捕捉 再传参数时，没有传 self，实例也可以不用传。 在TestCase类下，只有test_ 才执行，其他的只有被调用才执行。
        try:
            self.assertEqual("肖战_百度搜索", self.driver.title, msg="和预期结果不一致")
        except:
            self.save_errorImage(self.driver, "error.png")

    def save_errorImage(self, driver,fileName):
        # "./" 表示当前路径
        if not os.path.exists("./image"):
            os.makedirs("./image")
        # 错误截图的名称要不一样
        now = time.strftime("%Y%m%d-%H%M%S", time.localtime(time.time()))
        driver.get_screenshot_as_file("./image/" + now + fileName)

    # 关闭浏览器
    def tearDowm(self):
        self.driver.quit()

    # 一个入口
    if __name__ == "__main__":
        unittest.main()

在这里插入图片描述

二、数据驱动

  • 例如一个搜索框，如何进行功能测试？
    例如在新闻方面，在输入人名（中文。英文）等等，看是否搜索出对应信息。等等信息。
  • 自动化测试用例：
    • 1、定位输入框 “kw” ，在输入框中输入内容。
    • 2、定位百度一下按钮 “su”，点击按钮。
    • 3、判断搜索内容。
  • 上述方法太过冗杂，可以使用数据驱动，一次性完成需要多次执行的测试用例。

1、ddt 安装

  • 1、查看是否安装 ddt，在 cmd 中输入 pip show ddt 出现版本号就说明安装成功。
  • 2、没有安装，在 cmd 中输入 pip install ddt 安装。

2、ddt 使用

from selenium import webdriver
import time
import unittest
from ddt import ddt, unpack, data, file_data
import sys, csv, os

# 如何读取 TXT 文件中的数据形成数组
def getTxt(file_name):
    rows = []
    path = sys.path[0]
    with open(path + "/" + file_name, 'rt') as f:
        readers = csv.reader(f, delimiter=",", quotechar="|")
        next(readers, None)
        for row in readers:
            temprows=[]
            for i in row:
                temprows.append(i)
            rows.append(temprows)
        print(rows)
        return rows

@ddt
class Search(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.url = "https://www.baidu.com/"
        self.driver.maximize_window()
        self.driver.get(self.url)
        time.sleep(6)

    @unittest.skip("skipping")
    # 1、一次性传一个参数
    @data("肖战", "Lisa", "顾一野", "党章")
    def test_find1(self, value):
        self.driver.find_element_by_id("kw").clear()
        self.driver.find_element_by_id("kw").send_keys(value)
        self.driver.find_element_by_id("su").click()
        time.sleep(5)

    @unittest.skip("skipping")
    # 2、一次性传两个参数 u:utf-8,防止中文乱码 需要 unpack 来一一对应参数
    @data(['Lisa', u"Lisa_百度搜索"], [u'肖战', u"顾一野_百度搜索"])
    @unpack
    def test_find2(self, value, expected_value):
        self.driver.find_element_by_id("kw").clear()
        self.driver.find_element_by_id("kw").send_keys(value)
        self.driver.find_element_by_id("su").click()
        time.sleep(6)
        try:
            self.assertEqual(expected_value, self.driver.title, msg="与预期结果不一致")
        except:
            self.save_errorImage(self.driver, "error.png")
        time.sleep(5)

    @unittest.skip("skipping")
    # 3、使用 json 文件（一次传一个数据） ，测试数据在文件里，整洁、保存多
    @file_data('test_baidu_data.json')
    def test_find3(self, value):
        self.driver.find_element_by_id("kw").clear()
        self.driver.find_element_by_id("kw").send_keys(value)
        self.driver.find_element_by_id("su").click()
        time.sleep(5)

    # 4、传入一个 txt 文件（可以传多组数据）
    @data(*getTxt('test_baidu_data.txt'))
    @unpack
    def test_find4(self, value, expected_value):
        self.driver.find_element_by_id("kw").clear()
        self.driver.find_element_by_id("kw").send_keys(value)
        self.driver.find_element_by_id("su").click()
        time.sleep(5)
        # 断言
        self.assertEqual(expected_value, self.driver.title, msg="与预期结果不一致")
        print(expected_value)
        print(self.driver.title)
        time.sleep(5)


    def save_errorImage(self, driver, fileName):
        if not os.path.exists("./image"):
            os.makedirs("./image")
        now = time.strftime("%Y%m%d-%H%M%S",time.localtime(time.time()))
        driver.get_screenshot_as_file("./image/" + now + "-"+ fileName)

    # 关闭浏览器
    def tearDown(self):
       self.driver.quit()

    # 一个入口
    if __name__ == "__main__":
        unittest.main()

test_baidu_data.txt

data
周迅, 周迅_百度搜索
肖战, 肖战_百度搜索

test_baidu_data.json

[
"jolin",
"林俊杰"
]
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请联系我们举报，一经查实，本站将立刻删除。

发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/181842.html原文链接：https://javaforall.net

赞 (0)
0 0

关于作者

全栈程序员-站长的头像

全栈程序员-站长

133.5K 文章
3 粉丝
本网站汇聚当前互联网主流语音，持续更新，欢迎关注公众号“全栈程序员社区”
上一篇 2022年10月15日 上午8:00
下一篇 2022年10月15日 上午8:00


相关推荐

  • webdriver自动化测试工具的使用，将chromedriver配置到path环境变量中，并测试是否成功

    webdriver自动化测试工具的使用，将chromedriver配置到path环境变量中，并测试是否成功

    webdriver自动化测试工具的使用，将chromedriver配置到path环境变量中，并测试是否成功文章目录 webdriver 概述安装 chromewebdri 查看自己的 chrome 谷歌浏览器版本 2 去国内镜像地址下载对应浏览器版本的 webdriver3 下载好之后解压 将解压后的 exe 文件放到你的 chrome 浏览器的安装目录下配置 Chromedriver 的环境变量测试是否配置成功 webdriver 概述是一个非常好用的用来进行复杂重复的 web 自动化测试的工具 主要是 它可以用于我们进行爬虫 WebDriver Selenium2 它的主要新功能是集成了 Selenium1 0 以

    全栈程序员-站长的头像 全栈程序员-站长
    2026年3月18日
    2
  • Java实现矩阵相乘问题

    Java实现矩阵相乘问题

    Java实现矩阵相乘问题1 问题描述 1 1 实验题目设 M1 和 M2 是两个 n n 的矩阵 设计算法计算 M1 M2 的乘积 1 2 实验目的 1 提高应用蛮力法设计算法的技能 2 深刻理解并掌握分治法的设计思想 3 理解这样一个观点 用蛮力法设计的算法 一般来说 经过适度的努力后 都可以对其进行改进 以提高算法的效率 1 3 实验要求 1 设计并实现用 BF Brute Force 即蛮力法 方法求解矩阵相乘问题

    全栈程序员-站长的头像 全栈程序员-站长
    2026年3月17日
    2
  • Tomcat 的类加载机制

    Tomcat 的类加载机制

    Tomcat 的类加载机制Tomcat实际上只有WebAppClassLoader加载器中打破了双亲委派，其他类加载器还是遵循双亲委派的。这样做最主要原因是保证同个Web容器中的不同Web应用程序所使用的类库相互独立，避免相互影响

    全栈程序员-站长的头像 全栈程序员-站长
    2022年5月23日
    41
  • Serverlet理解

    Serverlet理解

    Serverlet理解部分转载自 https blog csdn net javaloveipho article details 从上图可以看出 Tomcat 的容器分为四个等级 真正管理 Servlet 的容器是 Context 容器 一个 Context 对应一个 Web 工程 除了将 Servlet 包装成 StandardWrap 并作为子容器添加到 Context 中 其它的

    全栈程序员-站长的头像 全栈程序员-站长
    2026年3月19日
    2
  • ubuntu 18.04 配置 ASF 挂卡环境

    ubuntu 18.04 配置 ASF 挂卡环境

    ubuntu 18.04 配置 ASF 挂卡环境我已经的身体不能离开 ASF 了 服务器运行 ubuntu18 04 详细请见 https www vultr com ref 官方中文 wikihttps github com JustArchiNET ArchiSteamFa wiki Home zh CN 环境搭建 1 安装 net 来自微软 RegisterMicr

    全栈程序员-站长的头像 全栈程序员-站长
    2025年10月4日
    7
  • RRC信令过程概述

    RRC信令过程概述

    RRC信令过程概述1RRC 协议功能为 NAS 层提供连接管理 消息传递等服务 对接入网的底层协议实体提供参数配置的功能 负责 UE 移动性管理相关的测量 控制等功能 2 nbsp RRC 状态 RRC IDLE nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp PLMN 选择 nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp NAS 配置的 DRX 过程 nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp 系统信息广播和寻呼 nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp nbsp 邻小区测量 nbsp nbsp nbsp nbsp

    全栈程序员-站长的头像 全栈程序员-站长
    2025年9月18日
    8

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注

关注全栈程序员社区公众号