pytest + yaml 框架 -42.支持 mock 功能

上海-悠悠

发布于 2023-08-22 11:48:11

1600

发布于 2023-08-22 11:48:11

前言

前面已经通过代理实现了抓包自动生成 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 == "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()
]

访问后返回结果