今天给大家分享一个简单的Python脚本，使用python进行http接口的自动化测试，脚本很简单，逻辑是：读取excel写好的测试用例，然后根据excel中的用例内容进行调用，判断预期结果中的返回值和返回报文中的值是否一致，如果不一致则根据用例标题把bug提交到bug管理系统，这里使用的bug管理系统是bugfree。最后统计测试结果：总共执行多少条用例、通过多少条用例，失败多少条用例，邮件标题加上当前时间，将测试结果发送到指定邮箱。
实现步骤：
1、读取excel，保存测试用例中的内容，这里需要定义一个读取excel的函数readExcel（）；
2、根据excel中的请求url和参数拼接请求报文，调用接口，并保存返回报文，这里需要定义一个将请求数据转换为字典的函数param_To_Dic（）；
3、读取返回报文，和预期结果对比，不一致的往bugfree数据库中写入一条bug，并且把请求报文、返回报文和测试结果写到测试用例的excel中，这里需要定义一个比对预期结果和返回结果的函数contrastRes（），一个往bugfree提交bug的函数writeBug（），一个将测试结果写入excel的函数copy_excel（），还需定义一个接口请求的函数interfaceTest（）。
4、统计测试结果，发送测试邮件。需定义一个send_email（）的函数。
http接口最常用的两种请求方式，POST和GET两种方法，这篇博客分享的就是最简单常用的url请求。例如：http://192.168.21.129/bugfree/index.php/info/edit?type=bug&action=opened&product_id=1
需要用的到几个模块有：requests、xlrd（读取excel）、xlutils（写excel）、pymysql（连接数据库）、yagmail（发送邮件）这五个模块都是第三方模块，需要自己单独安装。
首先在excel中写好用例，需要有的字段 项目、用例id、接口名称、用例描述、请求方式、请求url、请求数据（多个的参数话用&分号隔开）、预期结果、请求报文、返回报文、测试人员、测试结果
首先我们将项目框架建起来，如下图所示：
cases目录中是放excel测试用例，conf文件夹是配置文件，lib目录是存放主逻辑的，logs目录里是日志文件，bin目录下是启动文件，运行start.py即可运行该项目
1.excel中测试用例书写格式如下图所示：
2.我们编写lib中的主逻辑，将上面分析出的所需要的函数写在common.py中，代码如下：
class OpCase(object): #定义获取并操作excel的类 def get_case(self,file_path): #获取用例的函数 cases = [] #定义一个list，存放所有的case if file_path.endswith(‘.xls‘) or file_path.endswith(‘.xlsx‘): #如果文件类型正确 try: book = xlrd.open_workbook(file_path) #打开excel sheet = book.sheet_by_index(0) for i in range(1,sheet.nrows): #循环excel row_data = sheet.row_values(i) #将excel中每行数据存到row_data中 cases.append(row_data[4:8]) #将excel中的url,method,data,预期结果，返回报文都加入到cases列表中 atp_log.info(‘共读取%s条用例‘%(len(cases))) self.file_path = file_path except Exception as e: atp_log.error(‘【%s】用例获取失败，错误信息：%s‘%(file_path,e)) else: atp_log.error(‘用例文件不合法的，%s‘%file_path) return cases def my_request(self,url,method,data): #发起接口请求 method = method.upper() #将请求方式全部转换为大写 data = self.dataToDict(data) #将参数转为字典 try : if method==‘POST‘: res = requests.post(url,data).text elif method==‘GET‘: res = requests.get(url,params=data).text else: atp_log.warning(‘该请求方式暂不支持。。‘) res = ‘该请求方式暂不支持。。‘ except Exception as e: msg = ‘【%s】接口调用失败，%s‘%(url,e) atp_log.error(msg) res = msg return res def check_res(self,res,check): #校验实际结果与预期结果 res = res.replace(‘": "‘,‘=‘).replace(‘": ‘,‘=‘) #把实际返回结果转为excel中预期结果的格式形式 for c in check.split(‘,‘): #将预期结果按逗号分隔 if c not in res: #循环预期结果，如果预期结果中子元素不在实际返回结果中，那么校验失败 atp_log.info(‘结果校验失败，预期结果：【%s】，实际结果【%s】‘%(c,res)) return ‘失败‘ return ‘成功‘ def write_excel(self,cases_res): #将返回报文和测试结果写入excel # [ [‘dsfd‘,"通过"] ,[‘sdfsdf‘,‘失败‘] ] book = xlrd.open_workbook(self.file_path) new_book = copy.copy(book) sheet = new_book.get_sheet(0) row = 1 for case_case in cases_res: sheet.write(row,8,case_case[0]) #写第8列 sheet.write(row,9,case_case[1]) #写第9列 row+=1 new_book.save(self.file_path.replace(‘xlsx‘,‘xls‘)) def dataToDict(self,data): #把数据转成字典 res = {} data = data.split(‘,‘) for d in data: #a= k,v = d.split(‘=‘) res[k]=v return res
3.写日志的模块，代码如下：
可以引用一个简单的log模块，只需要pip install nnlog即可使用，详细教程参照牛牛博客：www:nnzhp.cn/archives/646
class MyLogger(): def __init__(self,file_name,level=‘info‘,backCount=5,when=‘D‘): logger = logging.getLogger() # 先实例化一个logger对象，先创建一个办公室 logger.setLevel(self.get_level(level)) # 设置日志的级别的人 cl = logging.StreamHandler() # 负责往控制台输出的人 bl = handlers.TimedRotatingFileHandler(filename=file_name, when=when, interval=1, backupCount=backCount, encoding=‘utf-8‘) fmt = logging.Formatter(‘%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s‘) cl.setFormatter(fmt) # 设置控制台输出的日志格式 bl.setFormatter(fmt) # 设置文件里面写入的日志格式 logger.addHandler(cl) logger.addHandler(bl) self.logger = logger def get_level(self,str): level = { ‘debug‘:logging.DEBUG, ‘info‘:logging.INFO, ‘warn‘:logging.WARNING, ‘error‘:logging.ERROR } str = str.lower() return level.get(str) path = os.path.join(setting.LOG_PATH,setting.LOG_NAME) #拼好日志的绝对路径 atp_log = MyLogger(path,setting.LEVEL).logger #直接在这里实例化，用的时候就不用再实例化了
4.发送邮件函数，代码如下：
def sendmail(title,content,attrs=None): m = yagmail.SMTP(host=setting.MAIL_HOST,user=setting.MAIL_USER ,password=setting.MAIL_PASSWORD,smtp_ssl=True ) m.send(to=setting.TO,subject=title, contents=content, attachments=attrs) atp_log.info(‘发送邮件完成‘)
5.conf中配置文件进行参数的配置，详细如下：
import os BASE_PATH = os.path.dirname( os.path.dirname(os.path.abspath(__file__)) ) MAIL_HOST=‘smtp.qq.com‘ MAIL_USER=‘729111761@qq.com‘ MAIL_PASSWORD = ‘xxxxxxxxxxx‘ TO = [ ‘1379440856@qq.com‘, ] LEVEL = ‘debug‘ #日志级别 LOG_PATH = os.path.join(BASE_PATH,‘logs‘) #存放日志的路径 CASE_PATH = os.path.join(BASE_PATH,‘cases‘) #存放日志的路径 LOG_NAME=‘atp.log‘ #日志的文件名
6.最后就是bin目录下的运行文件start.py，代码如下：
import os,sys BASE_PATH = os.path.dirname( os.path.dirname(os.path.abspath(__file__)) ) #获取到ATP目录的path sys.path.insert(0,BASE_PATH)#加入到环境变量中
from lib.common import OpCase from lib.send_mail import sendmail from conf import setting class CaseRun(object): def find_cases(self): op = OpCase() for f in os.listdir(setting.CASE_PATH):#从CASE_PATH目录中读取excel,每次循环的时候读一个excel abs_path = os.path.join(setting.CASE_PATH,f) #将读取的文件与目录拼接 case_list = op.get_case(abs_path) #获取读取excel中的用例 res_list = [] #定义一个空list，用来存实际返回报文和测试结果 pass_count,fail_count = 0,0 for case in case_list:#循环每个excel里面所有用例 url,method,req_data,check = case res = op.my_request(url,method,req_data) #调用完接口返回的结果 status = op.check_res(res,check) res_list.append([res,status]) if status==‘成功‘: pass_count+=1 else: fail_count+=1 op.write_excel(res_list) #将返回报文和是否通过的测试结果写入excel msg = ‘‘‘ xx你好： 本次共运行%s条用例，通过%s条，失败%s条。 ‘‘‘%(len(res_list),pass_count,fail_count) sendmail(‘测试用例运行结果‘,content=msg,attrs=abs_path) CaseRun().find_cases()
7.另外readme里面可以简单介绍下该框架的运行环境，所依赖的模块，简单的思路和顺序。
自此，基于数据驱动的接口自动化测试框架就已搭建成功，大家可以自己动手试试了。。。。
python学习笔记（三十）python实现http接口自动化测试（完善版）
原文：https://www.cnblogs.com/mululu/p/9146749.html