Python 接口测试之接口请求方法封装

目录

引言

请求方法封装

执行测试

总结

请求方法封装

下面就直接附上代码：

 

__author__ = 'Leo'
 
import requests
import json
 
class ApiRequest(object):
    """
    请求方法
    """
    # 请求方法get
    def get_method(self,url,data=None,header=None):
 
        if header is not None:
            res = requests.get(url,params=data,headers=header)
        else:
            res = requests.get(url,params=data)
        return res.json()
 
    # 请求方法post
    def post_method(self,url,data=None,header=None):
        global res
        if header is not None:
            res = requests.post(url,json=data,headers=header)
        else:
            res = requests.post(url,json=data)
        if  str(res) == "<Response [200]>":
            return res.json()
        else:
            return res.text
 
    # 请求方法put
    def put_method(self,url,data=None,header=None):
        if header is not None:
            res = requests.put(url,json=data,headers=header)
        else:
            res = requests.delete(url, json=data)
        return res.json()
 
    # 请求方法delete
    def delete_method(self, url, data=None, header=None):
        if header is not None:
            res = requests.delete(url, json=data, headers=header)
        else:
            res = requests.delete(url, json=data)
        return res.json()
 
    # 主方法
    def run_method(self,method,url,data=None,header=None):
        if method == 'get' or method == 'GET':
            res = self.get_method(url,data,header)
        elif method == 'post' or method =='POST':
            res = self.post_method(url,data,header)
        elif method == 'put' or method == 'PUT':
            res = self.post_method(url,data,header)
        elif method == 'delete' or method == 'DELETE':
            res = self.post_method(url,data,header)
        else:
            res = "你的请求方式不正确！"
        # return res
        return json.dumps(res, ensure_ascii=False, indent=4, sort_keys=True,separators=(',', ':')

简单的解释一下相关知识点：

 

ensure_ascii：默认值True，如果dict内含有non-ASCII的字符，则会类似\uXXXX的显示数据，设置成False后，就能正常显示。
 
indent：应该是一个非负的整型，如果是0，或者为空，则一行显示数据，否则会换行且按照indent的数量显示前面的空白，这样打印出来的json数据也叫pretty-printed json。
 
separators：分隔符，实际上是(item_separator, dict_separator)的一个元组，默认的就是(‘,’,’:’)；这表示dictionary内keys之间用“,”隔开，而KEY和value之间用“：”隔开。
 
encoding：默认是UTF-8，设置json数据的编码方式。
 
sort_keys：将数据根据keys的值进行排序。

执行测试

方法已经封装好，既然接口数据有了，请求方法也有了，我们不妨测试一下之前封装好的代码，用实际行动来检验一下：

 

# get请求
if __name__ == '__main__':
    url = "http://httpbin.org/get"
    ir = ApiRequest()
    result = ir.run_method(url=url,method='get')
    print(result)

运行结果：

 

 

# post请求
url2 = "http://httpbin.org/post"
ir2 = ApiRequest()
data = {'id': '测试'}
result = ir2.run_method(url=url2,method='post',data=data)
print(result)

运行结果：

 

有些细心人可能有疑问，这里没有使用excel中的接口测试数据，下面我就演示一下，调用excel中维护的接口测试数据，先维护数据：

 

 

 from basic_method import get_excelData
    get_data = get_excelData.getData()
    print("获取是否运行key： ",get_data.get_is_run(1))
    print("获取接口url： ",get_data.get_url(1))
    print("获取接口请求方法： ",get_data.get_method(1))
    print("获取接口请求数据： ",get_data.get_data(1))
    url2 = get_data.get_url(1)
    method = get_data.get_method(1)
    data = get_data.get_data(1)
    ir2 = ApiRequest()
    result = ir2.run_method(url=url2,method=method,data=data)
    print(result)

运行结果：

 

这次就是调用过程是从excel拿的数据，并且通过封装好的方法请求模拟请求接口。证明此前封装好的代码是可以正常运行。

这里献上另一种请求方法：