import unittest
import requests
class TestRecharge(unittest.TestCase):
    def setUp(self):
        pass
    def tearDown(self):
        pass
    def test_recharge(self):
        url = 'http://127.0.0.1:8888/recharge'
        payload = {
            "member_id": 100004312,
            "amount": 80
        }
        res = requests.post(url,json=payload)
        print(res.json())
if __name__ == '__main__':
    unittest.main()

def jsonpath(obj, expr, result_type='VALUE', debug=0, use_eval=True):
    """traverse JSON object using jsonpath expr, returning values or paths"""

from jsonpath import jsonpath
# json格式結(jié)果
res = {
    "code": 0,
    "msg": "OK",
    "data": {
        "token_info": {
            "token_type": "Bearer",
            "expires_in": "2020-05-28 13:07:29",
            "token": "eyJhbGciOiJIUzUxAiJ9.eyJtZW1iZZJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwNjQyNDQ5fQ.s6A7pzLILf9tqpEDAU9wIPGGLGkgKEQ6EBHq26l-eUM5seCb48DWalkE7u16iRZv3uzD5hIDFbw41Jmi9V0T_Q"
        }
    }
}
# 獲取res中的token
token = jsonpath(res,'$..token')[0]
print(token)

class DevConfig():
    # 項(xiàng)目的域名或IP
    host='http://127.0.0.1:8888'
config = DevConfig()

user:
  mobile_phone: '155********'
  pwd: '12345678'

from common.requests_handler import RequestsHandler
from config.setting import config
from common.yaml_handler import yaml_data
def login():
    """登錄，接口返回token"""
    req = RequestsHandler()
    res = req.visit('post', config.host+'/login', json=yaml_data['user'])
    return res
print(login())

{
    'code': 0,
    'msg': 'OK',
    'data': {
        'token_info': {
            'token_type': 'Bearer',
            'expires_in': '2020-05-26 13:31:41',
            'token': 'eyJhbGciOiJIUpUxMiJ9.eyJtZW1iZXJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwNDcxMTAxfQ.4QDNdQ-Wyecvpv7yRnMlCx1ZXpAlb8vpohxHht5OfNcjGn_yTaISyhmn7OMzjMGlz68sJ_uFbCh2NhIaO2p_Lg'
        }
    }
}

def save_token():
    """保存token信息"""
    res = login()
    token = jsonpath(res,'$..token')[0]
    token_type = jsonpath(res,'$..token_type')[0]
    token = " ".join([token_type, token])
    return token
print(save_token())

from common.requests_handler import RequestsHandler
from config.setting import config
from common.yaml_handler import yaml_data
from jsonpath import jsonpath
def login():
    """登錄，返回token信息"""
    req = RequestsHandler()
    res = req.visit('post', config.host+'/login', json=yaml_data['user'])
    return res
def save_token():
    """保存token信息"""
    res = login()
    token = jsonpath(res,'$..token')[0]
    token_type = jsonpath(res,'$..token_type')[0]
    token = " ".join([token_type, token])
    Context.token = token
    return token
class Context:
    """將token作為類屬性"""
    token = ''
if __name__ == '__main__':
    print(save_token())

import unittest
import requests
from middleware.helper import Context, save_token
class TestRecharge(unittest.TestCase):
    def setUp(self):
        pass
    def tearDown(self):
        pass
    def test_recharge(self):
        save_token()
        token = Context.token
        url = 'http://127.0.0.1:88888/recharge'
        payload = {
            "member_id": 100051315,
            "amount": 80
        }
        res = requests.post(url,json=payload)
        print(res.json())
if __name__ == '__main__':
    unittest.main()

{
    'code': 0,
    'msg': 'OK',
    'data': {
        'leave_amount': 240.0,
        'mobile_phone': '155********',
    }
}