在設計接口測試工具的時候,會涉及到的一個點是登錄后保持會話。保持會話有幾種方法,在java的HttpClient中可以通過CookieStore去實現,在python可以通過set_cookie方法將登錄獲取到的cookie寫到本地,之后的接口再通過get_cookie獲取本地文件中的cookie。本篇描述一種比較簡單的辦法,基於requests的requests.Session()方法實現保持會話。
1.先在登錄接口全局定義一個Session
import requests # 定義一個全局session s = requests.Session() class testlogin(): Baseurl = "" def test_login(self): url=BaseUrl+"" headers={} data ={} r =s.post(url,json=data,headers=headers) result=r.json() # print(r.cookies) return s
2.在其他接口使用登錄接口的Session
from xxx import testlogin s = testlogin().test_login() class testxx: def test01(self): url ="" r = s.get(url) result = r.json() print(result) if __name__ == "__main__": testxx().test01()