python接口测试

# -*- coding:utf-8 -*-
import requests,json

class RunMain():
    # def __init__(self,url,method,data=None,headers=None):
    #     self.res = self.run_main(url,method,data,headers)

    '''post接口封装，传入url,头部信息和主体信息，没有的时候默认为空'''
    def send_post(url,data=None,headers=None):
        # #返回json格式
        # res = requests.post(url=url,json=data,headers=headers).json()
        # return json.dumps(res,indent=3,sort_keys=True)
        #返回text格式
        res = requests.post(url=url, json=data, headers=headers)
        return res.json()

    #get接口封装，传入url,头部信息和主体信息，没有的时候默认为空
    def send_get(url,data=None,headers=None):
        # res = requests.get(url=url,headers=headers,json=data).json()
        # return json.dumps(res,indent=3,sort_keys=True)
        res = requests.get(url=url, json=data, headers=headers)
        return res.json()

    def run_main(self,url,method,data=None,headers=None):
        #如果method等于post，则是post接口，否则是get接口
        res = None
        if method == "POST":
            res = RunMain.send_get(url,data,headers)
        else:
            res = RunMain.send_post(url,data,headers)
        return res
if __name__ == '__main__':
    run = RunMain
    url = ''
    headers = {
        'token': '34F0CE94887E4B88954C30D63CC19700',
        'timestamp': '1',
        'Content-Type': 'application/json',
        'password': '123',
        'account': 't4'
    }
    data = {
        "city": "35",
        "packageCode": "P000001"
    }
    run = RunMain(url,'post',data,headers)
    print(run.res)
# print(run.run_main(url,data,'POST',headers))