28. Flask 使用unittest进行单元测试

2024-08-21 03:58

本文主要是介绍28. Flask 使用unittest进行单元测试,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

为什么要测试?

Web程序开发过程一般包括以下几个阶段:需求分析,设计阶段,实现阶段,测试阶段。其中测试阶段通过人工或自动来运行测试某个系统的功能。目的是检验其是否满足需求,并得出特定的结果,以达到弄清楚预期结果和实际结果之间的差别的最终目的。

测试的分类:

测试从软件开发过程可以分为:单元测试、集成测试、系统测试等。在众多的测试中,与程序开发人员最密切的就是单元测试,因为单元测试是由开发人员进行的,而其他测试都由专业的测试人员来完成。所以作为开发人员主要需要学习单元测试。

什么是单元测试?

程序开发过程中,写代码是为了实现需求。当我们的代码通过了编译,只是说明它的语法正确,功能能否实现则不能保证。 因此,当我们的某些功能代码完成后,为了检验其是否满足程序的需求。可以通过编写测试代码,模拟程序运行的过程,检验功能代码是否符合预期。

单元测试就是开发者编写一小段代码,检验目标代码的功能是否符合预期。通常情况下,单元测试主要面向一些功能单一的模块进行。

举个例子:一部手机有许多零部件组成,在正式组装一部手机前,手机内部的各个零部件,CPU、内存、电池、摄像头等,都要进行测试,这就是单元测试。

在Web开发过程中,单元测试实际上就是一些“断言”(assert)代码。

断言就是判断一个函数或对象的一个方法所产生的结果是否符合你期望的那个结果。 python中assert断言是声明布尔值为真的判定,如果表达式为假会发生异常。单元测试中,一般使用assert来断言结果。

断言方法的使用:
# 定义一个list
In [6]: a = [1,3,5,7,9]In [7]: b = 3# 断言判断 b 是否存在 a 中,如果正确,则不会报错
In [8]: assert b in a# 断言如果报错,可以自定义打印错误信息,这里定义错误为 False
In [9]: assert b not in a, 'False'
---------------------------------------------------------------------------
AssertionError                            Traceback (most recent call last)
<ipython-input-9-a3961e666e68> in <module>
----> 1 assert b not in a, 'False'AssertionError: False# 断言报错,但是如果没有自定义错误信息,则只会显示错误
In [10]: assert b not in a
---------------------------------------------------------------------------
AssertionError                            Traceback (most recent call last)
<ipython-input-10-57ad4b59ee88> in <module>
----> 1 assert b not in aAssertionError:In [11]: 

断言语句类似于:

if not expression:raise AssertionError

常用的断言方法:

assertEqual     如果两个值相等,则pass
assertNotEqual  如果两个值不相等,则pass
assertTrue      判断bool值为True,则pass
assertFalse     判断bool值为False,则pass
assertIsNone    不存在,则pass
assertIsNotNone 存在,则pass

如何测试?

写一个斐波那契数列 Fibonacci 来进行测试,验证以下的数字是否符合斐波那契数列

可以测试的数字:

1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233,377,610,987,1597,2584,4181,6765,
# 斐波那契数列 Fibonacci
def fibo(x):if x == 0:resp = 0elif x == 1:resp = 1else:return fibo(x-1) + fibo(x-2)return resp
assert fibo(5) == 5

测试执行判断断言如下:

13423234-9bb6d852954ab268.png

单元测试的基本写法:

首先,定义一个类,继承自unittest.TestCase

import unittest
class TestClass(unitest.TestCase):pass

其次,在测试类中,定义两个测试方法

import unittest
class TestClass(unittest.TestCase):#该方法会首先执行,方法名为固定写法def setUp(self):pass#该方法会在测试代码执行完后执行,方法名为固定写法def tearDown(self):pass

最后,在测试类中,编写测试代码

import unittest
class TestClass(unittest.TestCase):#该方法会首先执行,相当于做测试前的准备工作def setUp(self):pass#该方法会在测试代码执行完后执行,相当于做测试后的扫尾工作def tearDown(self):pass#测试代码def test_app_exists(self):pass

看清楚了上面关于unittest测试框架的基本写法之后,下面来写一个登录的视图函数,然后再写一个视图函数的单元测试。

登录视图函数的单元测试

1.编写一个模拟登录的视图函数 login.py

from flask import Flask, request, jsonifyapp = Flask(__name__)@app.route("/login", methods=["POST"])
def login():"""登录"""name = request.form.get("name")password = request.form.get("password")# ""  0  [] () {} None 在逻辑判断时都是假if not all([name, password]):# 表示name或password中有一个为空或者都为空return jsonify(code=65535, message="参数不完整")if name == "admin" and password =="123456":return jsonify(code=0, message="OK")else:return jsonify(code=65535, message="用户名或密码错误")if __name__ == '__main__':app.run(debug=True)

2. 使用postman测试login登录

首先输入正确的用户名和密码测试,如下:

13423234-f5153d035115192d.png

然后去除用户名或者密码,缺少参数进行请求,如下:

13423234-5a797654398ecdea.png

故意输错密码进行请求,如下:

13423234-eda3ac12e5c86887.png

通过postman测试接口这三种情况是可以的,但是如果每次都要手动去进行这样的单元测试,就会感觉很麻烦了。

那么下面可以将这三种情况写成单元测试的代码,来避免重复测试。

3.编写单元测试代码 test_login.py

import unittest
from login import app
import jsonclass TestLogin(unittest.TestCase):"""定义测试案例"""def setUp(self):"""在执行具体的测试方法前,先被调用"""self.app = app# 激活测试标志app.config['TESTING'] = True# 可以使用python的http标准客户端进行测试# urllib  urllib2  requests# 在这里,使用flask提供的测试客户端进行测试self.client = app.test_client()def test_empty_name_password(self):"""测试模拟场景,用户名或密码不完整"""# 使用客户端向后端发送post请求, data指明发送的数据,会返回一个响应对象response = self.client.post("/login", data={})# respoonse.data是响应体数据resp_json = response.data# 按照json解析resp_dict = json.loads(resp_json)# 使用断言进行验证:是否存在code字符串在字典中self.assertIn("code", resp_dict)# 获取code的返回码的值,验证是否为错误码 65535code = resp_dict.get("code")self.assertEqual(code, 65535)# 测试只传nameresponse = self.client.post("/login", data={"name": "admin"})# respoonse.data是响应体数据resp_json = response.data# 按照json解析resp_dict = json.loads(resp_json)# 使用断言进行验证self.assertIn("code", resp_dict)# 验证错误码 65535code = resp_dict.get("code")self.assertEqual(code, 65535)# 验证返回信息msg = resp_dict.get('message')self.assertEqual(msg, "参数不完整")def test_wrong_name_password(self):"""测试用户名或密码错误"""# 使用客户端向后端发送post请求, data指明发送的数据,会返回一个响应对象response = self.client.post("/login", data={"name": "admin", "password": "123456789"})# respoonse.data是响应体数据resp_json = response.data# 按照json解析resp_dict = json.loads(resp_json)# 使用断言进行验证self.assertIn("code", resp_dict)# 验证错误码code = resp_dict.get("code")self.assertEqual(code, 65535)# 验证返回信息msg = resp_dict.get('message')self.assertEqual(msg, "用户名或密码错误")if __name__ == '__main__':unittest.main()

执行测试如下:

13423234-5bea14e09e4e1706.png

从上面可以看出,大部分的Flask框架的单元测试就是这样的处理流程。下面再提供一个数据库单元测试的示例。

数据库单元测试:

数据单元测试的基本步骤方法如下:
1.替换使用一个创建的testdb测试库,避免影响项目的实际数据库
2.导入代码中构建数据库的模型类、app、db等对象,创建数据库以及创建数据
3.断言查询数据库的数据,正确则单元测试成功
4.测试完毕之后,删除创建的数据表

下面来看看实际代码,如下:

准备用来测试的项目代码 db_database.py
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
import pymysql
pymysql.install_as_MySQLdb()
from flask_migrate import Migrate,MigrateCommand
from flask_script import Shell,Managerapp = Flask(__name__)
manager = Manager(app)class Config(object):"""配置参数"""# 设置连接数据库的URLuser = 'root'password = '***************'database = 'flask_ex'app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://%s:%s@127.0.0.1:3306/%s' % (user,password,database)# 设置sqlalchemy自动更跟踪数据库SQLALCHEMY_TRACK_MODIFICATIONS = True# 查询时会显示原始SQL语句# app.config['SQLALCHEMY_ECHO'] = True# 禁止自动提交数据处理app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = False# 设置密钥,用于csrf_token的加解密app.config["SECRET_KEY"] = "xhosd6f982yfhowefy29f"# 读取配置
app.config.from_object(Config)# 创建数据库sqlalchemy工具对象
db = SQLAlchemy(app)#第一个参数是Flask的实例,第二个参数是Sqlalchemy数据库实例
migrate = Migrate(app,db)#manager是Flask-Script的实例,这条语句在flask-Script中添加一个db命令
manager.add_command('db',MigrateCommand)#定义模型类-作者
class Author(db.Model):__tablename__ = 'author'id = db.Column(db.Integer,primary_key=True)name = db.Column(db.String(32),unique=True)email = db.Column(db.String(64))au_book = db.relationship('Book',backref='author')def __str__(self):return 'Author:%s' %self.name#定义模型类-书名
class Book(db.Model):__tablename__ = 'books'id = db.Column(db.Integer,primary_key=True)info = db.Column(db.String(32),unique=True)leader = db.Column(db.String(32))au_book = db.Column(db.Integer,db.ForeignKey('author.id'))def __str__(self):return 'Book:%s,%s'%(self.info,self.leader)if __name__ == '__main__':# 通过管理对象来启动flaskmanager.run()
进行数据库单元测试的代码 test_db.py
import unittest
from db_database import app,db,Author,Book
import timeclass TestLogin(unittest.TestCase):"""定义测试案例"""def setUp(self):"""在执行具体的测试方法前,先被调用"""# 激活测试标志app.config['TESTING'] = True# 设置用来测试的数据库,避免使用正式数据库实例[覆盖原来项目中的数据库配置]user = 'root'password = '***********'# 设置数据库,测试之前需要创建好 create database testdb charset=utf8;database = 'testdb'app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://%s:%s@127.0.0.1:3306/%s' % (user, password, database)self.app = app# 创建数据库的所有模型表:Author、Book模型表db.create_all()def tearDown(self):# 测试结束操作,删除数据库db.session.remove()db.drop_all()# 测试代码def test_append_data(self):au = Author(name='quyuan')bk = Book(info='python_book')db.session.add_all([au, bk])db.session.commit()author = Author.query.filter_by(name='quyuan').first()book = Book.query.filter_by(info='python_book').first()# 断言数据存在self.assertIsNotNone(author)self.assertIsNotNone(book)# 休眠10秒,可以到数据库中查询表进行确认time.sleep(10)if __name__ == '__main__':unittest.main()

测试执行,执行过程查看mysql的数据库表,如下:

# 切换数据库testdb
mysql> use testdb;
Database changed
mysql> 
# 查看表为空
mysql> show tables;
Empty set (0.00 sec)# 执行过程,创建表成功
mysql> show tables;
+------------------+
| Tables_in_testdb |
+------------------+
| author           |
| books            |
+------------------+
2 rows in set (0.00 sec)mysql> 
# 执行完毕,表被全部删除
mysql> show tables;
Empty set (0.00 sec)mysql> 

查看单元测试执行成功,如下:

13423234-9d930e9eea068ee7.png
13423234-0e3934319aa622f6.png

这篇关于28. Flask 使用unittest进行单元测试的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1092059

相关文章

C++中assign函数的使用

《C++中assign函数的使用》在C++标准模板库中,std::list等容器都提供了assign成员函数,它比操作符更灵活,支持多种初始化方式,下面就来介绍一下assign的用法,具有一定的参考价... 目录​1.assign的基本功能​​语法​2. 具体用法示例​​​(1) 填充n个相同值​​(2)

MySQL深分页进行性能优化的常见方法

《MySQL深分页进行性能优化的常见方法》在Web应用中,分页查询是数据库操作中的常见需求,然而,在面对大型数据集时,深分页(deeppagination)却成为了性能优化的一个挑战,在本文中,我们将... 目录引言:深分页,真的只是“翻页慢”那么简单吗?一、背景介绍二、深分页的性能问题三、业务场景分析四、

Spring StateMachine实现状态机使用示例详解

《SpringStateMachine实现状态机使用示例详解》本文介绍SpringStateMachine实现状态机的步骤,包括依赖导入、枚举定义、状态转移规则配置、上下文管理及服务调用示例,重点解... 目录什么是状态机使用示例什么是状态机状态机是计算机科学中的​​核心建模工具​​,用于描述对象在其生命

使用Python删除Excel中的行列和单元格示例详解

《使用Python删除Excel中的行列和单元格示例详解》在处理Excel数据时,删除不需要的行、列或单元格是一项常见且必要的操作,本文将使用Python脚本实现对Excel表格的高效自动化处理,感兴... 目录开发环境准备使用 python 删除 Excphpel 表格中的行删除特定行删除空白行删除含指定

SpringBoot结合Docker进行容器化处理指南

《SpringBoot结合Docker进行容器化处理指南》在当今快速发展的软件工程领域,SpringBoot和Docker已经成为现代Java开发者的必备工具,本文将深入讲解如何将一个SpringBo... 目录前言一、为什么选择 Spring Bootjavascript + docker1. 快速部署与

深入理解Go语言中二维切片的使用

《深入理解Go语言中二维切片的使用》本文深入讲解了Go语言中二维切片的概念与应用,用于表示矩阵、表格等二维数据结构,文中通过示例代码介绍的非常详细,需要的朋友们下面随着小编来一起学习学习吧... 目录引言二维切片的基本概念定义创建二维切片二维切片的操作访问元素修改元素遍历二维切片二维切片的动态调整追加行动态

prometheus如何使用pushgateway监控网路丢包

《prometheus如何使用pushgateway监控网路丢包》:本文主要介绍prometheus如何使用pushgateway监控网路丢包问题,具有很好的参考价值,希望对大家有所帮助,如有错误... 目录监控网路丢包脚本数据图表总结监控网路丢包脚本[root@gtcq-gt-monitor-prome

Python通用唯一标识符模块uuid使用案例详解

《Python通用唯一标识符模块uuid使用案例详解》Pythonuuid模块用于生成128位全局唯一标识符,支持UUID1-5版本,适用于分布式系统、数据库主键等场景,需注意隐私、碰撞概率及存储优... 目录简介核心功能1. UUID版本2. UUID属性3. 命名空间使用场景1. 生成唯一标识符2. 数

linux解压缩 xxx.jar文件进行内部操作过程

《linux解压缩xxx.jar文件进行内部操作过程》:本文主要介绍linux解压缩xxx.jar文件进行内部操作,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录一、解压文件二、压缩文件总结一、解压文件1、把 xxx.jar 文件放在服务器上,并进入当前目录#

SpringBoot中如何使用Assert进行断言校验

《SpringBoot中如何使用Assert进行断言校验》Java提供了内置的assert机制,而Spring框架也提供了更强大的Assert工具类来帮助开发者进行参数校验和状态检查,下... 目录前言一、Java 原生assert简介1.1 使用方式1.2 示例代码1.3 优缺点分析二、Spring Fr