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 (已报名的同学学习过程中有问题，都可以协助解决)