python - 如何为 Rester 服务（flask）的上传、下载功能编写单元测试用例？

Question

我是 python 和单元测试的新手，我如何为下面的烧瓶Flask 文档上传、重定向功能编写单元测试用例。提前致谢

from flask import request

@app.route('/upload', methods=['GET', 'POST'])
def upload_file():
    if request.method == 'POST':
        f = request.files['the_file']
        f.save('/uploaded_file.txt')

Answer 1

您应该在测试中放置一个模拟对象而不是文件来请求。在您的测试中调用此方法并验证在您的模拟中使用正确的参数调用了 save 方法之后。