Python Flask，单元测试案例，unittest-蒲公英云

Python Flask，单元测试案例，unittest

向右看齐 2022-03-31 12:38 429阅读 0赞

demo.py（Flask，定义视图）：

# coding:utf-8
from flask import Flask, request, jsonify
app = Flask(__name__)
@app.route("/dologin", methods=["POST"])
def login():
    """登录"""
    name = request.form.get("name")
    password = request.form.get("password")
    if not all([name, password]):
        # 如果name或password中有一个为空或者都为空
        return jsonify(code=1, message=u"参数不完整")
    if name == "admin" and password =="python":
        return jsonify(code=0, message=u"OK")
    else:
        return jsonify(code=2, message=u"用户名或密码错误")
if __name__ == '__main__':
    app.run()

test.py（unittest，单元测试案例，测试Flask视图函数）：

# coding:utf-8
import unittest
from demo import app
import json
class TestLogin(unittest.TestCase):
    """定义测试案例"""
    # 测试代码执行之前调用 (方法名固定)
    def setUp(self):
        """在执行具体的测试方法前，先被调用"""
        # 可以使用python的http标准客户端进行测试
        # urllib  urllib2  requests
        # app.config['TESTING'] = True  # 指定app在测试模式下运行
        app.testing = True   # 指定app在测试模式下运行。 (测试模式下,视图中的意外异常可以正常打印显示出来)
        # 使用flask提供的测试客户端进行测试 (Flask客户端可以模拟发送请求)
        self.client = app.test_client()
    # 测试代码。 (方法名必须以"test_"开头)
    def test_empty_name_password(self):
        """测试模拟场景，用户名或密码不完整"""
        # 使用Flask客户端向后端发送post请求, data指明发送的数据，会返回一个响应对象
        response = self.client.post("/dologin", data={})
        # 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, 1)
        # 测试只传name
        response = self.client.post("/dologin", data={"name": "admin"})
        # 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, 1)
    # 测试代码。 (方法名必须以"test_"开头)
    def test_wrong_name_password(self):
        """测试用户名或密码错误"""
        response = self.client.post("/dologin", data={"name": "admin", "password": "xxx"})
        # 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, 2)
if __name__ == '__main__':
    unittest.main()  # 进行测试