Python的简单接口自动化测试源代码

# coding:utf-8 import unittest import requests # import HTMLTestRunner class MyTestCase(unittest.TestCase): def setUp(self): print("starting test") self.token = self.test_login() # 调用登录接口并获取token self.headers = {"token": self.token} # self.headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.token}"} # 设置请求头 def test_login(self): url = "https://ajkgj.huimaibuy.net/public/admin/admin/login" data = {"username": "admin", "password": "123456"} res = requests.post(url=url, json=data) self.assertEqual(res.json()["errorCode"], 0) token = res.json()["data"]["token"] return token def test_pagingDiseaseOffice(self): url = "https://ajkgj.huimaibuy.net/public/admin/DiseaseOffice/pagingDiseaseOffice" data = {"page": 1, "limit": 10} res = requests.post(url=url, headers=self.headers, json=data) self.assertEqual(res.json()["errorCode"], 0) def test_info(self): url = "https://ajkgj.huimaibuy.net/public/admin/admin/info" data = {"token": self.token} res = requests.post(url=url, headers=self.headers, json=data) self.assertEqual(res.json()["data"]["id"], 1) def test_getDiseaseOfficeList(self): url = "https://ajkgj.huimaibuy.net/public/admin/DiseaseOffice/getDiseaseOfficeList" data = {} res = requests.post(url=url, headers=self.headers, json=data) self.assertEqual(res.json()["data"][0]["label"], "内科") def tearDown(self): print("end test") if __name__ == '__main__': unittest.main()