完善环境变量的设置方法

#coding:utf-8
import requests,unittest

class EnvironmentVariableNew(unittest.TestCase):
    '''设置环境变量'''
    def setUp(self):
        pass

    def tearDown(self):
        self.s.close()
        pass

    def get_url(self):
        '''获取url'''
        url = "111"     #uat环境域名
        # url = "0000"       #线上环境
        return url

    def get_token(self):
        '''获取token'''
        token = "222"      #测试环境
        # token = "333"      #生产环境
        return token

    def log_in(self,url_api,payload1):
        '''登录接口'''
        ev = EnvironmentVariableNew()
        s = requests.session()
        #获取域名
        url = ev.get_url()
        #拼接请求地址
        url2 = url + url_api
        #获取对应环境token
        token = ev.get_token()
        headers2 = {
        "Content-Type": "application/json",
        "Authorization": token
         }
        #发送请求
        r = s.post(url=url2,headers=headers2,json=payload1)
        return r.json()

改class中主要封装获取域名,token,接口请求方式

posted @ 2021-01-06 16:00  Camillezxl  阅读(85)  评论(0)    收藏  举报