软件测试学习笔记丨Flask框架-请求与响应

本文转自测试人社区，原文链接：https://ceshiren.com/t/topic/23408

请求方法

from flask import Flaskapp = Flask(__name__)@app.route("/cases", methods=["get"])
def get_case():return {"code": 0, "msg": "get success"}# post请求不能直接通过url访问,可以使用postman或者requests库操作
@app.route("/case", methods=["post"])
def post_case():return {"code": 0, "msg": "post success"}if __name__ == '__main__':app.run()

处理请求数据

注意：是request对象（不是requests）

json可以处理一些post请求的数据

from flask import Flask, requestfrom log_utils import loggerapp = Flask(__name__)# get
@app.route("/login/", methods=["get"])
def hogwarts():logger.info(f"请求参数为{request.args}")result = request.argsa = result.get('a')b = result.get('b')print(a, b)return {"code": 0, "msg": "get success"}# post请求的json数据
@app.route("/regist", methods=["post"])
def post_regist():logger.info(f"请求参数为{request.json}")return {"code": 0, "msg": "post success"}# form表单
@app.route("/put", methods=["put"])
def put_regist():name = request.form.get('name')p = request.form.get('password')email = request.form.get('email')logger.info(f"请求参数为{request.form}")logger.info(f"name：{name},password：{p},email：{email}")return {"code": 0, "msg": "post success"}# 文件上传
@app.route("/file", methods=["post"])
def post_file():obj = request.files.get("file")logger.info(obj)filename = obj.filenamelogger.info(f"文件名为：{filename}")# 将接口接收的文件保存在本地obj.save("./1.txt")return {"code": 0, "msg": "post success"}if __name__ == '__main__':app.run()

由于postman对文件上传有一些bug，所以写了测试代码上传文件:

import requestsdef test_file():url = "http://127.0.0.1:5000/file"file = {"file": open("C:/Users/Administrator/Desktop/1.txt", 'rb')}r = requests.post(url=url, files=file)assert r.status_code == 200

处理响应信息

返回的响应类型：
- 文本：直接return
- 元组
- json
- html
- 额外数据

返回元组有三种情况：

（response，status）
（response，headers）
（response，status，headers）

@app.route("/tuple")
def tuple_result():return "hogwarts", 200, {"test": "demo"}

返回json：

直接返回dict会转换成json
使用jsonify()方法，通过参数传入键值对

@app.route("/dict")
def json_result():return {'status': 0}@app.route("/json")
def json_re():return jsonify({'status': 0})# 也可以通过参数指定# return jsonify(status=1,name="joker")

返回html–render_template(文件名称.html)

使用模板渲染技术

注意：html文件必须在同级名为templates的目录下

@app.route("/html")
def html_demo():return render_template("demo.html")

设置额外数据–make_response()：

设置cookie
设置响应头信息等

@app.route("/")
def index():# 设置返回页面模板res = make_response(render_template('demo.html'))# 设置cookieres.set_cookie('username', 'password')# 设置请求头res.headers['hogwarts'] = 'joker'return res