python在rest接口测试中的应用

2023-01-31 05:01:59 python 接口 测试中

    近期一直在学习python,这两天正好做rest接口的接口测试,就尝试着用Python写一个测试rest接口的通用类,省去用工具的繁琐,也显得比较高大上一些。

    Http rest接口最常用的两种请求方式,POST和GET两种方法,请求都是类似下面例子的url。

    例子:http://127.0.0.1:8080/rest/login?oper_no=marry。

    需要用到python的几个内置模块,urllib、urllib2、re、os、time。urllib和urllib2是网络模块,re是正则表达式模块,os是系统模块,用它来执行一些系统命令,time模块用它来取当前系统时间。

    代码如下,都加了对应的注释:

import urllib,urllib2
import re,os
import time
class RestTest(object):
   def __init__(self):
       '''
       python的魔术方法,类似java的构造函数,该类在实例化的时候会调用这个函数
       '''
       self.__ResPath__='/Users/niuhanyang/Desktop/TestRes/'#测试结果路径
   def Rest(self,method,url,restName,**param):
       '''
       :param method: 请求方法,GET/POST,入参类型是str
       :param url: 请求url地址,入参类型是str
       :param restName: 接口名称,入参,类型是str
       :param param: 请求参数是一个可选参数,例如oper_id=11,会把传入的值都放到一个dict中,如{'oper_id':11}
       :return:如果是登录接口的话,会返回jessionid
       '''
       jsessionStr=r'"jsessionid":"(.*?)"'#正则表达式,抓出jsessionid
       new_jsessionStr = re.compile(jsessionStr)#编译正则
       if method=='GET':
           data = self.UrlParam(param)
           new_url=url+'?'+data#生成新的url
           result = urllib.urlopen(new_url).read()#保存返回结果
           print  restName+'\n'+result
           self.WriteRes(result,restName)#保存测试结果和返回报文
           return ''.join(new_jsessionStr.findall(result))
       if method=='POST':
           data = self.UrlParam(param)#参数转换
           req = urllib2.Request(url,data)#生成一个request对象
           response = urllib2.urlopen(req)#发送request请求
           result = response.read()#保存结果
           print  restName+'\n'+result
           self.WriteRes(result,restName)#保存测试结果和返回报文
           return ''.join(new_jsessionStr.findall(result))#把jsessionid转化成字符串返回,findall方法的返回值是一个list
   def UrlParam(self,param):
       '''
       参数转换,入参是一个dict,把这个list变为xx=xx这样
       示例:
       入参:{'oper_id':11}
       出参:oper_id=11    
       '''
       return urllib.urlencode(param)
   def WriteRes(self,result,restName):
       '''
       :param result: 返回报文
       :param restName: 接口名,用来做报文结果的文件名
       :return: 无返回值
       '''
       res= result.find('success')#返回结果是否成功
       fw_flag = open('%sTestRes.txt'%self.__ResPath__,'a')#追加方式打开测试结果文件
       if res>0:
          fw_flag.write('%s : pass\n'%restName)#写入测试结果
       else:
           fw_flag.write('%s : fail\n'%restName)#写入测试结果
       fw_flag.close()
       fw_response = open('%s%s.txt'%(self.__ResPath__,restName),'w')#打开返回报文文件,前面一个%s是路径,后面的%s是以接口名命名的文件名
       fw_response.write(result)#写入返回报文
       fw_response.close()
   def BakRes(self):
       '''
       该函数的作用是用来备份测试结果
       '''
       now_time= time.strftime("%m-%d_%H:%M", time.localtime())
       os.system('cd %s&&mkdir %s&&mv *.txt ./%s'%(self.__ResPath__,now_time,now_time))
   def RmRes(self):
       '''
       清空测试目录下的所有测试结果
       '''
       os.system('cd %s&&rm -rf *'%self.__ResPath__)
if __name__ == '__main__':
   Test = RestTest()#创建一个实例
   session_id = Test.Rest('POST','http://127.0.0.1:8080/nn_WEB/rest/oper/login','login',login_flag=0,oper_no='HEHH',oper_pwd=1)#调用Rest方法测试登录接口,获取session_id
Test.Rest('POST','http://127.0.0.1:8080/nn_web/rest/oper/getOperInfo','getOperInfo',jsessionid=session_id)#调用获取工号信息接口,传入登录获取到的seession_id

接下来继续研究python在webservice接口测试中的应用,有哪些不足的地方大家提出来哦。

相关文章