一盏小灯 接口测试框架整理:python unittest + pytest + allure

Jerry li · December 28, 2018 · Last by mingmingjiu replied at December 19, 2022 · 4785 hits

api_test_demo

github 地址: https://github.com/jerrylizilong/api_test_demo

简介

整理如何用 unittest 编写接口测试用例,和使用 pytest+allure 生成报表。
每个接口只需要写上接口地址、加密参数列表,即可快速编写对应的测试用例。

如何运行

  • 下载源码。
  • 安装 requirements
  • 运行 run.bat/ run.sh 文件开始执行。
  • 报告生成: 每次运行会生成新的报告目录, 打开 api_demo/allurereport/时间日期/html 下的 index.html 文件查看 allure 报告。(需要安装 allure )

结构介绍

1. 通用接口功能

api_demo.api_manage.api_base 通用方法: 如何为登录接口进行签名、拼接参数和 url,及发送。

1.1 : 签名加密方法

def getOsign(user,osign_list):   # 签名方法。  这里举例是进行 MD5 加密, 具体需要替换为实际的签名方法。
    user['osign'] = util.getOsign(user, osign_list)
    return user

 def getOsign(user,osignList):
    paraPand = ''
    for para in osignList:
        paraPand += str(user[para])
    return md5(paraPand)

1.2 :接口参数拼接和发送

def api_send(user,osign_list,url,need_osign = True,isMock=False):     # 具体的接口url 拼接、参数生成和发送方法
    if need_osign:
        getOsign(user,osign_list)
    return send(user, url,isMock=isMock)

# http post 发送
 def send(user,url,isMock=False,host=host,needJson=True):
    body = json.dumps(user, default=lambda user: user.__dict__, sort_keys=True, skipkeys=True)
    body = eval(body)
    list = util.dict_2_str(body)
    url = host+url+ '?' + list
    print(url)
    if isMock:
        return mockData.mockData(url)
    else:
        response, content = util.sendRequest(url)
        print(content)
        if needJson:
            content = json.loads(content)
        return content

2. 模拟接口功能实现

api_demo.mock 示例如何为登录接口生成 mock 数据。

使用说明: 如果接口已开发完成,可以使用,可忽略这步。

def login(self,query):
    data = {}
    osign = query['osign'][0]
    userName = query['userName'][0]
    password = query['password'][0]
    verifyCode = query['verifyCode'][0]
    if osign != util.md5(userName +password+verifyCode) or verifyCode !='123456':      # 签名不匹配,或者验证码错误
        data['code']=4010
        data['msg']='invalid request!'
    elif userName !='correctuser' or password !='correctpassword' :                # 用户名或密码错误
        data['code']=500
        data['msg']='username or password is wrong ,please try again!'
    else:                                                                               # 正常登录
        data['code']=200
        data['msg']='success!'
        data['loginTime'] = self.loginTime
    return data

3. 测试用例(可根据 test_login.py 作为模板,编写你的测试用例)

api_demo.test 示例如何编写不同场景的测试用例。

使用说明: 如果不需要使用 mock ,将 isMock 改为 False 即可; 填写具体的签名列表、接口地址

isMock = True    # 如果不需要使用 mock ,直接使用接口,此处改为 False
osign_list = ['userName', 'password','verifyCode']    # 定义签名参数列表,例如签名方法为 username+password+verifycode 做md5 。  具体需要替换为实际的签名参数列表。
url = '/login'    # 具体的接口 url 相对路径, 测试时会拼凑为完整路径:  http://host/login

根据实际场景编写对应测试用例

def setUp(self):
    import warnings
    warnings.simplefilter("ignore", ResourceWarning)
    print("start")

    # 通用参数初始化
    self.testuser = {}
    self.testuser['userName']='correctuser'
    self.testuser['password']='correctpassword'
    self.testuser['verifyCode']='123456'

# 正常场景:login 是否成功。
def test_login(self):
    result =login(self.testuser,isMock=True)
    self.assertEqual(result['code'],code_success)
    self.assertEqual(result['msg'],msg_success)


# 异常场景: userName错误。
def test_login_wrong_userName(self):
    self.testuser['userName']='username'
    result = login(self.testuser, isMock=True)
    self.assertEqual(result['code'],code_login_fail)
    self.assertEqual(result['msg'],msg_login_fail)

# 异常场景: 签名错误。
def test_login_osign_error(self):
    from api_demo.api_manage import api_base
    self.assertEqual(api_base.test_osign_error(self.testuser,osign_list,url,code_sign_error,msg_sign_error,isMock=isMock), 0)

通用的签名错误校验方法:

# 传入:签名参数列表 osign_list;   返回签名不匹配的参数数量,如果返回 0 ,表示测试通过
def test_osign_error(user,osign_list,url,code_sign_error,msg_sign_error,isMock=False):
        osignFailCount = 0
        for para in osign_list:
            result = osign_error(user,osign_list,url,code_sign_error,msg_sign_error,para,isMock=isMock)
            if not result:
                osignFailCount += 1
        print('osign para lenth : %d' % len(osign_list))
        return osignFailCount

# 单个参数的签名校验:在生成完签名后修改某个参数值,看接口能否正确校验
def osign_error(user,osign_list,url,code_sign_error,msg_sign_error, para,isMock=False):
    if para in ['']:    # 过滤的加密参数,部分参数不需要加密验证
        return True
    else:
        user = getOsign(user,osign_list)
        if isinstance(user[para],int):
            user[para]= user[para] + 1
        else :
            user[para]=str(user[para] + '1')
    result = api_send(user, osign_list, url,need_osign=False,isMock=isMock)
    if result['code'] == code_sign_error and result['msg'] == msg_sign_error:
        return True
    else:
        print('osign error : %s, %s' % (para, result))
        return False

4. pytest 执行和 allure 报告生成

run_pytest_entry.py 、 run_pytest.py

4.1 通过 pytest mark 过滤需执行的用例:

@pytest.mark.skipif(environmentFlag =='1', reason='skip')   # 如果environmentFlag =='1', 跳过
# 异常场景: userName错误。
def test_login_wrong_userName(self):
    print(environmentFlag =='1')
    print('evironment is : ',environmentFlag)
    self.testuser['userName']='username'
    result = api_send(self.testuser,osign_list,url,isMock=isMock)
    self.assertEqual(result['code'],code_login_fail)
    self.assertEqual(result['msg'],msg_login_fail)

@pytest.mark.skipif(isMock , reason='skip')     # 如果 isMock 为 true, 跳过
# 异常场景: password错误。
def test_login_wrong_password(self):
    self.testuser['password']='password'
    result = api_send(self.testuser,osign_list,url,isMock=isMock)
    self.assertEqual(result['code'],code_login_fail)
    self.assertEqual(result['msg'],msg_login_fail)

5. 环境配置

api_demo.init.py 文件中可以定义多个不同的测试环境地址,并通过 environmentFlag 进行切换:

environmentFlag='1'

if environmentFlag=='1':
    host = 'http://test1.com'
elif environmentFlag=='2':
    host = 'http://test2.com'
elif environmentFlag=='3':
    host = 'http://test3.com'

6. 命令行执行:run.bat/run.sh

python run_pytest_entry.py 1

其中最后的参数 1 为 environmentFlag, 如果需要切换不同的测试环境,只需传入不同的标记位。

7. 与 jenkins 的集成

  1. freestyle job:

  2. pipeline script:

pipeline {
    agent {label 'apitest' }   // 指定运行节点
    stages {
        stage('update') {
            steps {
                echo 'update..'
                sh '/yourfolder/svn_update.sh'
            }
        }
        stage('test') {
            steps {
                echo 'test..'
                sh 'python3 /yourfolder/run_pytest_entry.py 1 4 ${JOB_NAME}'
            }
        }
        stage('allure-report') {
            steps {
                echo 'allure-report..'
            }
            post {
                        always {
                        allure includeProperties: false, jdk: '', results: [[path: 'report']]
                       }
                }
            }
        stage('sendding email'){
            steps {
                echo 'sendding email..'
            }
            post {
                 always {
                    script {
                            emailext attachLog: true, body: '${DEFAULT_CONTENT} ', recipientProviders: [[$class: 'RequesterRecipientProvider']], subject: '$DEFAULT_SUBJECT', to: 'youremail@test.com'
            }

        }
            }
        }

    }
}
共收到 18 条回复 时间 点赞

2019-01-03 update:
1.每次执行生成独立的报告目录
2.修改生成方法,用列表方式更便捷。
3.将验证签名合并为同一个用例

2Floor has deleted

2019-02-02 update:

  • 抽取公共的接口和签名测试方法
  • 补充 pytest 的 mark 功能跳过某些用例
Jerry li #4 · May 06, 2019 Author

2019-05-06 update:

  • 与 jenkins 的集成

非常棒的案例,谢谢

非常急,非常急,请问作者大神,pytest-allure-adaptor 和 allure-pytest 这两个你这边安装的是哪个啊,感觉这两个插件老是傻傻分不清楚,感谢感谢感谢

朱zhu 回复

我用的是 allure-Python, github 代码的 requirements 文件里有写

allure-Python 吗?还是 allure-pytest

朱zhu 回复

sorry,allure-pytest

大神,我如何修改你的签名方法呢?能有机会加个微信向你学习下你这个框架嘛?

Jerry li #11 · August 25, 2019 Author

签名方法是根据你要测的接口来定的,你要拿到对应的接口文档或许找开发要对应的签名算法才能改

大神一直是这样的问题,不知道是为啥,求解

Build step 'Execute shell' marked build as failure
[allure] $ /Users/Shared/Jenkins/Home/tools/ru.yandex.qatools.allure.jenkins.tools.AllureCommandlineInstallation/allure/bin/allure generate /Users/Shared/Jenkins/Home/workspace/allure/allure -c -o /Users/Shared/Jenkins/Home/workspace/allure/allure-report
env: sh: Not a directory
ERROR: Build step failed with exception
ru.yandex.qatools.allure.jenkins.exception.AllurePluginException: Can not generate Allure Report, exit code: 126
at ru.yandex.qatools.allure.jenkins.AllureReportPublisher.generateReport(AllureReportPublisher.java:314)
at ru.yandex.qatools.allure.jenkins.AllureReportPublisher.perform(AllureReportPublisher.java:231)
at hudson.tasks.BuildStepCompatibilityLayer.perform(BuildStepCompatibilityLayer.java:79)
at hudson.tasks.BuildStepMonitor$1.perform(BuildStepMonitor.java:20)
at hudson.model.AbstractBuild$AbstractBuildExecution.perform(AbstractBuild.java:741)
at hudson.model.AbstractBuild$AbstractBuildExecution.performAllBuildSteps(AbstractBuild.java:690)
at hudson.model.Build$BuildExecution.post2(Build.java:186)
at hudson.model.AbstractBuild$AbstractBuildExecution.post(AbstractBuild.java:635)
at hudson.model.Run.execute(Run.java:1843)
at hudson.model.FreeStyleBuild.run(FreeStyleBuild.java:43)
at hudson.model.ResourceController.execute(ResourceController.java:97)
at hudson.model.Executor.run(Executor.java:429)
Build step 'Allure Report' marked build as failure
Finished: FAILURE

Jerry li #13 · August 27, 2019 Author

env: sh: Not a directory

你的日志里是说这个目录不存在,你最好检查一下对应的两个目录,一个 result,一个 report

14Floor has deleted
Jerry li #15 · August 27, 2019 Author
sherry-bai 回复

env: sh: Not a directory

你的日志里是说这个目录不存在,你最好检查一下对应的两个目录,一个 result,一个 report

Jerry li 回复

对应的目录都是有的 result 里也有 json 文件 配置是这样的

Jerry li #17 · August 27, 2019 Author
sherry-bai 回复

不用加 workspace 的参数

Jerry li 回复

去掉了也还是一样的错误

Jerry li #19 · August 27, 2019 Author
sherry-bai 回复

execute shell 里的路径不用变; 下面 post action 里的 report 把 workspace 去掉

大佬,请问你是怎么处理关联接口的?修改接口的数据你是怎么预置的?

需要 Sign In 后方可回复, 如果你还没有账号请点击这里 Sign Up