python 设置统一的接口返回 python接口返回json

需求场景:

公司微服务接口使用数字签名的方式, 使用Postman调试接口每次都需要修改源码将验签临时关闭, 但是关闭后,其他微服务不能正常调用该服务

从ELK中抓取到的接口参数, 需要Json格式化, 我通常使用Sublime Text + PrettyJson插件, 使用Postman调试, 有些表单类接口, 还需要把每个参数及值复制出来, 特别麻烦.

实现原理:

Sublime text3可以自定义编译系统

Python方便的json数据处理

使用Python实现签名算法

requests发送接口请求

实现目标:

采用json文件数据驱动, 支持一个json包含多个接口

支持GET/POST/DELETE/...方法

支持表单类/Json类请求

支持headers, cookies, params, data, files, timeout

支持session依赖, 中间变量

数据文件格式:

Get请求

{
"url": "http://127.0.0.1:5000/api/user/logout/?name=张三"
}
或
{
"url": "http://127.0.0.1:5000/api/user/logout/",
"params": {"name": "张三"}
}
Post表单请求:
{
"url": "http://127.0.0.1:5000/api/user/login/",
"data": {"name": "张三","password": "123456"}
}
Json格式请求:
{
"url": "http://127.0.0.1:5000/api/user/reg/",
"headers": {"Content-Type": "application/json"},
"data": { "name": "张991", "password": "123456"}
}
或
{
"url": "http://127.0.0.1:5000/api/user/reg/",
"type": "json",
"data": { "name": "张991","password": "123456"}
}
断言格式(ASSERT):
{
"url": "http://127.0.0.1:5000/api/user/login/",
"data": {"name": "张三","password": "123456"},
"ASSERT": ["response.status_code == 200", "'成功' in response.text"]
}
需要签名(SIGN):
{ "url": "http://127.0.0.1:5000/api/user/delUser/",
"type": "json",
"data": {"name": "张三"},
"SIGN": true
}
多接口,session依赖, 及中间变量(STORE)
[
{
"url": "http://127.0.0.1:5000/api/user/getToken/",
"method": "get",
"params": {"appid": "136425"},
"STORE": {"token": "response.text.split('=')[1]"}
},
{ "url": "http://127.0.0.1:5000/api/user/updateUser/?token=%(token)s",
"headers": {"Content-Type": "application/json"},
"data": {"name": "张三","password": "123456"}
}
]
工具实现:
sign.py
import hashlib
appid = user
appsecret = NTA3ZTU2ZWM5ZmVkYTVmMDBkMDM3YTBi
def md5(str):
m = hashlib.md5()
m.update(str.encode('utf8'))
return m.hexdigest() #返回摘要，作为十六进制数据字符串值
def makeSign(params):
if not isinstance(params, dict):
print("参数格式不正确，必须为字典格式")
return None
if 'sign' in params:
params.pop('sign')
sign = ''
for key in sorted(params.keys()):
sign = sign + key + '=' + str(params[key]) + '&'
sign = md5(sign + 'appsecret=' + appsecret)
params['sign'] = sign
return params
post_json.py
import json
import os
import requests
from sign import makeSign
timeout = 10
def postJson(path, timeout=60):
try:
with open(path, encoding='utf-8') as f:
apis = json.load(f)
except IOError as e:
print(e)
except json.decoder.JSONDecodeError:
print("json文件格式有误")
if isinstance(apis, dict):
apis=[apis]
session = requests.session()
for api in apis:
# 处理全局变量
str_api = json.dumps(api)
if '%' in str_api:
api = json.loads(str_api % globals())
# 获取接口相关项
url = api.get('url')
method = api.get('method')
type = api.get('type')
headers = api.get('headers')
cookies = api.get('cookies')
params = api.get('params')
data = api.get('data')
files = api.get('files')
# 特定动作(系统标签)
_sign = api.get('SIGN') # 对数据进行签名
_store = api.get('STORE') # 存储变量
_assert = api.get('ASSERT') # 结果断言
# 处理签名
if _sign:
data = makeSign(data)
# 如果json请求，转换一下数据
if type and type == 'json' or headers and 'json' in json.dumps(headers):
data = json.dumps(data)
# 根据method发送不同请求
if not method or method.lower() == 'post': # 有data字段默认使用post方法
response = session.post(url=url, headers=headers, cookies=cookies, params=params, data=data, files=files, timeout=timeout)
elif not data or method.lower() == 'get': # 没有data字段默认采用get方法
response = session.get(url=url, headers=headers, cookies=cookies, params=params, timeout=timeout)
elif method.lower() == 'delete':
response = session.delete(url=url, headers=headers, cookies=cookies, params=params, data=data, files=files, timeout=timeout)
elif method.lower() == 'put':
response = session.put(url=url, headers=headers, cookies=cookies, params=params, data=data, files=files, timeout=timeout)
elif method.lower() == 'patch':
response = session.patch(url=url, headers=headers, cookies=cookies, params=params, data=data, files=files, timeout=timeout)
elif method.lower() == 'head':
response = session.head(url=url, headers=headers, cookies=cookies, params=params, data=data, files=files, timeout=timeout)
else:
print("不支持当前方法")
# 存储中间结果
if _store:
for key in _store:
globals()[key]=eval（_store[key])
# 处理响应
try:
response_text = json.dumps(response.json(), ensure_ascii=False, indent=2)
except json.decoder.JSONDecodeError: # only python3
try:
response_text = response.text
except UnicodeEncodeError:
# print(response.content.decode("utf-8","ignore").replace('\xa9', ''))
response_text = response.content
finally:
pass
# 处理断言
status = "PASS"
if _assert:
assert_results = []
for item in _assert:
try:
assert eval（item)
assert_results.append("PASS: " % item)
except AssertionError:
assert_results.append("FAIL: " % item)
status = "FAIL"
except Exception as e:
assert_results.append("ERROR: \n%s" % (item, repr(e)))
status = "ERROR"
# 打印结果
print("="*80)
print("请求:")
print("Url: %s\nHeaders: %s\nData: %s" % (url, headers, data if isinstance(data, str) else json.dumps(data)))
print("-"*80)
print("响应:")
print(response_text)
if _assert:
print("-"*80)
print("断言:")
for assert_result in assert_results:
print(assert_result)
def main():
if len(sys.argv) != 2:
print("缺少参数：json文件")
else:
postJson(sys.argv[1])
main()
**集成到Sublime **
Sublime Text3 -> Tools -> Build System -> New Build System
复制以下代码, 修改Python为你的Python3环境, 脚本路径为你的post_json.py路径, 保存为PostJson.sublimet-build
Windows 系统
{
"cmd": ["python3","C:/PostJson/post_json.py","$file"],
"file_regex": "^[ ] *File \"(...*?)\", line ([0-9]*)",
"selector": "source.json",
"encoding":"cp936",
}
Linux & Mac
{
"cmd": ["/Library/Frameworks/Python.framework/Versions/3.6/bin/python3","-u","/Users/apple/Applications/PostJson/post_json.py","$file"],
"file_regex": "^[ ] *File \"(...*?)\", line ([0-9]*)",
"selector": "source.json",
"env": {"PYTHONIOENCODING": "utf8"},
}
使用方法
建议安装PrettyJson插件
在Sublime text3中, 新建json文件,并保存为*.json
{
"url": "http://127.0.0.1:5000/api/user/login/",
"data": {"name": "张三","password": "123456"}
}

按Ctrl + B运行

项目路径

补充了以下功能:

用例discover, 自动发现和批量执行

生成报告

命令行参数

verbose: 显示级别: 1. 只显示用例结果 2. 只显示响应文本(默认) 3. 显示请求,响应及断言信息

host: 指定服务器地址

env: 指定环境

timeout: 指定timeout

html: 指定输出html报告路径

log: 指定输出log路径

collect-only: 只列出所有用例

ToDo

完善测试报告

支持数据库对比

支持执行Sql, Shell, Cmd命令等

使用协程, 进行异步并发

支持分布式执行

支持定时执行

进行持续集成