python接口自动化读取json,yaml配置文件+封装requests+unittest+HTMLRunner实现全自动化

时间:2021-10-10 06:55:37
# 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 
版权声明:本文为博主原创文章,转载请附上博文链接!