使用没有 mocking 的 Flask 测试客户端上传 Excel 文件

Question

查看我的 API：

@api.resource('/project/<string:val>/upload')
class UploadView(Resource):
    @classmethod
    @jwt_required
    def post(self, val):
        uploaded_file = request.files['file']
        file_name = secure_filename(uploaded_file.filename)
        file_path = os.path.join('data','files',filename)
        try:
            uploaded_file.save(file_path)
            status_code = 200
            response["result"] = "done"
        except Exception as e:
            print(e)
        return Response(response, status_code)

PyTest function 具有客户端，在功能 scope 中作为夹具传递的标头用于测试 ZA2F2ED4F8EBC2CBBDZC2A：

def test_upload_view(self, client, headers):
    file_path = rootdir + 'file.xlsx'
    file = FileStorage(
            stream=io.BytesIO(open(file_path, "rb").read()),
            filename="uploadedFile.xlsx",
            content_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
       )
    url = "/v1/project/K6400/upload"
    response = client.post(
                url,
                headers=headers,
                data=dict(file=file),
                follow_redirects=True,
                content_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
               )
    assert response.status_code == 200

我已经正确实施了设置和拆卸方法，这些方法也关闭了打开的文件以避免错误。 我仍然在 POST 方法上收到 400 错误。 我的 API 在 UI 上正常工作并下载文件，但在单元测试中，它失败了。 我知道 request.files['file'] 是 FileStorage object 所以我试图在 stream 参数中转换为 BytesIO。 不知道是什么，我做错了。 帮助将不胜感激。

Answer 1

试试下面的代码：

def test_upload_view(self, client, headers):
    file_path = rootdir + 'file.xlsx'
    with open(file_path, 'rb') as xlsxFile:
        fileBytes = ByteIO(xlsxFile.read())

    fileBytes.seek(0)

    url = "/v1/project/K6400/upload"
    response = client.post(
                url,
                headers=headers,
                data=dict(files=[(xlsxFile, myFile.xlsx)]),
                follow_redirects=True
               )
    assert response.status_code == 200

这也将处理同一请求中的多个文件。 读取文件使用：

request.files.to_dict(flat=False).get('files', None)

或删除单个文件的字典

Answer 2

我想出了答案，我不得不使用 BytesIO：

def test_upload_view(self, client, headers):
    file_path = rootdir + 'file.xlsx'
    with open(file_path, "rb") as fp:
        url = "/v1/project/K6400/upload"
        response = client.post(
                url,
                headers=headers,
                data={'file': (io.BytesIO(fp.read()), file_path)},
                follow_redirects=True
               )
    assert response.status_code == 200