requests-mock：轻松模拟HTTP请求的利器

一、 简介

requests-mock一个python库，用于单元测试中模拟HTTP请求的响应，它可以进行来模拟接口的各种场景。

安装：

?

1	pip install requests-mock

　　

二、 使用方法

模拟post请求

 

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21

import requests   import requests_mock     def test_01():     with requests_mock.Mocker() as m:         # 模拟post请求内容，返回的json格式，返回码为200         m.post('https://www.wxwxwx.com', json={'name': 'wx'}, status_code=200)           # 根据模拟的请求进行通过requests进行发送模拟信息，查看返回结果内容         r = requests.post('https://www.wxwxwx.com', json={'name': 'wx'})           print(r.text)           assert r.status_code == 200           assert r.json()['name'] == 'wx'     test_01()

　　

模拟动态post请求

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

import requests   import requests_mock     def test_01():     with requests_mock.Mocker() as m:         # 模拟post请求内容，返回的json格式，返回码为200         def echo_response(request, context):             data = request.json()             return {'json': data, 'status_code': 200}           m.post('https://www.wxwxwx.com', json=echo_response)           data = {'name': 'wx'}         r = requests.post('https://www.wxwxwx.com', json=data)         print(r.text)         assert r.status_code == 200         assert r.json()['json']['name'] == 'wx'     test_01()

　　

模拟请求错误异常

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20

import pytest import requests import requests_mock from requests.exceptions import ConnectionError, Timeout     def test_01():     with requests_mock.Mocker() as m:         # 模拟请求超时处理         m.get('https://www.wxwxwx.com', exc=Timeout)           # 通过pytest.raises进行捕捉异常，如果存在异常，则判断为pass         with pytest.raises(Timeout):             requests.get('https://www.wxwxwx.com', timeout=1)           # 模拟服务器错误         m.get('https://www.wxwxwx.com/api', exc=ConnectionError)         # 发送请求并断言是否抛出了预期的异常         with pytest.raises(ConnectionError):             requests.get('https://www.wxwxwx.com/api')