使用python test测试http接口

使用python test测试http接口

1. 获取token和控制session，后面大多数接口要带上这些信息

   import time
   import requests
   
   from common.aes_algorithm import AES
   from config.config import Config
   from config.log import log
   
   
   class Common:
       username = "admin"
       password = "888888"
       token = None
       user_id = None
       controlSession = None
   
       def __init__(self):
           pass
   
       @classmethod
       def login(cls):
           timestamp = int(time.time())
           password = AES.encrypt(cls.password, str(timestamp))
           data = {
               "username": cls.username,
               "password": password,
               "timestamp": timestamp,
           }
           response = requests.post(f"{Config.base_url}/user/login", json=data)
           response.raise_for_status()
           # 打印响应内容
           dataJson = response.json()
           log.debug(dataJson)
           if dataJson["status"] == 200:
               data = dataJson.get("data")
               cls.token = data.get("Authorization")
               cls.user_id = data.get("userID")
   
       @classmethod
       def get_control_session(cls):
           if cls.controlSession is None:
               timestamp = int(time.time())
               password = AES.encrypt(cls.password, str(timestamp))
               response = requests.post(
                   f"{Config.base_url}/user/verifyControlPasswd",
                   headers={"Authorization": cls.token},
                   json={
                       "userID": cls.user_id,
                       "password": password,
                       "timestamp": timestamp,
                   },
               )
               response.raise_for_status()
               dataJson = response.json()
               log.debug(dataJson)
               if dataJson["status"] == 200:
                   data = dataJson.get("data")
                   cls.controlSession = data.get("controlSession")
   
       def get_token(cls):
           cls.login()
           cls.get_control_session()
   
   if __name__ == "__main__":
       common = Common()
       common.get_token()
   

2. 使用unittest编写单元测试

   import time
   import unittest
   
   import requests
   
   from common.login import Common
   from config.config import Config
   from config.log import log
   
   
   class StgyModeTest(unittest.TestCase):
       @classmethod
       def setUpClass(cls):
           cls.common = Common()
           cls.common.get_token()  # 只执行一次
   
           timestamp = int(time.time())
           cls.data = [
               {
                   "userID": Common.user_id,
                   "controlSession": Common.controlSession,
                   "timestamp": timestamp,
                   "mode": "powerBalanced",
               },
               {
                   "userID": Common.user_id,
                   "controlSession": Common.controlSession,
                   "timestamp": timestamp,
                   "mode": "socBalanced",
               },
               {
                   "userID": Common.user_id,
                   "controlSession": Common.controlSession,
                   "timestamp": timestamp,
                   "mode": "customRatio",
                   "powerWeights": [
                       {"pcsCode": "pcs1", "ratio": 0.3},
                       {"pcsCode": "pcs2", "ratio": 0.7},
                       {"pcsCode": "pcs3", "ratio": 0},
                   ],
               },
           ]
   
       def test_set_power_balanced_stgy(self):
           response = requests.post(
               f"{Config.base_url}/strategy/setStrategyMode",
               headers={"Authorization": Common.token},
               json=self.data[0],
           )
           log.info(f"set stgy mode response: {response.text}")
           response.raise_for_status()
           # 打印响应内容
           dataJson = response.json()
           if dataJson["status"] == 200:
               data = dataJson.get("data")
               log.info(f"set stgy mode success, data: {data}")
   
       def test_set_soc_balanced_stgy(self):
           response = requests.post(
               f"{Config.base_url}/strategy/setStrategyMode",
               headers={"Authorization": Common.token},
               json=self.data[1],
           )
           log.info(f"set stgy mode response: {response.text}")
           response.raise_for_status()
           # 打印响应内容
           dataJson = response.json()
           if dataJson["status"] == 200:
               data = dataJson.get("data")
               log.info(f"set stgy mode success, data: {data}")
   
       def test_set_custom_power_weights_stgy(self):
           response = requests.post(
               f"{Config.base_url}/strategy/setStrategyMode",
               headers={"Authorization": Common.token},
               json=self.data[2],
           )
           log.info(f"set stgy mode response: {response.text}")
           response.raise_for_status()
           # 打印响应内容
           dataJson = response.json()
           if dataJson["status"] == 200:
               data = dataJson.get("data")
               log.info(f"set stgy mode success, data: {data}")
   
       # 获取策略接口
       def test_get_stgy_mode(self):
           response = requests.get(
               f"{Config.base_url}/strategy/getStrategyMode",
               headers={"Authorization": Common.token},
           )
           log.info(f"get stgy mode response: {response.text}")
   
   
   if __name__ == "__main__":
       unittest.main()
   

vscode里面启用unittest插件

1. 代开vscode设置，搜索python test，启用unittest

   

2. 在左侧可以运行单个测试