我有一个请求重试功能,每个会话发送 3 次 post 请求,如果 post 请求在 3 次尝试后没有返回响应,那么它将引发错误。功能如下:
def request_retry_session():
session = requests.Session()
retry = Retry(total=3, backoff_factor=0.1,
method_whitelist=frozenset(['POST']))
adapter = HTTPAdapter(max_retries=retry)
session.mount('http://', adapter)
try:
response = session.post(
url=url,
data=json.dumps(data),
headers={"Content-Type": "application/json"},
verify=False,
)
response_data = response.json()
logger.info(f"{ref}processing time cost for {api_name}: {time_cost}s")
except Exception as e:
raise e
我想为此函数编写单元测试。如果您有任何建议,我将不胜感激。