持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第15天,点击查看活动详情
大家好~我是
米洛
!
我正在从0到1打造一个开源的接口测试平台, 也在编写一套与之对应的教程
,希望大家多多支持。
欢迎关注我的公众号米洛的测开日记
,获取最新文章教程!
回顾
上一节我们讲述了用例录制
相关内容,但是有一些websocket的改造和http请求结果的转换没有讲,这一节我们就来完成他们。
Websocket改造
websocket改造分为2部分:
- 后端部分要支持发送录制好的http请求数据,其实也就是json数据,之前已经支持,我们稍微改动下即可
- 前端部分需要监听新的类型,收到后端发来的消息,根据消息type做出不同的操作
先来看后端部分:
其实我们只需要在websocketManager加入send_data方法即可,也就是能自由地根据用户id发送消息。
再来看看前端部分:
稍微有点变化,根据type为2,判断是录制成功
的消息,我们走存储录制消息的逻辑。
![](/qrcode.jpg)
其实readRecord就是往录制列表里面插入数据
。
Redis部分
@staticmethod
@awaitable
def get_address_record(address: str):
"""
获取ip是否已经开启录制
:param address:
:return:
"""
key = RedisHelper.get_key(f"record:ip:{address}")
return RedisHelper.pity_redis_client.get(key)
@staticmethod
@awaitable
def cache_record(address: str, request):
"""
:param address:
:param request:
:return:
"""
key = RedisHelper.get_key(f"record:{address}:requests")
RedisHelper.pity_redis_client.rpush(key, request)
ttl = RedisHelper.pity_redis_client.ttl(key)
if ttl < 0:
RedisHelper.pity_redis_client.expire(key, 3600)
@staticmethod
@awaitable
def set_address_record(user_id: int, address: str, regex: str):
"""
设置录制状态
:param user_id:
:param address:
:param regex: 录制的url正则
:return:
"""
# 默认录制1小时
value = json.dumps({"user_id": user_id, "regex": regex}, ensure_ascii=False)
RedisHelper.pity_redis_client.set(RedisHelper.get_key(f"record:ip:{address}"), value, ex=3600)
# 清楚上次录制数据
RedisHelper.pity_redis_client.delete(RedisHelper.get_key(f"record:{address}:requests"))
@staticmethod
@awaitable
def remove_address_record(address: str):
"""
停止录制任务
:param address:
:return:
"""
return RedisHelper.pity_redis_client.delete(RedisHelper.get_key(f"record:ip:{address}"))
复制代码
主要由这4个方法组成:
- 获取该ip录制的状态,是开启还是关闭
- 缓存录制结果(上一节已经介绍过)
- 设置录制状态,1小时后过期,也就贴合之前说的,1小时后任务自动停止
- 关闭录制,直接删除key,提前停止录制
所以对应页面上我们会有开始录制
,停止录制
等操作按钮。
HTTP请求解析
这块内容其实不难,就是整合mitmproxy的flow对象里面有的跟http请求相关的内容,我这里只选取了一些用得到的部分,比如headers/body/cookies等等。
__author__ = "woody"
import json
from typing import TypeVar
import pydantic
from loguru import logger
"""
translate mitmproxy request and response data
"""
body = TypeVar("body", bytes, str)
class RequestInfo(pydantic.BaseModel):
url: str = None
body: body = None
request_method: str = None
request_data: str = None
request_headers: str = None
response_headers: str = None
cookies: str = None
request_cookies: dict = None
response_content: str = None
status_code: int = None
def __init__(self, flow):
super().__init__()
self.status_code = flow.response.status_code
self.url = flow.request.url
self.request_method = flow.request.method
self.request_headers = json.dumps(dict(flow.request.headers), indent=4, ensure_ascii=False)
self.response_headers = json.dumps(dict(flow.response.headers), indent=4, ensure_ascii=False)
self.response_content = self.get_response(flow.response)
self.body = self.get_body(flow.request)
self.cookies = json.dumps(dict(flow.response.cookies), indent=4, ensure_ascii=False)
self.request_cookies = dict(flow.request.cookies)
@classmethod
def translate_json(cls, text):
try:
return json.dumps(json.loads(text), indent=4, ensure_ascii=False)
except Exception as e:
logger.bind(name=None).warning(f"解析json格式请求失败: {e}")
return text
@classmethod
def get_response(cls, response):
content_type = response.headers.get("Content-Type")
if "json" in content_type.lower():
return cls.translate_json(response.text)
if "text" in content_type.lower() or "xml" in content_type.lower():
return response.text
return response.data.decode('utf-8')
@classmethod
def get_body(cls, request):
if len(request.content) == 0:
return None
content_type = request.headers.get("Content-Type")
if "json" in content_type.lower():
return cls.translate_json(request.text)
if "text" in content_type.lower() or "xml" in content_type.lower():
return request.text
return request.data.decode('utf-8')
def dumps(self):
return json.dumps(self.dict(), ensure_ascii=False)
复制代码
值得注意的是,dumps方法,是因为我们把requestInfo对象存入redis的时候需要序列化,因为redis不允许存入Python数据,需要转换为bytes或者str。
main.py开启proxy
最后根据proxy的开关判断,是否开启配置,由于proxy也是一个服务,我们单独用一个线程启动它(create_task,并且不等待)
前端部分
其实比较干燥,就是一个很简单的表格
,加上对应的url输入框和具体的按钮:
后续我们需要能够勾选
录制到的请求,并且做一些智能操作,自动提取里面的参数数据,最后完成用例的生成操作。
其中智能提取参数是比较通用的,需要兼容其他比如har文件格式的数据,最后怎么导入到pity成为pity的一员,这个笔者还在筹划。
今天的内容就到这里了,感谢大家的收看,再会~