1、登录获取cookies
import requests
def login():
# 登录接口
url = "登录接口"
payload = {'userNick': '登录账号', 'password': '登录密码'}
login_res = requests.post(url, data=payload)
print(login_res.json())
# 获取响应结果的cookies
cookies = login_res.cookies
print("cookies: %s" % cookies)
# 返回cookies
return cookies
2、在其他接口中获取cookies:
import unittest
import requests
import os, sys
parentdir = os.path.dirname(os.path.abspath(__file__))
sys.path.insert(0, parentdir)
# from db_fixture import test_data
from common.requests_handler import login
import json
class AddAgentGradeTest(unittest.TestCase):
''' 添加代理商等级 '''
def setUp(self):
pass
def tearDown(self):
print(self.result)
def test_add_grade_success(self):
''' 添加成功 '''
cookies = login()
self.base_url = "添加等级接口"
headers = {
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.130 Safari/537.36",
"Content-Type": "application/json;charset=UTF-8",
"Referer": ""
}
payload = {
'name': '95星',
'first_pct': 1.25,
'renew_pct': 2.20,
'custom_pct': 3.20,
'dock_pct': 2.55
}
data_json = json.dumps(payload)
r = requests.post(self.base_url, headers=headers, data=data_json, cookies=cookies)
self.result = r.json()
self.assertEqual(self.result['success'], True)
if __name__ == '__main__':
unittest.main()
一开始因为headers没加,在运行时报错系统异常,后来加上headers就成功了!~