接口测试框架搭建#1 -- 创建get post请求类

学习mushishi第一堂课

相对应url去除

 

相对课程代码，加了一个对请求类型的判空处理，对于请求数据的json格式转换，post请求的header添加（表明请求为json格式）。

处理经验：

需注意请求数据的格式转换（字典转json），和header添加

 

import requests
import json

class send_method:
    header = {
            'Content-Type': 'application/json; charset=utf-8',
        }

    def __init__(self,url,method,data = None):
        self.res = self.run_main(url,method,data)

    def send_post(self,url,data,header):
        res = requests.post(url=url,data= data,headers=header).json()
        return json.dumps(res,indent=2,sort_keys=True)

    def send_get(self,url,data=None):
        res = requests.get(url=url,data=data).json()
        return json.dumps(res,indent=2,sort_keys=True)

    def run_main(self,url,method,data):
        data = json.dumps(data)

        if method == 'GET':
            result = self.send_get(url,data)
        elif method == 'POST':
            result = self.send_post(url,data,self.header)

        else:
            result = 0
            print("method is wrong!! it is ",method)
        return result



if __name__ == '__main__':
    url = '****'
    url_1 = '****'

    data = {
        'deviceType': '0',
        'language': 'zh_CN_%23Hans'
    }
    data_1 = {"country": "CN"}

    res = send_method(url,'GET',data)

    print(res.res)