建站
咨詢
售后
建站咨詢
新聞中心
在發(fā)送請(qǐng)求的時(shí)候，我們希望在發(fā)送請(qǐng)求參數(shù)前，帶上簽名的值，或者返回的內(nèi)容需要二次處理，解密后返回。
此功能我們可以用 hooks 鉤子來實(shí)現(xiàn)
pip 安裝插件
pip install pytest-yaml-yoyohooks 功能在v1.0.4版本上實(shí)現(xiàn)
response 鉤子功能requests 庫(kù)只支持一個(gè) response 的鉤子，即在響應(yīng)返回時(shí)可以捎帶執(zhí)行我們自定義的某些方法。
可以用于打印一些信息，做一些響應(yīng)檢查或想響應(yīng)對(duì)象中添加額外的信息
# 作者-上海悠悠 微信/QQ交流:283340479
# blog地址 https://www.cnblogs.com/yoyoketang/
import requests
url = 'https://httpbin.org/get'
def response_status(resopnse, *args, **kwargs):
print('url', resopnse.url)
resopnse.status = 'PASS' if resopnse.status_code< 400 else 'FAIL'
res = requests.get(url, hooks={'response': response_status})
print(res.status)以上是基于requests 庫(kù)的鉤子功能實(shí)現(xiàn)的基本方式
yaml 用例中添加response 鉤子在yaml 文件中添加response 鉤子功能，跟上面代碼方式差不多, 有2種方式
- 1.寫到config 全局配置，每個(gè)請(qǐng)求都會(huì)帶上hooks
- 2.寫到單個(gè)請(qǐng)求的request 下，僅單個(gè)請(qǐng)求會(huì)帶上hooks功能
先看單個(gè)請(qǐng)求的response 鉤子
test_hook1.yml
# 作者-上海悠悠 微信/QQ交流:283340479
# blog地址 https://www.cnblogs.com/yoyoketang/
config:
name: post示例
teststeps:
-
name: post
request:
method: POST
url: http://httpbin.org/post
json:
username: test
password: "123456"
hooks:
response: ['hook_response']
extract:
url: body.url
validate:
- eq: [status_code, 200]
- eq: [headers.Server, gunicorn/19.9.0]
- eq: [$.code, 0]
-
name: post
request:
method: POST
url: http://httpbin.org/post
json:
username: test
password: "123456"
extract:
url: body.url
validate:
- eq: [status_code, 200]
- eq: [headers.Server, gunicorn/19.9.0]
- eq: [$.code, 0]在conftest.py 中注冊(cè)鉤子函數(shù)
# 作者-上海悠悠 微信/QQ交流:283340479
# blog地址 https://www.cnblogs.com/yoyoketang/
def hook_response(response, *args, **kwargs):
# print(response.text) 原始數(shù)據(jù)
print("執(zhí)行response hook 函數(shù)內(nèi)容....")
class NewResponse:
text = '{"code": 0, "data": {"token": "yo yo"}}' # response.text解密
history = response.history
headers = response.headers
cookies = response.cookies
status_code = response.status_code
raw = response.raw
is_redirect = response.is_redirect
content = b'{"code": 0, "data": {"token": "yo yo"}}' # response.text解密
elapsed = response.elapsed
@staticmethod
def json():
# 拿到原始的response.json() 后解碼
return {"code": 0, "data": {"token": "yo yo"}}
return NewResponse
my_builtins.hook_response = hook_response由于上面用例只在第一個(gè)請(qǐng)求中使用了hooks功能，所以第二個(gè)請(qǐng)求的斷言- eq: [$.code, 0]會(huì)失敗
鉤子方法調(diào)用語(yǔ)法
- 1.層級(jí)是在request 下
- 2.hooks 關(guān)鍵字對(duì)應(yīng)的是一個(gè)字典 {“response”: []}
- 3.response 的值可以是單個(gè)函數(shù)名稱，也可以是多個(gè)
func1, func2,或者是一個(gè)list類型[func1, func2] - 4.response 的值必須是一個(gè)可以調(diào)用的函數(shù)，此函數(shù)需在conftest 中注冊(cè)綁定到
my_builtins模塊 - 5.調(diào)用的函數(shù)第一個(gè)參數(shù)是
response， 可以重寫response內(nèi)容（如需要對(duì)返回結(jié)果解密），也可以不用重寫
request:
method: POST
url: http://httpbin.org/post
hooks:
response: ['hook_response']在config 中配置全局hooks功能，格式如下
config:
name: post示例
hooks:
response: ['hook_response']test_hook2.yml完整示例
config:
name: post示例
hooks:
response: ['hook_response']
teststeps:
-
name: post
request:
method: POST
url: http://httpbin.org/post
json:
username: test
password: "123456"
hooks:
response: ['hook_response']
extract:
url: body.url
validate:
- eq: [status_code, 200]
- eq: [headers.Server, gunicorn/19.9.0]
- eq: [$.code, 0]
-
name: post
request:
method: POST
url: http://httpbin.org/post
json:
username: test
password: "123456"
extract:
url: body.url
validate:
- eq: [status_code, 200]
- eq: [headers.Server, gunicorn/19.9.0]
- eq: [$.code, 0]全局配置hooks， 那么該用例下所有的請(qǐng)求都會(huì)帶上hooks
請(qǐng)求預(yù)處理鉤子如果需要對(duì)請(qǐng)求參數(shù)預(yù)處理，我們還新增了一個(gè)request 請(qǐng)求鉤子，可以獲取到發(fā)送請(qǐng)求時(shí)的request參數(shù)
在conftest.py
# 作者-上海悠悠 微信/QQ交流:283340479
# blog地址 https://www.cnblogs.com/yoyoketang/
def func1(req):
print(f'請(qǐng)求預(yù)處理：{req}')
def func2():
print(f'請(qǐng)求預(yù)處理-----------')
my_builtins.func1 = func1
my_builtins.func2 = func2在 yaml 文件中使用示例
config:
name: post示例
teststeps:
-
name: post
request:
method: POST
url: http://httpbin.org/post
json:
username: test
password: "123456"
hooks:
request: ['func1', 'func2']
response: ['hook_response']
extract:
url: body.url
validate:
- eq: [status_code, 200]
- eq: [headers.Server, gunicorn/19.9.0]
- eq: [$.code, 0]
-
name: post
request:
method: POST
url: http://httpbin.org/post
json:
username: test
password: "123456"
hooks:
response: ['hook_response']
extract:
url: body.url
validate:
- eq: [status_code, 200]
- eq: [headers.Server, gunicorn/19.9.0]
- eq: [$.code, 0]在config 中設(shè)置全局hooks示例
config:
name: post示例
hooks:
request: ['func1', 'func2']
response: ['hook_response']由于request 變量是pytest的一個(gè)內(nèi)置fixture，此變量保留著，獲取請(qǐng)求參數(shù)的函數(shù)使用req代替。
利用request hook功能可以實(shí)現(xiàn)請(qǐng)求參數(shù)的預(yù)處理，比如請(qǐng)求body簽名和加密處理等需求。
你是否還在尋找穩(wěn)定的海外服務(wù)器提供商？創(chuàng)新互聯(lián)www.cdcxhl.cn海外機(jī)房具備T級(jí)流量清洗系統(tǒng)配攻擊溯源，準(zhǔn)確流量調(diào)度確保服務(wù)器高可用性，企業(yè)級(jí)服務(wù)器適合批量采購(gòu)，新人活動(dòng)首月15元起，快前往官網(wǎng)查看詳情吧
網(wǎng)頁(yè)名稱：pytest+yaml框架-6.hooks鉤子功能實(shí)現(xiàn)-創(chuàng)新互聯(lián)
文章轉(zhuǎn)載：http://www.dlmjj.cn/article/dshsdg.html
