pytest + yaml 框架 -42.支持 mock 功能
前言
前面已经通过代理实现了抓包自动生成 yaml 用例的功能,通过代理也可以实现 mock 功能。
mock 有2种场景:
1.直接拦截发出去的请求,还未到达服务端,模拟自定义返回结果
2.发出去的请求,服务端有反回,拦截返回的结果,篡改返回内容,模拟自己需要的数据
(备注:从v1.2.4 以后新版本不再公开,新功能内部 VIP 学员可以使用,公开版本仅解决bug, 不提供新功能了。)
拦截发出去的请求
先看第一种场景:直接拦截发出去的请求,还未到达服务端,模拟自定义返回结果
mt_mock.py 内容如下
from mitmproxy import http
class MockAPI:
def request(self, flow: http.HTTPFlow):
print("------------拦截请求----------------")
if flow.request.pretty_url == "https://www.cnblogs.com/yoyoketang/":
# 构造自定义 response
flow.response = http.Response.make(
200, # 返回状态码
"自定义返回内容: 上海-悠悠", # 返回content str or bytes
{"Content-Type": "text/html"} # 返回 headers
)
addons = [
MockAPI()
]
启动服务
>mitmweb -s ./mt_mock.py -p 8099
基本环境准备参考前面这篇https://www.cnblogs.com/yoyoketang/p/17495374.html
本机开启代理,设置8099端口。
浏览器访问https://www.cnblogs.com/yoyoketang/
地址,就会看到模拟的返回结果
拦截返回的结果,篡改返回内容
第二种场景:发出去的请求,服务端有反回,拦截返回的结果,篡改返回内容,模拟自己需要的数据
使用示例:访问http://www.example.com/
本来返回的是html格式,我改下返回json格式
from mitmproxy import http
class MockAPI:
def request(self, flow: http.HTTPFlow):
print("------------拦截请求----------------")
if flow.request.pretty_url == "https://www.cnblogs.com/yoyoketang/":
# 构造自定义 response
flow.response = http.Response.make(
200, # 返回状态码
"自定义返回内容: 上海-悠悠", # 返回content str or bytes
{"Content-Type": "text/html"} # 返回 headers
)
def response(self, flow: http.HTTPFlow):
if flow.request.pretty_url == "http://www.example.com/":
# 修改返回结果
print(f'状态码: {flow.response.status_code}')
flow.response.headers["Content-Type"] = "application/json"
flow.response.set_text('{"code": 0, "message": "success"}')
addons = [
MockAPI()
]
访问后返回结果
网易云完整视频课程https://study.163.com/course/courseMain.htm?courseId=1213419817&share=2&shareId=480000002230338
报名咨询wx:283340479 (已报名的同学学习过程中有问题,都可以协助解决)