python基于requests的接口测试框架，建立全局的等待时间

Posted 2021-05-21 CrissChan

在使用python的requests库接口测试过程中，很多时候为了让测试代码可以顺利的全部执行，我们都会设置一些等待时间（也叫思考时间），一般的做法如下：

requests.get('https://blog.csdn.net/crisschan', timeout=0.001)

这样的代码会出现在全部的请求中，其实我们可以创建一个等待时间的类，通过全局设置完成这一步。

from requests.adapters import HTTPAdapter

DEFAULT_TIMEOUT = 5 # seconds

class TimeoutHTTPAdapter(HTTPAdapter):
    def __init__(self, *args, **kwargs):
        self.timeout = DEFAULT_TIMEOUT
        if "timeout" in kwargs:
            self.timeout = kwargs["timeout"]
            del kwargs["timeout"]
        super().__init__(*args, **kwargs)

    def send(self, request, **kwargs):
        timeout = kwargs.get("timeout")
        if timeout is None:
            kwargs["timeout"] = self.timeout
        return super().send(request, **kwargs)

上面的使用方法如下：

import requests

http = requests.Session()

# 把配置复制给http和https请求
adapter = TimeoutHTTPAdapter(timeout=2.5)
http.mount("https://", adapter)
http.mount("http://", adapter)

# 使用全局等待时间2.5秒
response = http.get("https://blog.csdn.net/crisschan")

# 不用全局等待时间，设置请求自己的等待时间10秒
response = http.get("https://blog.csdn.net/crisschan", timeout=10)