Python Flask，单元测试案例，unittest

 

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()  # 进行测试