# coding=utf-8 import json import requests class TestApi(object): """ /* @param: @session ,@cookies the request can be divided into session request and cookie request according to user's own choice however,url and header is must ,other parameters are given by user to make it is None or not */ """ def get(self,url,param,header,cookie=None,session=None,**kwargs): if session: return session.request("GET",url,param,headers=header,**kwargs) elif cookie: return requests.get(url,params=param,headers=header,cookies=cookie,**kwargs) """ /* @param: @session ,@cookies 传入的是dict类型 python object 对象 header is form data: application/x-www-urlencoded transfer data to data directly ,finally requests's submit will be like 'aa=dd&bb=ff' formation header is json :application/json due to the data can be 'str','dict'and tuple and so on ,so when we choose data and data is given by dict,we must transfer it to json str，but when is json type str ,we must must transfer python object dict to json str with json.dumps(), finally the request submit data format is str like: 'aa=dd&bb=ff',but when choose json the submit will become like {'a': 'cc' ,'b': 'dd'} , data and json cant not be used in the requests at the same time */ """ def post_data(self,url,type,data,header,cookie=None,session=None,**kwargs): if cookie: if type is "data": return requests.post(url,data=data,headers=header,cookies=cookie,**kwargs) elif type is "json": return requests.post(url,data=json.dumps(data),headers=header,cookies=cookie,**kwargs) elif session: if type is "data": return session.request("POST",url,data=data,headers=header,cookies=cookie,**kwargs) elif type is "json": return session.request("POST",url,data=json.dumps(data),headers=header,cookies=cookie,**kwargs) """ /* @:param：@json object json的value为传入的json对象 请求header默认：ContentType: application/json */ """ def post_json(self,url,header,json,cookie=None,session=None,**kwargs): if cookie: return requests.post(url,headers=header,json=json,cookies=cookie,**kwargs) elif session: return session.request("POST",url,headers=header,json=json,**kwargs) """ /* @:param： @url,@data,@**kwargs Tip: header you need to according to your api to be given in **kwargs position */ """ def put(self,url,data,cookie=None,session=None,**kwargs): if cookie: return requests.put(url,data,cookies=cookie,**kwargs) elif session: return session.request("PUT",url,data,**kwargs) """ /* @:param： @url,@data,@**kwargs Tip: header you need to according to your api to given in **kwargs position */ """ def delete(self,url,data,cookie=None,session=None,**kwargs): if cookie: return requests.delete(url,data,cookies=cookie,**kwargs) elif session: return session.request("DELETE",url,data,**kwargs) # coding=utf-8 from ruamel import yaml from API.apitest import * """ /*@param: python version 3.7 第一步制造配置文件yaml或者json都可以保存请求报文接口参数的： 写入方法很简单见：Jsread.py的Yml，Js类的write()方法 */ """ class Yml(object): def __init__(self, yml_path): self.yml_path = yml_path def read(self): with open(self.yml_path, 'r', encoding='utf-8')as f: data = yaml.load(f,Loader=yaml.Loader) return data class EnvParameter(object): def __init__(self, con_path): defaults = {"url": None, "header": None, "data": None, "method": None, "param": None, "type": None, "json": None} self.cookies = None self.session = None self.con_path = con_path dict = Yml(self.con_path).read() defaults.update(dict) self.url = defaults["url"] self.header = defaults["header"] self.method = defaults["method"] self.data = defaults["data"] self.param = defaults["param"] self.json = defaults["json"] self.type = defaults["type"] class TestSend(EnvParameter): def __init__(self,config_path,cookie1=None,session1=None): # EnvParameter.__init__(self,conpath=None ,url=None,method=None,header=None,type=None,data=None,param=None,json=None,cookies=None,session=None) EnvParameter.__init__(self,config_path) self.session=session1 self.cookie1=cookie1 # print(self.param,self.type) #测试下类继承效果 def send(self): if self.method.upper()=="GET": rep=TestApi().get(self.url,self.param,self.header,cookie=self.cookie1,session=self.session) return rep elif self.method.upper()=="POST": rep=TestApi().post_data(self.url,self.type,self.data,self.header,cookie=self.cookie1,session=self.session1) return rep elif self.method.upper()=="PUT": rep=TestApi().put(self.url,self.data,cookie=self.cookie1,session=self.session1) return rep elif self.method.upper()=="DELETE": rep=TestApi().delete(self.url,self.data,cookie=self.cookie1,session=self.session1) return rep # if __name__ == "__main__": # TestSend('./conf.yaml') import unittest import requests from API.testyaml import * class Interface(unittest.TestCase): @classmethod def setUpClass(cls): global session s = requests.session() requests.get(url="http://www.baidu.com") session =s print("---------------开始测试所有接口--------------") @classmethod def tearDownClass(cls): """清除cookie""" session.cookies.clear() #也可以这样写 session.cookies=None print("---------------加载所有接口结束销毁cookie--------------") def test_001(self): response=TestSend('./conf.yaml',session1=session).send() print(response.status_code) if __name__ =="__main__": unittest.main() 详情也可以见我的csdn地址 --------------------- 作者：流浪的python 来源：CSDN 原文：https://blog.csdn.net/chen498858336/article/details/86619178 版权声明：本文为博主原创文章，转载请附上博文链接！