请求重试功能的单元测试

时间:2021-07-07 15:30:11

标签: python unit-testing

我有一个请求重试功能,每个会话发送 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

我想为此函数编写单元测试。如果您有任何建议,我将不胜感激。

0 个答案:

没有答案