石頭智能洗地機A10 Plus體驗：雙向自清潔治好了我的懶癌
一、前言和介紹專為家庭請假懶人而生的石頭科技在近日又帶來了自己的全新旗艦新品，石頭智能洗地機A10 Plus。從這個產品名上就不難看出，這次石頭推出的并不是常見的掃地機器
在自動化接口測試中，我們經常需要發送HTTP請求來模擬用戶的操作并驗證接口的正確性。夠靈活處理請求參數、添加認證信息以及處理依賴參數。這正是鉤子函數的用武之地。
下面勇哥將以一個實際的示例場景為例，詳細介紹如何利用鉤子函數進行自動化接口測試，并展示如何設計和應用這些鉤子函數。
遙想當年，勇哥一開始對【鉤子】也是懵逼的!!鉤子函數就是在特定事件發生前后執行的函數。通過鉤子我們可以在 HTTP 請求的不同階段執行花里胡哨的自定義操作，如修改URL、更新請求頭、請求參數加密加簽、處理響應結果...
上代碼：
```pythonclass Hooks: def __init__(self): self.before_request_funcs = {} self.after_request_funcs = {} def before_request(self, func): """ 注冊 before_request 鉤子函數 """ self.before_request_funcs[func.__name__] = func return func def after_request(self, func): """ 注冊 after_request 鉤子函數 """ self.after_request_funcs[func.__name__] = func return func def run_before_request_hooks(self, func_names, request, json_data): """ 執行 before_request 鉤子函數 """ for func_name in func_names: if func_name in self.before_request_funcs: func = self.before_request_funcs[func_name] json_data = func(request, json_data) return json_data def run_after_request_hooks(self, func_names, request, response): """ 執行 after_request 鉤子函數 """ for func_name in func_names: if func_name in self.after_request_funcs: func = self.after_request_funcs[func_name] response = func(request, response) return response模擬用戶注冊，并驗證注冊接口的正確性。同時，在每個請求中自動添加認證信息和處理依賴參數。
有錢才能吃飯，所以我們看看怎么封裝代碼：
def req(url, method, **kwargs): """ 發送請求并返回響應對象 """ before_hooks = kwargs.pop('before_hooks', []) after_hooks = kwargs.pop('after_hooks', []) json_data = kwargs.pop('json', {}) request = requests.Request(method=method, url=url, **kwargs) prepared_request = session.prepare_request(request) json_data = hooks.run_before_request_hooks(before_hooks, prepared_request, json_data) prepared_request.body = json.dumps(json_data) response = session.send(prepared_request) response = hooks.run_after_request_hooks(after_hooks, prepared_request, response) return response上述代碼中，勇哥先封裝一個http請求，在里面實現不同位置調用鉤子函數
常規玩法，干百度。
def test_user_registration(): url = "http://www.baidu.com" # url = "http://jsonplaceholder.typicode.com/posts" data = { "userId": "testuser", "title": "password123", "body": "測試玩家勇哥" } headers = { "Content-Type": "application/json" } before_hooks = [add_authentication_headers.__name__, handle_dependent_parameters.__name__] after_hooks = [after_dependent_parameters.__name__] kwargs = {"json": data, "headers": headers} return req(url, "post", before_hooks=before_hooks, after_hooks=after_hooks, **kwargs)在上述測試用例中，我們使用req函數發送了一個POST請求來模擬用戶注冊，并對返回的響應進行了斷言驗證。
寫N個前置鉤子函數，發送注冊請求前自動添加認證信息：
@hooks.before_requestdef add_authentication_headers(request, json_data): """ 添加認證頭信息 """ print("前置鉤子函數，添加認證頭信息", request) request.headers["Authorization"] = "Bearer YOUR_AUTH_TOKEN" return json_data@hooks.before_requestdef handle_dependent_parameters(request, json_data): """ 處理依賴參數 """ print("前置鉤子函數,處理依賴參數", request) json_data["verification_code"] = get_verification_code() return json_data定義一個add_authentication_headers函數，它是一個前置鉤子函數。在這個鉤子函數中，我們通過調用get_access_token函數獲取訪問令牌，并將其添加到請求的Authorization頭信息中。
通過注冊該鉤子函數，每次發送請求前都會自動添加認證信息，無需在每個測試用例中重復編寫。
def get_verification_code(): # 實現獲取驗證碼的邏輯 return "YOUR_VERIFICATION_CODE"上述代碼是可以是一些公共函數，以便全局使用，比如我們這里寫一些獲取驗證碼的方法
處理一些依賴參數，例如在注冊用戶之前需要獲取一個驗證碼，并將驗證碼添加到注冊請求中。
@hooks.after_requestdef after_dependent_parameters(request, response): """ 處理后置 """ print("發送請求后執行", request, "后置鉤子函數,處理依賴參數", response) return response我們又定義了一個后置鉤子函數。在這個鉤子函數中，我們調用after_dependent_parameters函數處理一些依賴數據,并返回任意結果。
通過注冊該鉤子函數，每次發送請求前以及發送請求后都會自動處理依賴參數，使我們的測試用例更加簡潔和可維護。
可以按照這種方式執行測試用例，并觀察鉤子函數的作用。：
if __name__ == "__main__": res = test_user_registration # 執行測試 print("打印看看響應結果",res)輸出結果如下：
執行test_user_registration()函數來執行我們的測試用例,但是這個測試用例在前面已經被我們注冊了很多鉤子函數，所以不需要額外處理，我們在買菜這部分就已經實現了，通過不同的函數名，執行不同的鉤子函數，非常靈活，擴展性非常強。
以上就是勇哥使用鉤子函數在自動化接口測試中實現了對HTTP請求的增強和擴展的示例和思路，希望能夠為各位在自動化接口測試中的工作帶來一些啟發和幫助。
本文鏈接：http://www.tebozhan.com/showinfo-26-10475-0.html優化自動化接口測試：利用鉤子函數增強HTTP請求處理
聲明：本網頁內容旨在傳播知識，若有侵權等問題請及時與本網聯系，我們將在第一時間刪除處理。郵件：2376512515@qq.com
上一篇： C語言操作寄存器和函數指針
下一篇： 別再用 “! = null” 做判空了！
Copyright ? 2016-2023 天津谷騏科技有限公司 版權所有 sitemap.xml
違法及侵權請聯系：2376512515@qq.com 津ICP備18001702號
津公網安備 12010102000574號