flask框架单元测试小案例

前言

单元测试，及测试某块代码的逻辑是否正确。
通过flask对象的客户端来模拟发送请求。
使用断言assert进行判断，断言成功，程序继续执行；断言失败，抛出断言失败。

登录代码（login.py）

# -*- coding:utf-8 -*-


from flask import Flask
from flask import request, jsonify

app = Flask(__name__)


@app.route("/login", methods=["POST"])
def login():
	username = request.form.get("username")
	password = request.form.get("password")
	
	if not all([username, password]):
		resp = {
			"code": 1,
			"message": "invalid params"
		}
		return jsonify(resp)
	
	if username == "admin" and password == "python":
		resp = {
			"code": 0,
			"message": "login success"
		}
		return jsonify(resp)
	else:
		resp = {
			"code": 2,
			"message": "wrong use