1. 程式人生 > 其它 >python+requests介面自動化測試1--基本請求方法封裝+獲取token

python+requests介面自動化測試1--基本請求方法封裝+獲取token

前段時間用python對公司專案的介面進行了自動化測試,由於存在很多不完善的地方,在這過程中也遇到了一些問題。為了便於以後溫習回顧,以下做了一些總結:

1.請求登入介面

import hashlib
from random import randint, choice
import requests
import json
import time

# 生成隨機數
def get_nonce():
    ret = ""
    for i in range(6):
        num = randint(0, 9)
        s = str(num)
        ret += s
    
return ret # 獲取時間戳 def get_timestamp(): return str(int(time.time() * 1000)) # 加密 def md5_sys(param): str_md5 = hashlib.md5(param.encode(encoding='utf-8')).hexdigest() return str_md5 # 簽名(省略) def get_sign(parameter, ...): ...return sign
nonce
= get_nonce() timestamp = get_timestamp() url
= "https://xxx/login/password"
params
= {"username": "admin", "password": "4297f44b13955235245b2497399d7a93"}
sign = get_sign(params, ...)
header = {"Content-Type": "application/json", "client": "xxx", "sign": sign, "timestamp": timestamp,"nonce": nonce} 
res
= requests.post(url=url, headers=header, data=str(params))
#或 res = requests.post(url=url, headers=header, json=params)
print(res.text)  # 以text形式顯示響應內容
print(res.json())  # 以json格式解析響應內容
print(res.encoding)  # 檢視requests使用的編碼方式

print(res.status_code) # 伺服器返回的響應狀態碼
print(res.headers)  # 返回的響應頭
print(res.request) # 請求方式
print(res.url)  # 返回請求url

2.方法封裝

class BaseRequest:
    def __init__(self, url):
        self.url = url

    def get(self, header, data=None):
        if data != None:
            r = requests.get(url=self.url, headers=header, params=data)

        else:
            r = requests.get(url=self.url, headers=header)
        return r

    def post(self, header, data=None):
        if data != None:
            r = requests.post(url=self.url, headers=header, json=data)

        else:
            r = requests.post(url=self.url, headers=header)
        return r

    def sureMethod(self, header, method, data):
        if method == "get":
            res = self.get(header, data)
        elif method == "post":
            res = self.post(header, data)
        return res


if __name__ == '__main__':
    url = "https://xxx/login/password" 
  nonce
= get_nonce()
  timestamp
= get_timestamp()
  params
= {"username": "admin", "password": "4297f44b13955235245b2497399d7a93"}
  sign
= get_sign(params, ...)
  header
= {"Content-Type": "application/json", "client": "xx", "sign": sign, "timestamp": timestamp, "nonce": nonce}
  br
= BaseRequest(url)
  re
= br.sureMethod(header, "post", params)
  print(re.json())

提取token:

token = re.json()["data"]["token"]["access_token"]
   yamlpath = r'D:\Python37\Workspace\IndustryApi_Test\test_data\token.yaml'  # yaml檔案位置
   tokenValue = {'token': token}  # 儲存token格式
   with open(yamlpath, "w", encoding='utf-8') as f:
       yaml.dump(tokenValue, f, Dumper=yaml.Dumper)  # 寫入資料

token.yaml

token: eyJhbGciOiJSUzI1NiIsImtpZCI6IjU5MUVBQjVEMEZEMUUzNUZGRTA5RjQ0QzY1QkM0RTY4MDU1MEMxMDAiLCJ0eXAiOiJKV1QiLCJ4NXQiOiJXUjZyWFFfUjQxXy1DZlJNWmJ4T2FBVlF3UUEifQ.eyJuYmYiOjE2MjM5MDExMDgsImV4cCI6MTYyMzkwODMwOCwiaXNzIjoiaHR0cDovL2xvY2FsaG9zdDoxMDAxIiwiYXVkIjpbImh0dHA6Ly9sb2NhbGhvc3Q6MTAwMS9yZXNvdXJjZXMiLCJhcmltLndlYi5zY29wZSJdLCJjbGllbnRfaWQiOiJhcmltLndlYiIsInN1YiI6ImlkZW50aXR5IiwiYXV0aF90aW1lIjoxNjIzOTAxMTA4LCJpZHAiOiJsb2NhbCIsIkxvZ2luSWQiOiI0NzIyIiwiTG9naW5UaW1lIjoiMjAyMS0wNi0xNyAxMTozODoyOCIsIkNsaWVudElkIjoiOSIsIkNsaWVudCI6ImFyaW0ud2ViIiwiUGxhdGZvcm1JZCI6IjIiLCJBY2NvdW50SWQiOiIyMCIsIlVzZXJOYW1lIjoiY3NseXkiLCJSb2xlSWQiOiIxMSIsIlJvbGUiOiLotoXnuqfnrqHnkIYiLCJOaWNrbmFtZSI6IuWImOiOueiOuSIsIlNjaG9vbElkIjoiMSIsIlNjaG9vbCI6IuenkeWNjuWkqeW6nCIsIkRldmljZUlkIjoiMCIsInNjb3BlIjpbIm9wZW5pZCIsInByb2ZpbGUiLCJhcmltLndlYi5zY29wZSIsIm9mZmxpbmVfYWNjZXNzIl0sImFtciI6WyJjdXN0b20iXX0.Fo_XxxAwHgwg5op10JsYrCL_rTqS7ldRSv964HphQhtvEWLSskmRcBahJyQrbUukkCdppsfQ7De3nPTPVvxolJFULRWKOrbg-AjZJdMc5y-Xg4AIviiYXCj78oPa2TR6gieoYB0Z8hlgVmpmg16e7V9S1YD3FC8UBuSTsBFsg4ofWUyGJdhwlswlywpu2vW2KircO3ACt9dM-d83aQPm2O4QvZiub6lnQbjhCXfEb8hdiaYzKIhBc8Xkbtj4UcBRsgN-Jfhjbwg1uIt6JABdauHKgeonqzqiaA0NahL_6c_akg03_KhSp3xkY0eRsWLfTVUdY1B23aQek1kkVhodgA