接口自动化测试——文件上传/下载

转载请注明出处❤️

作者：测试蔡坨坨

原文链接：caituotuo.top/9cf3344.html

你好，我是测试蔡坨坨。

我们在做接口自动化测试的时候，经常会碰到文件上传接口和文件下载接口。

那么，文件接口跟普通接口有什么区别呢？又该如何实现呢？

〇、前言

文件上传/下载接口与普通接口类似，但是有细微的区别。

如果需要发送文件到服务器，例如：上传文档、图片、视频等，就需要发送二进制数据，上传文件一般使用的都是 Content-Type: multipart/form-data 数据类型，可以发送文件，也可以发送相关的消息体数据。

反之，文件下载就是将二进制格式的响应内容存储到本地，并根据需要下载的文件格式来写文件名，例如：F:/caituotuo-file.pdf。

一、文件上传接口

1. 接口文档

Request URL： /createfile

Request Method： POST

Content-Type： multipart/form-data

名称 类型 是否必须 描述
file File 是 文件
title String 是 文件名称
fileType String 是 文件类型：doc, docx, txt, pdf, png, gif, jpg, jpeg, tiff, html, rtf, xls, txt

2. 代码实现

（1）实现步骤：

  1. 构造文件数据，通过open()函数以二进制方式打开文件

    文件上传接口参数与普通post请求一样，需要写成Key和Value模式，Key为参数名称file（也是组件的name属性），Value为一个元组（与普通接口不同的地方）

    1
    2
    3
    4
    5
    "file": (
        "caituotuo-file.pdf", # 元组第一个值为文件名称，没有则取None
        open(r"F:\caituotuo-file.pdf", "rb"), # 若第一个值非None，则取文件open打开的二进制流，否则直接写文件路径，如"F:\caituotuo-file.pdf"
        "pdf" # 文件类型
    )
    1
    2
    3
    4
    "file": (
        None,
        "F:\caituotuo-file.pdf"
    )

  2. 构造其他数据

    1
    2
    3
    4
    {
        "title": "接口上传的文件",
        "fileType": "pdf"
    }

  3. 发送请求，将文件数据以 files 参数传入，其他消息体数据通过 data 、 headers 、 cookies 等传入

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    req = {
                "url": "127.0.0.1/createfile",
                "method": "POST",
                "headers": {},
                "files": {"file": ("caituotuo-file.pdf", open(r"F:\caituotuo-file.pdf", "rb"), "pdf")},
                "data": {
                    "title": "接口上传的文件",
                    "fileType": "pdf"
                }
            }
（2）完整代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# -*- coding:utf-8 -*-
# 作者：测试蔡坨坨
# 时间：2022/3/12 21:04
# 功能：上传文件demo

import requests


class Createfile:
    
    def createfile(self):
        req = {
            "url": "127.0.0.1/createfile",
            "method": "POST",
            "headers": {},
            "files": {"file": ("", open(r"F:\caituotuo-file.pdf", "rb"), "pdf")},
            "data": {
                "title": "接口上传的文件",
                "fileType": "pdf"
            }
        }
        res = requests.request(**req)
        assert res.status_code == 200
        res_json = res.json()
        return res_json["result"]["id"]


if __name__ == '__main__':
    Createfile().createfile()

二、文件下载接口

1. 接口文档

Request URL：/download

Request Method：GET

名称 类型 是否必须 描述
id Long 是 文件组id
downloadItems String[] 否 下载可选项
needCompressForOneFile Boolean 是，默认单文件也压缩 当下载的文件仅一份时，是否压缩

2. 代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# -*- coding:utf-8 -*-
# 作者：测试蔡坨坨
# 时间：2022/4/5 2:56
# 功能：下载文件demo

import requests


class Download:
    def download(self):
        req = {
            "url": "127.0.0.1/download",
            "method": "GET",
            "headers": {},
            "params": {
                "id": 2947403045981869536,
                "downloadItems": ["NORMAL"],
                "needCompressForOneFile": False
            },
        }
        res = requests.request(**req).content # 注意“.content"获取返回内容
        # with open("F:/response.zip", "wb") as f: # 多份文件返回压缩包
        with open("F:/response.pdf", "wb") as f:
            f.write(res)
        return res


if __name__ == '__main__':
    Download().download()