獲得400在Flask單元測試

Question

我正在爲Flask中的文件上傳函數編寫單元測試。

該應用程序在本地運行良好，但單元測試會拋出400.謝謝！

這裏是我的app.py

from flask import Flask, render_template, request 
app = Flask(__name__) 
app.config["TESTING"] = True 


@app.route('/') 
def index(): 
    return render_template('index.html') 

@app.route('/uploader', methods = ['GET','POST']) 
def upload_photo(): 
    if request.method == 'POST': 
     photo = request.files['photo'] 
     resp = app.make_response(photo.read()) 
     resp.mimetype = 'image/jpeg' 
     return resp 

if __name__ == '__main__': 
    app.run(debug=True) 

這裏是我的測試：

import unittest 
import io 
from app import app 


class TestCase(unittest.TestCase): 

    def setUp(self): 
     self.client = app.test_client() 

    def test_upload_photo(self):    
     result = self.client.post('/uploader', 
    content_type='multipart/form-data', follow_redirects=True, 
    data=dict(upload_var=(io.BytesIO(b'photo'), 'photo'))) 
     print(result.data) 
     self.assertTrue(result.status_code == 200) 


if __name__ == "__main__": 
    unittest.main()` 

Answer 1

變化在數據字典upload_var到photo，因爲這是你的處理函數需要的文件被調用。