前言
前面已经通过代理实现了抓包自动生成 yaml 用例的功能，通过代理也可以实现 mock 功能。
mock 有2种场景：
1.直接拦截发出去的请求，还未到达服务端，模拟自定义返回结果
2.发出去的请求，服务端有反回，拦截返回的结果，篡改返回内容，模拟自己需要的数据
拦截发出去的请求
先看第一种场景：直接拦截发出去的请求，还未到达服务端，模拟自定义返回结果
mt_mock.py 内容如下
from mitmproxy import http
class MockAPI:
def request(self, flow: http.HTTPFlow):
print("------------拦截请求----------------")
if flow.request.pretty_url == "
# 构造自定义 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基本环境准备参考前面这篇 本机开启代理，设置8099端口。
浏览器访问 地址，就会看到模拟的返回结果
拦截返回的结果，篡改返回内容
第二种场景：发出去的请求，服务端有反回，拦截返回的结果，篡改返回内容，模拟自己需要的数据
使用示例：访问http://www.example.com/ 本来返回的是html格式，我改下返回json格式
from mitmproxy import http
class MockAPI:
def request(self, flow: http.HTTPFlow):
print("------------拦截请求----------------")
if flow.request.pretty_url == ":
# 构造自定义 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()
]访问后返回结果
