利用trip包实现并发POST请求测试

工作需要，今天写了一个人员注册的压力测试程序，以前都是利用requests+threading来实现的，感觉还是有点麻烦。所以网上找了一下有没有其他轮子，果不其然，找到了 trip这个轮子，作者结合Tornado与Requests来实现。其中参考文档https://trip.readthedocs.io/zh_CN/latest/。

如下代码部分，伪数据产生用faker和random来实现。其中faker库用来造数据再好不过了，常见的接口字段都可以随机生成，顺带提一下fake2db库，是一个伪数据库生成库。
由于是测试人脸识别图片功能，需要post带人脸的图片，这里只好取巧，本地收集了一堆人脸照片到某个文件夹，利用random随机选取。其实如果只是普通图片的话，可以自己写个爬虫脚本在线随机爬取。
值得注意的是post需要传递 [{"tag": "姓名", "value": self.fake.name()}, {"tag": "身份证号", "value": self.fake.ssn()}, {"tag": "性别", "value": random.choice(["男", "女"])}] 这一堆，需要将其强制转成字符串，但是会导致其中的双引号变成单引号，不符合json格式，所以需要用replace替换回来。

# -*- coding: utf-8 -*-
import trip
import faker
import random
import os

class Regiser():

  def __init__(self, num=1):
    self.fake = faker.Faker("zh_CN")
    self.url = "http://xxx.xxx.x.x:9180/seeta/register/enrollment"
    self.base_image_url = "C:/Users/YuChou/Desktop/人员/"
    self.image_href_list = os.listdir(self.base_image_url)
    self.num = num  # 同时注册数据量

  # 生成伪数据表单
  def product_formdata(self):
    data_list = [{"tag": "姓名", "value": self.fake.name()},
                 {"tag": "身份证号", "value": self.fake.ssn()},
                 {"tag": "性别", "value": random.choice(["男", "女"])}]

    formData = {
      "secret_key": "b3a9b9bb20544f9eb0e42a5b93a73aa1",
      "meeting_key": "6c16544d0a004774b5d21c85bd0322be",
      "attributes": str(data_list).replace("\'", "\"") #传json 转字符串会把"转成' 需要重新替换回来 不然会报json错误
    }
    return formData

  # 生成随机图片
  def product_image(self):
    files = {"image": open(self.base_image_url + random.choice(self.image_href_list), "rb")}
    return files

  # 并发
  @trip.coroutine
  def request(self):
    req = yield [trip.post(url=self.url, data=self.product_formdata(), files=self.product_image()) for i in range(self.num)]
    print(list(map(lambda x: x.text, req)))

  # 执行协程
  def main(self):
    trip.run(self.request)

if __name__ == "__main__":
  obj = Regiser(num=16)
  obj.main()

这里设置了16个同时注册，16个总时间算下来小于2.5s，前面有试过单个注册就需要2s左右，基本实现并发功能，运行结果：

image.png