开始免费使用
当前位置：首页 > Python > Python利用Template自动生成接口测试用例详解

Python利用Template自动生成接口测试用例详解

键盘旁的猫
键盘旁的猫
发布时间：2026-10-13| 6人浏览过
原创

本文介绍如何利用Python的string.Template模块，通过定义模板字符串动态生成PyUnit框架下的接口测试用例。示例展示了如何将测试参数化，自动构建包含请求头、数据及断言的测试方法，并结合BSTestRunner执行测试并生成报告。该方法适用于需要复用测试框架并批量生成用例的场景，代码基于Python 2.7.6环境编写。

在Python中，Template类允许我们将字符串格式固定下来，从而实现内容的重复利用。为了构建同一套可复用的测试框架，我们可以将测试用例部分进行参数化处理，进而将其应用到不同的项目中。

具体实现代码如下：

coding=utf-8
'''
作者：大石
功能：自动生成pyunit框架下的接口测试用例
环境：python2.7.6
用法：将用户给的参数处理成对应格式,然后调用模块类生成函数，并将参数传入即可
'''
 
from string import Template
#动态生成单个测试用例函数字符串
def singleMethodCreate(MethodList,interfaceNamePara):
  code=Template('''\n  def test_${testcase}(self):
    u"""${testcaseName}"""
    headers = $headers
    data = $data
    re = requests.$method(url='$url',headers=headers,data=data)
    status_code = re.status_code
    s = str(status_code)
    json = re.text
    logging.info('-'*5+'返回状态码是'+s+'-'*5)
    logging.info('-'*5+'返回结果集是'+json+'-'*5)
    assert status_code == 200
    assert json['status'] == 'ok'
''')
 
  string = code.substitute(testcase=MethodList["testcase"],testcaseName=MethodList["TestcaseName"],
               method=MethodList['method'],url=MethodList['url'],headers=MethodList['headers'],data=MethodList['data'],
               )
  return string
 
#拼接单个的测试用例函数字符串为完整字符串并传回主函数
#MethodParaList获取测试用例部分list
def methodCreate(MethodParaList,interfaceNamePara):
  string = ""
  for MethodPara in MethodParaList:
    string2=singleMethodCreate(MethodPara,interfaceNamePara)
    string=string+string2
  return string
 
#构造单个测试集
def singleTestsuitCreate(MethodList,parameters):
  code = Template('''suite.addTest(${className}("test_${testcase}"))''')
  string = code.substitute(testcase = MethodList["testcase"],className = parameters[0])
  return string
 
#添加测试集
def addtestsuit(MethodParaList,interfaceNamePara):
  string = ""
  for MethodPara in MethodParaList:
    string2 = singleTestsuitCreate(MethodPara,interfaceNamePara)
    string=string+string2
  return string
 
#生成测试用例类函数字符串
def modelClassCreate(parameters):
  modelCode = methodCreate(parameters[2],parameters[1])
  adtestsuit = addtestsuit(parameters[2],parameters)
  code = Template('''#coding: utf-8
"""
作者：大石
功能：待执行的接口测试用例
环境：python2.7.6
用法：通过框架自动触发调用
"""
import unittest,requests,datetime,sys,logging,BSTestRunner,time,os
from Log import Log
class ${className}(unittest.TestCase):
  u"""待测试接口：${interfaceName}"""
  def setUp(self):
    logging.info('-'*5+"begin test"+"-"*5)
  def tearDown(self):
    logging.info('-'*5+"end test"+'-'*5)
  ${model}
if __name__ == "__main__":
  #解决UnicodeDecodeError: 'ascii' codec can't decode byte 0xe5 in position 97: ordinal not in range(128)
  reload(sys)
  sys.setdefaultencoding('utf8')
  #构造测试集
  suite = unittest.TestSuite()
  ${testsuite}
  #定义date为日期，time为时间
  date=time.strftime("%Y%m%d")
  time1=time.strftime("%H%M%S")
  now=time.strftime("%Y-%m-%d-%H_%M_%S",time.localtime(time.time()))
  #创建路径
  path='F:/test/study/yaml/test_log/'+now+"/"
  #解决多次执行时报路径已存在的错误
  try:
    os.makedirs(path)
  except:
    if path!= None:
      logging.error(u'当前路径已经存在')
  filename=path+'Report.html'
  fp=file(filename,'wb')
  #日志记录
  Log.log()
  #执行测试
  runner =BSTestRunner.BSTestRunner(stream=fp,title=u'下单平台接口测试用例',description=u'接口用例列表：')
  runner.run(suite)
  fp.close()
''')
  fileStr = code.substitute(className=parameters[0],interfaceName=parameters[1],testsuite=adtestsuit,model=modelCode)
  f=open(parameters[0]+".py",'w')
  f.write(fileStr)
  f.close()

随后，测试用例部分的参数配置如下：

parameters=["Testcase_Orders",
        "/login",
        [
          {"TestcaseName":"测试登录","method":"post","url":"http://www.senbaba.cn/login","headers":{'content-type': 'application/json',
          'User-Agent':'Mozilla/5.0 (Windows NT 6.1; WOW64; Trident/7.0; rv:11.0) like Gecko',
          'Accept':'application/x-ms-application, image/jpeg, application/xaml+xml, image/gif, image/pjpeg, application/x-ms-xbap, */*',
          'Accept-Language':'zh-CN'},"data":{"uname":"187071484771","pwd":"123456"},
            "testcase":"login"},
 
          {"TestcaseName":"测试登录","method":"post","url":"http://www.senbaba.cn/login1","headers":{'content-type': 'application/json',
          'User-Agent':'Mozilla/5.0 (Windows NT 6.1; WOW64; Trident/7.0; rv:11.0) like Gecko',
          'Accept':'application/x-ms-application, image/jpeg, application/xaml+xml, image/gif, image/pjpeg, application/x-ms-xbap, */*',
          'Accept-Language':'zh-CN'},"data":{"uname":"187071484771","pwd":"123457"},
            "testcase":"login_failed"}
        ]
      ]

根据上述配置自动生成的测试用例代码如下：

#coding: utf-8
"""
作者：大石
功能：待执行的接口测试用例
环境：python2.7.6
用法：通过框架自动触发调用
"""
import unittest,requests,datetime,sys,logging,BSTestRunner,time,os
from Log import Log
class Testcase_Orders(unittest.TestCase):
  u"""待测试接口：/login"""
  def setUp(self):
    logging.info('-'*5+"begin test"+"-"*5)
 
  def tearDown(self):
    logging.info('-'*5+"end test"+'-'*5)
 
  
  def test_login(self):
    u"""测试登录"""
    headers = {'Accept-Language': 'zh-CN', 'content-type': 'application/json', 'Accept': 'application/x-ms-application, image/jpeg, application/xaml+xml, image/gif, image/pjpeg, application/x-ms-xbap, */*', 'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64; Trident/7.0; rv:11.0) like Gecko'}
    data = {'uname': '187071484771', 'pwd': '123456'}
    re = requests.post(url='http://www.senbaba.cn/login',headers=headers,data=data)
    status_code = re.status_code
    s = str(status_code)
    json = re.text
    logging.info('-'*5+'返回状态码是'+s+'-'*5)
    logging.info('-'*5+'返回结果集是'+json+'-'*5)
    assert status_code == 200
    assert json['status'] == 'ok'
 
  def test_login_failed(self):
    u"""测试登录"""
    headers = {'Accept-Language': 'zh-CN', 'content-type': 'application/json', 'Accept': 'application/x-ms-application, image/jpeg, application/xaml+xml, image/gif, image/pjpeg, application/x-ms-xbap, */*', 'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64; Trident/7.0; rv:11.0) like Gecko'}
    data = {'uname': '187071484771', 'pwd': '123457'}
    re = requests.post(url='http://www.senbaba.cn/login1',headers=headers,data=data)
    status_code = re.status_code
    s = str(status_code)
    json = re.text
    logging.info('-'*5+'返回状态码是'+s+'-'*5)
    logging.info('-'*5+'返回结果集是'+json+'-'*5)
    assert status_code == 200
    assert json['status'] == 'ok'
 
 
if __name__ == "__main__":
  #解决UnicodeDecodeError: 'ascii' codec can't decode byte 0xe5 in position 97: ordinal not in range(128)
  reload(sys)
  sys.setdefaultencoding('utf8')
  #构造测试集
  suite = unittest.TestSuite()
  
  suite.addTest(Testcase_Orders("test_login"))
 
  suite.addTest(Testcase_Orders("test_login_failed"))
 
  #定义date为日期，time为时间
  date=time.strftime("%Y%m%d")
  time1=time.strftime("%H%M%S")
  now=time.strftime("%Y-%m-%d-%H_%M_%S",time.localtime(time.time()))
  #创建路径
  path='F:/test/study/yaml/test_log/'+now+"/"
  #解决多次执行时报路径已存在的错误
  try:
    os.makedirs(path)
  except:
    if path!= None:
      logging.error(u'当前路径已经存在')
  filename=path+'Report.html'
  fp=file(filename,'wb')
  #日志记录
  Log.log()
  #执行测试
  runner =BSTestRunner.BSTestRunner(stream=fp,title=u'下单平台接口测试用例',description=u'接口用例列表：')
  runner.run(suite)
  fp.close()

20171019添加测试集的一个简单方法：

#添加测试集
def addtestsuit(parameters):
  string = ""
  temp = Template('''\n  suite.addTest(${className}("test_${testcase}"))
''')
  l = len(parameters[2])
  for i in range(0,l):
    testcase1 = parameters[2][i]['testcase']
    string2 = temp.substitute(className = parameters[0],testcase = testcase1)
    string=string+string2
    print string
  return string

以上这篇关于Python自动生成接口测试的示例讲解就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持本站。

相关文章

相关标签: Python 字符串处理

免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。

作者最新文章

全部文章

Python 相关产品和应用

全部下载
PythonSelenium
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

Python机器学习
scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Python代码质量
Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Python代码质量
Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

Pythonpytest
pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

PythonFastAPI
FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架，支持类型提示和自动接口文档。

PythonFlask
Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

PythonDjango
Django

Django

6.1.2 · Windows

成熟的 Python Web 框架，适合构建内容站点和业务系统。

PythonSpyder
Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

PythonSciPy
SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。