目录
封装方法
完整代码
总结：
封装方法
在编写自动化脚本的时候，都要求代码简介，上一篇unittest---unittest断言中代码重复性比较多，我们进行一次简单的优化（优化方法不同，请勿喷）
我们可以看到下面的代码每个用例中都填写了不同的参数和请求相同的接口地址，我们可以通过把相同的代码进行提取出来，然后调用这个方法，完成简化代码
# coding:utf-8
import unittest
import requests
class Music(unittest.TestCase):
def test01(self):
url = 'https://api.apiopen.top/searchMusic'
data = {
"name":"断桥残雪"
}
r = requests.post(url,data=data)
b = r.json()['result'][0]['title']
a = '断桥残雪'
self.assertEqual(a,b)
print('第一个用例通过')
def test02(self):
url = 'https://api.apiopen.top/searchMusic'
data = {
"name":"说好不哭"
}
r = requests.post(url,data=data)
a = '周杰伦'
b = r.text
self.assertIn(a,b)
print('第二个用例通过')
def test03(self):
url = 'https://api.apiopen.top/searchMusic'
data = {
"name":"芒种"
}
r = requests.post(url,data=data)
a = '抖音'
b = r.text
try:
self.assertIn(a,b,msg='\n抖音不存在芒种歌曲信息中')
except Exception as msg:
print('错误信息%s'%msg)
print('第三个用例失败')
if __name__ == '__main__':
unittest.main()1、相同的代码提取出来，单独封装成一个函数
# 相同的代码提取出来，返回歌名，方便断言
def select(self,name):
url = 'https://api.apiopen.top/searchMusic'
data = {
"name":name
}
r = requests.post(url,data=data)
b = r.json()['result'][0]['title']
return b2、通过unittest的编写用例格式，进行编写用例，调用上面的函数
def test01(self):
b = '断桥残雪'
a = self.select(b)
self.assertEqual(b,a)
def test02(self):
a = '说好不哭'
b = self.select(a)
self.assertEqual(a,b)
def test03(self):
a = '芒种'
c = '抖音'
b = self.select(a)
try:
self.assertIn(c,b,msg='\n抖音不存在芒种歌曲信息中')
except Exception as msg:
print('错误信息%s'%msg)3、完整代码
# coding:utf-8
import unittest
import requests
class Music(unittest.TestCase):
def select(self,name):
url = 'https://api.apiopen.top/searchMusic'
data = {
"name":name
}
r = requests.post(url,data=data)
b = r.json()['result'][0]['title']
return b
def test01(self):
b = '断桥残雪'
a = self.select(b)
self.assertEqual(b,a)
def test02(self):
a = '说好不哭'
b = self.select(a)
self.assertEqual(a,b)
def test03(self):
a = '芒种'
c = '抖音'
b = self.select(a)
try:
self.assertIn(c,b,msg='\n抖音不存在芒种歌曲信息中')
except Exception as msg:
print('错误信息%s'%msg)
if __name__ == '__main__':
unittest.main()通过上面的接口我们只是简单的了解下如何可以把自己的代码写的更加好看，更加的简介。（让别人看到代码就会更加的崇拜你~~~~）
我们在通过一个小例子来练习。通过请求豆瓣网的登录接口
1、豆瓣网登录接口封装完成
封装请求的url，进行对参数化，返回断言数据
def login(self,name,password):
# 登录请求地址
url = 'https://accounts.douban.com/j/mobile/login/basic'
# 请求头
headers = {
"User-Agent":"Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.142 Safari/537.36"
}
# body数据
data = {
'name':name, # 账号
"password":password, # 密码
"remember":"false"
}
r = requests.post(url,headers=headers,data=data)
b = r.text
return b2、不同的账号密码进行请求
3、通过断言判断用例是否成功
完整代码
# coding:utf-8
import requests
import unittest
class Test_login(unittest.TestCase):
def login(self,name,password):
# 登录请求地址
url = 'https://accounts.douban.com/j/mobile/login/basic'
# 请求头
headers = {
"User-Agent":"Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.142 Safari/537.36"
}
# body数据
data = {
'name':name, # 账号
"password":password, # 密码
"remember":"false"
}
r = requests.post(url,headers=headers,data=data)
b = r.text
return b
def test01(self):
# 正确的账号，密码
b = self.login('xxxxxxx','xxxxxxx')
a = '处理成功'
self.assertIn(a,b)
print('用例通过')
def test02(self):
# 错误的账号，密码
b = self.login('12345456','821006052')
a = '用户名或密码错误'
self.assertIn(a,b)
print('用例通过')
if __name__ == '__main__':
unittest.main()安静简单的优化了一些代码，可能优化的不是很好，大佬们勿喷~~~
总结：
感谢每一个认真阅读我文章的人！！！
我个人整理了我这几年软件测试生涯整理的一些技术资料，包含：电子书，简历模块，各种工作模板，面试宝典，自学项目等。欢迎大家点击下方名片免费领取，千万不要错过哦。
Python自动化测试学习交流群：全套自动化测试面试简历学习资料获取点击链接加入群聊【python自动化测试交流】：
http://qm.qq.com/cgi-bin/qm/qr?_wv=1027&k=DhOSZDNS-qzT5QKbFQMsfJ7DsrFfKpOF&authKey=eBt%2BF%2FBK81lVLcsLKaFqnvDAVA8IdNsGC7J0YV73w8V%2FJpdbby66r7vJ1rsPIifg&noverify=0&group_code=198408628
