0
点赞
收藏
分享

微信扫一扫

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

前言

前面已经通过代理实现了抓包自动生成 yaml 用例的功能,通过代理也可以实现 mock 功能。
mock 有2种场景:
1.直接拦截发出去的请求,还未到达服务端,模拟自定义返回结果
2.发出去的请求,服务端有反回,拦截返回的结果,篡改返回内容,模拟自己需要的数据

拦截发出去的请求

先看第一种场景:直接拦截发出去的请求,还未到达服务端,模拟自定义返回结果

pytest + yaml 框架 -42.支持 mock 功能_自定义

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端口。

浏览器访问 地址,就会看到模拟的返回结果

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

拦截返回的结果,篡改返回内容

第二种场景:发出去的请求,服务端有反回,拦截返回的结果,篡改返回内容,模拟自己需要的数据

pytest + yaml 框架 -42.支持 mock 功能_返回结果_03

使用示例:访问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()
]

访问后返回结果

pytest + yaml 框架 -42.支持 mock 功能_返回结果_04




举报

相关推荐

0 条评论