pytest + yaml 框架 - 支持 mock 功能

傷城~ 2023-10-12 21:28 30阅读 0赞

![32891de8b2904d0996959f01ed4ea750.png][]

前面已经通过代理实现了抓包自动生成 yaml 用例的功能，通过代理也可以实现 mock 功能。  
mock 有2种场景：

1.直接拦截发出去的请求，还未到达服务端，模拟自定义返回结果

2.发出去的请求，服务端有反回，拦截返回的结果，篡改返回内容，模拟自己需要的数据

## 拦截发出去的请求 ##

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

![4de5344cefe04475943685e0d6801532.png][]

mt\_mock.py 内容如下

启动服务

>mitmweb -s ./mt_mock.py -p 8099

基本环境准备参考前面这篇https://www.cnblogs.com/yoyoketang/p/17495374.html  
本机开启代理，设置8099端口。

浏览器访问`https://www.cnblogs.com/yoyoketang/` 地址，就会看到模拟的返回结果

![61e94852e1c0423aa7c1d7ac2cb0a7d6.png][]

## 拦截返回的结果，篡改返回内容 ##

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

![f5e85e0f6c4043a4a2124e471f557a68.png][]

使用示例：访问`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,   # 返回状态码
                    "自定义返回内容: 上海-XX",  # 返回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()
    ]

访问后返回结果

![91d6216b36194e108021de5dffe05aba.png][]

**最后：**下方这份完整的软件测试视频教程已经整理上传完成，需要的朋友们可以自行领取**【保证100%免费】**

![8b0171187dce405e878f274a8cdd9ec0.png][]

#### **软件测试面试文档** ####

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

![a3e4226c935c4242804cbdf26b571bc6.png][]

#### 全部资料获取 ####

![69778fb8780b412bb6cfd6cfe695ab2f.gif][]

[32891de8b2904d0996959f01ed4ea750.png]: https://img-blog.csdnimg.cn/32891de8b2904d0996959f01ed4ea750.png
[4de5344cefe04475943685e0d6801532.png]: https://img-blog.csdnimg.cn/4de5344cefe04475943685e0d6801532.png
[61e94852e1c0423aa7c1d7ac2cb0a7d6.png]: https://img-blog.csdnimg.cn/61e94852e1c0423aa7c1d7ac2cb0a7d6.png
[f5e85e0f6c4043a4a2124e471f557a68.png]: https://img-blog.csdnimg.cn/f5e85e0f6c4043a4a2124e471f557a68.png
[91d6216b36194e108021de5dffe05aba.png]: https://img-blog.csdnimg.cn/91d6216b36194e108021de5dffe05aba.png
[8b0171187dce405e878f274a8cdd9ec0.png]: https://img-blog.csdnimg.cn/8b0171187dce405e878f274a8cdd9ec0.png
[a3e4226c935c4242804cbdf26b571bc6.png]: https://img-blog.csdnimg.cn/a3e4226c935c4242804cbdf26b571bc6.png
[69778fb8780b412bb6cfd6cfe695ab2f.gif]: https://img-blog.csdnimg.cn/69778fb8780b412bb6cfd6cfe695ab2f.gif