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

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

〇、前言

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

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

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

一、文件上传接口

1. 接口文档

Request URL:/createfile

Request Method:POST

Content-Type:multipart/form-data

2. 代码实现

(1)实现步骤:

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

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

     
      
    1. "file": (

    2.     "caituotuo-file.pdf", # 元组第一个值为文件名称,没有则取None

    3.     open(r"F:\caituotuo-file.pdf", "rb"), # 若第一个值非None,则取文件open打开的二进制流,否则直接写文件路径,如"F:\caituotuo-file.pdf"

    4.     "pdf" # 文件类型

    5. )

     
      
    1. "file": (

    2.     None,

    3.     "F:\caituotuo-file.pdf"

    4. )

  2. 构造其他数据

     
      
    1. {

    2.     "title": "接口上传的文件",

    3.     "fileType": "pdf"

    4. }

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

     
      
    1. req = {

    2.             "url": "127.0.0.1/createfile",

    3.             "method": "POST",

    4.             "headers": {},

    5.             "files": {"file": ("caituotuo-file.pdf", open(r"F:\caituotuo-file.pdf", "rb"), "pdf")},

    6.             "data": {

    7.                 "title": "接口上传的文件",

    8.                 "fileType": "pdf"

    9.             }

    10.         }

(2)完整代码

 
  1. # -*- coding:utf-8 -*-

  2. # 时间:2022/3/12 21:04

  3. # 功能:上传文件demo

  4. import requests

  5. class Createfile:

  6.     

  7.     def createfile(self):

  8.         req = {

  9.             "url": "127.0.0.1/createfile",

  10.             "method": "POST",

  11.             "headers": {},

  12.             "files": {"file": ("", open(r"F:\caituotuo-file.pdf", "rb"), "pdf")},

  13.             "data": {

  14.                 "title": "接口上传的文件",

  15.                 "fileType": "pdf"

  16.             }

  17.         }

  18.         res = requests.request(**req)

  19.         assert res.status_code == 200

  20.         res_json = res.json()

  21.         return res_json["result"]["id"]

  22. if __name__ == '__main__':

  23.     Createfile().createfile()

二、文件下载接口

1. 接口文档

Request URL:/download

Request Method:GET

 

2. 代码实现

 
  1. # -*- coding:utf-8 -*-

  2. # 时间:2022/4/5 2:56

  3. # 功能:下载文件demo

  4. import requests

  5. class Download:

  6.     def download(self):

  7.         req = {

  8.             "url": "127.0.0.1/download",

  9.             "method": "GET",

  10.             "headers": {},

  11.             "params": {

  12.                 "id": 2947403045981869536,

  13.                 "downloadItems": ["NORMAL"],

  14.                 "needCompressForOneFile": False

  15.             },

  16.         }

  17.         res = requests.request(**req).content # 注意“.content"获取返回内容

  18.         # with open("F:/response.zip", "wb") as f: # 多份文件返回压缩包

  19.         with open("F:/response.pdf", "wb") as f:

  20.             f.write(res)

  21.         return res

  22. if __name__ == '__main__':

  23.     Download().download()

以上,完。

感谢每一个认真阅读我文章的人,礼尚往来总是要有的,虽然不是什么很值钱的东西,如果你用得到的话可以直接拿走:

这些资料,对于【软件测试】的朋友来说应该是最全面最完整的备战仓库,这个仓库也陪伴上万个测试工程师们走过最艰难的路程,希望也能帮助到你!有需要的小伙伴可以点击下方小卡片领取   

更多推荐