python实战-setup_hooks/teardown_hooks

最新推荐文章于 2024-06-11 11:38:39 发布

王小柯_0314

最新推荐文章于 2024-06-11 11:38:39 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python

原文链接：https://blog.csdn.net/qq_27371025/article/details/118101494

python 专栏收录该内容

17 篇文章

订阅专栏

etup_hooks: 在 HTTP 请求发送前执行 hook 函数，主要用于准备工作；也可以实现对请求的 request 内容进行预处理。
teardown_hooks: 在 HTTP 请求发送后执行 hook 函数，主要用于测试后的清理工作；也可以实现对响应的 response 进行修改，例如进行加解密等处理。

def hook_prepare(request=None):
    print('执行准备函数')
    if request:
        request['json']['username']='800002'

def hook_teardown(response=None):
    print('执行结束函数')
    if response:
        response.status_code =300

2、在yml测试用例文件中，通过调用setup_hooks、teardown_hooks调用自定义的函数

#hooks函数应用

- config:
    name: 验证能否打开淘宝主页
    base_url: https://www.taobao.com/
    setup_hooks:
      - ${setup_case(case01 验证能否打开淘宝主页)}
    teardown_hooks:
      - ${teardown_case(case01 验证能否打开淘宝主页)}

- test:
    name: open hnxmxit mainpage api
    setup_hooks:
      - ${setup_step(case01 打开淘宝主页)}
    teardown_hooks:
      - ${teardown_step(case01 打开淘宝主页)}
    request:
      url: /
      method: GET
    validate:
      - eq: ['status_code',200]

前言
httprunner 3.x可以支持hook机制，在发送请求前做一些预处理或在请求完成后后置处理

setup_hook ：主要用于处理接口的前置的准备工作，也可以对请求 request 参数签名加密等操作
teardown_hook：主要用于后置清理工作，也可以对返回 respone 解密等操作

测试步骤添加hook
# debugtalk.py

def hook_up():
    print("前置操作：setup!")/

def hook_down():
    print("后置操作：teardown!")

只在第一个步骤添加 setup_hooks 和 teardown_hooks
# blog地址 https://www.cnblogs.com/yoyoketang/

config:

name: logincase
base_url: ${ENV(base_url)}
variables: {}
export:
   - token

teststeps:


    name: step1 login
    request:
        url: /api/v1/login/
        method: POST
        headers:
            Content-Type: application/json
            User-Agent: python-requests/2.18.4
        json:
            username: test
            password: 123456
    setup_hooks:
        - ${hook_up()}
    teardown_hooks:
        - ${hook_down()}
    extract:
        token: content.token         # 提取token
    validate:
        - eq: [status_code, 200]
        - eq: [headers.Content-Type, application/json]
        - eq: [content.msg, login success!]
        - eq: [content.code, 0]

对应的pytest脚本

# NOTE: Generated By HttpRunner v3.1.4
# FROM: testcases\demo_hook.yml

from httprunner import HttpRunner, Config, Step, RunRequest, RunTestCase
class TestCaseDemoHook(HttpRunner):    config = Config("logincase").base_url("${ENV(base_url)}").export(*["token"])    teststeps = [
        Step(
            RunRequest("step1 login")
            .setup_hook("${hook_up()}")
            .post("/api/v1/login/")
            .with_headers(
                **{
                    "Content-Type": "application/json",
                    "User-Agent": "python-requests/2.18.4",
                }
            )
            .with_json({"username": "test", "password": 123456})
            .teardown_hook("${hook_down()}")
            .extract()
            .with_jmespath("body.token", "token")
            .validate()
            .assert_equal("status_code", 200)
            .assert_equal('headers."Content-Type"', "application/json")
            .assert_equal("body.msg", "login success!")
            .assert_equal("body.code", 0)
        ),
    ]


if __name__ == "__main__":
    TestCaseDemoHook().test_start()

从运行结果可以看到，setup_hooks 在 request 请求之前执行，teardown_hooks 在 response 之后，assert 断言之前执行

testcases\demo_hook_test.py 2021-06-21 18:45:17.051 | INFO | httprunner.runner:test_start:451 -
Start to run testcase: logincase, TestCase ID: 8043485f-1c2f-4e00-b00b-f951dc8cf6fa
2021-06-21 18:45:17.054 | INFO | httprunner.runner:__run_step:292 - run step begin: step1 login >>>>>>
2021-06-21 18:45:17.056 | INFO | httprunner.runner:__call_hooks:112 - call hook actions: setup request
2021-06-21 18:45:17.056 | DEBUG | httprunner.runner:__call_hooks:121 - call hook function: ${hook_up()}
前置操作：setup!
2021-06-21 18:45:17.197 | DEBUG | httprunner.client:request:186 - client IP: 192.168.1.125, Port: 30848
2021-06-21 18:45:17.200 | DEBUG | httprunner.client:request:194 - server IP: 49.235.92.12, Port: 8201
2021-06-21 18:45:17.202 | DEBUG | httprunner.client:log_print:40 -
....
2021-06-21 18:45:17.205 | INFO | httprunner.client:request:218 - status_code: 200, response_time(ms): 139.3 ms, response_length: 0 bytes
2021-06-21 18:45:17.207 | INFO | httprunner.runner:__call_hooks:112 - call hook actions: teardown request
2021-06-21 18:45:17.210 | DEBUG | httprunner.runner:__call_hooks:121 - call hook function: ${hook_down()}
后置操作：teardown!
2021-06-21 18:45:17.211 | INFO | httprunner.response:extract:176 - extract mapping: {'token': '392cb8c08440bde6f7ed72745e052a5c3f090aff'}
2021-06-21 18:45:17.213 | INFO | httprunner.response:validate:246 - assert status_code equal 200(int) ==> pass

请求 request 预处理
针对请求request 发出去的参数预处理，也可以用到 setup_hooks，需传一个内置 request 参数,debugtalk.py代码如下

# debugtalk.py
def request_sign(request):
"""请求sign签名"""
print("请求body:",request.get("req_json"))
# 新增 sign 参数
request["req_json"]["sign"] = "sign xxxxxxxxxxxxxxx"
print("sign 签名后请求body:", request.get("req_json"))
yaml脚本引用函数${request_sign($request)}

config:
    name: logincase
    base_url: ${ENV(base_url)}
    variables: {}
    export:
        - token

teststeps:
-
    name: step1 login
    request:
        url: /api/v1/login/
        method: POST
        headers:
            Content-Type: application/json
            User-Agent: python-requests/2.18.4
        json:
            username: test
            password: 123456
    setup_hooks:
        - ${request_sign($request)}
    extract:
        token: content.token         # 提取token
    validate:
        - eq: [status_code, 200]
        - eq: [headers.Content-Type, application/json]
        - eq: [content.msg, login success!]
        - eq: [content.code, 0]
1

pytest脚本

# NOTE: Generated By HttpRunner v3.1.4
# FROM: testcases\demo_hook.yml


from httprunner import HttpRunner, Config, Step, RunRequest, RunTestCase


class TestCaseDemoHook(HttpRunner):

    config = Config("logincase").base_url("${ENV(base_url)}").export(*["token"])

    teststeps = [
        Step(
            RunRequest("step1 login")
            .setup_hook("${request_sign($request)}")
            .post("/api/v1/login/")
            .with_headers(
                **{
                    "Content-Type": "application/json",
                    "User-Agent": "python-requests/2.18.4",
                }
            )
            .with_json({"username": "test", "password": 123456})
            .extract()
            .with_jmespath("body.token", "token")
            .validate()
            .assert_equal("status_code", 200)
            .assert_equal('headers."Content-Type"', "application/json")
            .assert_equal("body.msg", "login success!")
            .assert_equal("body.code", 0)
        ),
    ]


if __name__ == "__main__":
    TestCaseDemoHook().test_start()

运行后查看日志可以看到，已经签名成功

请求body: {'username': 'test', 'password': 123456}
sign 签名后请求body: {'username': 'test', 'password': 123456, 'sign': 'sign xxxxxxxxxxxxxxx'}
2021-06-21 18:58:57.876 | DEBUG | httprunner.client:request:186 - client IP: 192.168.1.125, Port: 31471
2021-06-21 18:58:57.878 | DEBUG | httprunner.client:request:194 - server IP: 49.235.92.12, Port: 8201
2021-06-21 18:58:57.881 | DEBUG | httprunner.client:log_print:40 -
================== request details ==================
method : POST
url : http://127.0.0.1:8000/api/v1/login/
headers : {
"User-Agent": "python-requests/2.18.4",
"Accept-Encoding": "gzip, deflate",
"Accept": "*/*",
"Connection": "keep-alive",
"Content-Type": "application/json",
"HRUN-Request-ID": "HRUN-911bb03a-a24a-40a6-ad32-9d62a9d000fd-137716",
"Content-Length": "72"
}
cookies : {}
body : {
"username": "test",
"password": 123456,
"sign": "sign xxxxxxxxxxxxxxx"
}

返回 response 处理
如果需要在接口返回后，对返回的结果处理，可以添加respone请求参数，比如我把返回的状态码200改成了203

# debugtalk.py

def response_status(response):
    """修改返回状态码"""
    print("返回response status_code:", response.status_code)
    response.status_code = 203
    print("修改后返回response status_code:", response.status_code)

yaml 引用示例

setup_hooks:
        - ${request_sign($request)}
    teardown_hooks:
        - ${response_status($response)}

pytest引用示例

.with_json({"username": "test", "password": 123456})
            .teardown_hook("${response_status($response)}")

运行结果

2021-06-21 19:07:13.729 | INFO | httprunner.client:request:218 - status_code: 200, response_time(ms): 157.17 ms, response_length: 0 bytes
2021-06-21 19:07:13.731 | INFO | httprunner.runner:__call_hooks:112 - call hook actions: teardown request
2021-06-21 19:07:13.736 | DEBUG | httprunner.runner:__call_hooks:121 - call hook function: ${response_status($response)}
返回response status_code: 200
修改后返回response status_code: 203

httprunner 3.x 版本在 config 中不支持setup_hook和teardown_hook