- 目的在于token实现复用
第一种:在要运行接口前加上登录的接口
class login(unittest.TestCase):
@classmethod
#登录函数
def test_login(self):
url =用户登陆链接
body = json.dumps({
'account': '***',
'password': '***'
})
token = requests.post(url, data=body).json()['data']['token']
return token #获取到token
#下面就是我们需要测试的接口
class Test_getExportFiles(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.web=login( ) #cls.web就等于调用上面的class login
def test_getExportFiles(self):
url=被测试链接
token1 = self.web.test_login() #调用login中的test_login函数
headers = {
'Content-Type': 'application/json',
'token':token1
}
response = requests.get(url, headers=headers).json()
print(response)
第二种方法更简单:新建一个login文件,每次用到直接用。
'''
这是login文件中的代码
'''
class Testncms(unittest.TestCase):
@classmethod
def test_login(self):
login_url = 链接
body = json.dumps({
'account': '***',
'password': '***'
})
token = requests.post(login_url, data=body).json()['data']['token']
return token
'''
下面是需要测试的代码
'''
from login import Testncms #在新文件中导入login的Testncms模块
class Test_updateFileRecord(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.web=Testncms( ) #cls.web就等于login里面的Testncms
def test_updateFileRecord(self):
url=链接
token1 = self.web.test_login()
headers = {
'token': token1,
'Content-Type': 'application/json'
}
data=json.dumps({
"id": "***",
"remark": "789"
})
response=requests.post(url, headers=headers, data=data).json()
print(response)