python 接口自动化测试框架-使用类封装测试脚本

一、使用类对post请求和get请求进行封装
1、首先我们来了解一下什么是类及类的作用：

class run_main: # 定义一个class类
	pass
run = run_main() # 创建一个run_main实例
run.post = "POST"  # 为实例添加 POST 属性

2、下面我们来定义实例方法

import requests

class RunMain:

	def __init__(self,url,data,headers,method):
         self.response = self.run_main(url,data,headers,method)

    def send_post(self,url,data,headers):
        res = requests.post(url=url,data=data,headers=headers).json()

    def send_get(self,url,data,headers):
        res = requests.get(url=url,data=data,headers=headers).json()

    def run_main(self,url,data,headers,method):
        res = None
        if method.upper() == 'POST':
            res = self.send_post(url,data,headers)
        elif method.upper() == 'GET':
            res = self.send_get(url,data,headers)
        return res

if __name__ == '__main__':
    run = RunMain()
    url = 'http://www.oktest.org.cn/siteApp/command/ecGzSubSearch?fid=t_xmosta&siteId=2808&search=1'
    data = {
        'searchKey': "测试",
        'x': '30',
        'y': '8'
    }
    headers = {
        'Content-Type': 'application/json'
    }
    ret = run.run_main(url=url,data=data,headers=None,method='POST')
    print(ret)