前言
httprunner 3.x可以支持hook机制,在发送请求前做一些预处理或在请求完成后后置处理
- setup_hook :主要用于处理接口的前置的准备工作,也可以对请求 request 参数签名加密等操作
- teardown_hook:主要用于后置清理工作,也可以对返回 respone 解密等操作
测试步骤添加hook
# debugtalk.py
# 上海悠悠,QQ交流群:750815713
# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/
def hook_up():
print("前置操作:setup!")
def hook_down():
print("后置操作:teardown!")
只在第一个步骤添加 setup_hooks
和 teardown_hooks
# 作者-上海悠悠 QQ交流群:717225969
# 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
# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/
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)}
# 作者-上海悠悠 QQ交流群:717225969
# 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:
- ${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]
pytest脚本
# NOTE: Generated By HttpRunner v3.1.4
# FROM: testcases\demo_hook.yml
# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/
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
# 作者-上海悠悠 QQ交流群:717225969
# blog地址 https://www.cnblogs.com/yoyoketang/
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