如何在自动化测试中使用MitmProxy获取数据返回？

背景介绍

当我们在接口或UI自动化项目中，常常会出现这种现象——明明是正常请求，却无法获取到想要的数据返回。

比如：

场景A：页面是动态数据，第一次进入页面获取到的数据，和下次进入页面获取到的数据完全不一样。

场景B：页面数据有展示限制，页面某些模块或整个页面数据仅展示3次，且通过服务端返回控制。

这时我们要怎么办？

如果是手工测试时，我们大可以直接使用Charles、Fiddler进行模拟，毕竟代理软件基本能解决大部分测试需求。

那如果是在自动化项目中呢？如果还是使用代理软件——其实也不是完全不行，但我们还有更佳的实践方式。

也就是使用MitmProxy作为正向代理拦截请求和返回数据。

如何使用MitmProxy作为正向代理

实现Maplocal

MitmProxy是一个三方库，我们在程序中可以把它当成Charles来使用。

这里以实现Maplocal功能进行说明：

（1）Maplocal功能，简单理解就是当针对特定请求，返回特定数据。

比如请求"www.baidu.com"，正常应该返回：

1. {

2. "post": {

3. "title": "百度一下",

4. },

5. "type": 1

6. }

AI写代码python运行

但我们修改返回数据，只要有"www.baidu.com"请求过来，就让它自动返回：

1. {

2. "post": {

3. "title": "谷歌一下啦啦啦",

4. },

5. "type": 1

6. }

AI写代码python运行

这就是一个最基础的maplocal案例。

（2）在MitmProxy里如何实现Maplocal功能呢？

这里以get请求实现进行说明：

get.json 配置文件，在这里记录所有需要拦截的get请求：

1. {

2. "https:www.baidu.com":[

3. {

4. "jsonPath":"../data/maplocal/first_tab.json"

5. }

6. ]

7. }

AI写代码python运行

MitmProxy脚本中：

1. from mitmproxy import http

2. from mitmproxy import ctx

3. class InterceptRequests:

4. def __init__(self):

5. def request(self, flow: http.HTTPFlow) -> None:

6. """

7. 使用request事件实现map local

8. :param flow:

9. :return:

10. """

11. # GET 请求处理

12. if method == "GET":

13. pure_url = get_url_host(interceptUrl)

14. url_data = get_json.get(pure_url)

15. if url_data:

16. interce_params = get_url_params(interceptUrl)

17. params = get_json_value(get_path, pure_url)

18. for para in params:

19. if para.items() <= interce_params.items():

20. jsonpath = para.get("jsonPath")

21. jsonpath_abs = path(jsonpath)

22. self.file_maplocal(flow, jsonpath_abs)

23. addons = [

24. InterceptRequests()

25. ]

在执行的脚本中（如本地shell 或 其他执行的py脚本里）：

1. # kill mitmproxy

2. lsof -n -i:8999 | grep LISTEN | awk '{print $2}' | xargs kill

3. # start mitmproxy

4. mitmdump -p 8999 -s mitmHandler.py

AI写代码python运行

启动脚本后，连接MitmProxy的设备的请求全部都会被脚本监控。

如果出现"www.baidu.com"的请求，就会到get.json里找到对应的json，自动返回给设备。

这样，就实现了一个最简单的maplocal。

当然基于目前的MitmProxy脚本，我们还可以进行一些功能扩展。

如针对接口类型get或post请求，针对请求参数类型json或x-www-form-urlencoded，都可以进行更多延伸扩展。

感谢每一个认真阅读我文章的人！！！

作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。

软件测试面试文档

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

视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取。