python单元测试框架（继承、unittest参数化、断言、测试报告）
一、继承
继承能解决什么问题？
unittest每个模块都要用到前提条件以及清理，如果有上百个模块，我们要改域名和浏览器，就会工作量很大特别麻烦，这时我们可以用继承的思想只用改一次
我们可以将前提和清理提出来单独放到一个文件里，具体代码如下
from selenium import webdriver
import unittest
class Init(unittest.TestCase):
@classmethod
def setUpClass(cls) -> None:
cls.driver=webdriver.Chrome()
cls.driver.maximize_window()
cls.driver.get('http://www.baidu.com')
cls.driver.implicitly_wait(30)
@classmethod
def tearDownClass(cls) -> None:
cls.driver.quit()然后我们写测试用例的时候可以继承它，具体代码如下
from selenium import webdriver
import unittest
from 单元测试框架.test.init import Init
class BaiduTest(Init):
def test_baidu_title(self):
'''百度测试:验证百度首页的title'''
# assert self.driver.title=='百度一下，你就知道'
self.assertEqual(self.driver.title,'百度一下，你就知道')
def test_baidu_url(self):
'''百度测试:验证百度首页的url'''
assert self.driver.current_url=='https://www.baidu.com/'
def test_baidu_video(self):
'''百度测试：验证点击视频后跳转到视频的页面'''
nowhandler=self.driver.current_window_handle
self.driver.find_element_by_link_text('视频').click()
allhandlers=self.driver.window_handles
for handler in allhandlers:
if handler!=nowhandler:
self.driver.switch_to.window(handler)
self.assertTrue(self.driver.current_url,'https://haokan.baidu.com/?sfrom=baidu-top')
self.driver.close()
self.driver.switch_to.window(nowhandler)
def test_baidu_map(self):
'''百度测试：验证点击地图后跳转到地图的页面'''
nowhandler=self.driver.current_window_handle
self.driver.find_element_by_link_text('地图').click()
allhandlers=self.driver.window_handles
for handler in allhandlers:
if handler!=nowhandler:
self.driver.switch_to.window(handler)
self.assertTrue(self.driver.current_url.startswith('https://map.baidu'))
self.driver.close()
self.driver.switch_to.window(nowhandler)
if __name__ == '__main__':
unittest.main()二、参数化
在unittest测试框架中，参数化使用的库为：parameterized 安装方式为：pip3 install parameterized
参数化：
我们把相同的测试步骤，应用于不同的测试场景，那么我们就可以使用参数化了
可以解决的问题是可以使用少量的测试代码，来覆盖更多的测试场景
例如：我们测一下sina邮箱的登录模块，代码如下：
from selenium import webdriver
import unittest
import time as t
class BaiduTest(unittest.TestCase):
def setUp(self) -> None: #前提
self.driver=webdriver.Chrome()
self.driver.get('https://mail.sina.com.cn/')
self.driver.maximize_window()
self.driver.implicitly_wait(30)
def tearDown(self) -> None: #清理
self.driver.quit()
def test_sina_null(self):
'''sina邮箱验证：登录账户为空'''
self.driver.find_element_by_class_name('loginBtn').click()
divText=self.driver.find_element_by_xpath('/html/body/div[3]/div/div[2]/div/div/div[4]/div[1]/div[1]/div[1]/span[1]')
self.assertEqual(divText.text,'请输入邮箱名')
def test_sina_email_format(self):
'''sina邮箱验证：登录邮箱格式不正确'''
self.driver.find_element_by_id('freename').send_keys('qwert')
self.driver.find_element_by_class_name('loginBtn').click()
divText=self.driver.find_element_by_xpath('/html/body/div[3]/div/div[2]/div/div/div[4]/div[1]/div[1]/div[1]/span[1]')
self.assertEqual(divText.text,'您输入的邮箱名格式不正确')
def test_sina_username_error(self):
'''sina邮箱验证：登录账户不匹配'''
self.driver.find_element_by_id('freename').send_keys('[email protected]')
self.driver.find_element_by_id('freepassword').send_keys('asdfg')
self.driver.find_element_by_class_name('loginBtn').click()
t.sleep(3)
divText=self.driver.find_element_by_xpath('/html/body/div[3]/div/div[2]/div/div/div[4]/div[1]/div[1]/div[1]/span[1]')
self.assertEqual(divText.text,'登录名或密码错误')
if __name__ == '__main__':
unittest.main()由于登录模块主要是⽤户名和密码的input表单的验证以及错误信息的验证，我们可以把用户名、密码、错误信息的验证参数化，具体实现的代码如下：
from selenium import webdriver
import unittest
import time as t
from parameterized import parameterized,param
class BaiduTest(unittest.TestCase):
def setUp(self) -> None: #前提
self.driver=webdriver.Chrome()
self.driver.get('https://mail.sina.com.cn/')
self.driver.maximize_window()
self.driver.implicitly_wait(30)
def tearDown(self) -> None: #清理
self.driver.quit()
@parameterized.expand([
param('','','请输入邮箱名'),
param('wertasd', 'asdf', '您输入的邮箱名格式不正确'),
param('[email protected]', 'asdf', '登录名或密码错误')
])
def test_sina_login(self,username,password,result):
self.driver.find_element_by_id('freename').send_keys(username)
t.sleep(3)
self.driver.find_element_by_id('freepassword').send_keys(password)
t.sleep(3)
self.driver.find_element_by_class_name('loginBtn').click()
t.sleep(3)
divText=self.driver.find_element_by_xpath('/html/body/div[3]/div/div[2]/div/div/div[4]/div[1]/div[1]/div[1]/span[1]')
self.assertEqual(divText.text,result)三、断言
assertEqual
assertEqual()是验证两个值相等，值的是数据类型与内容也是相等的,⻅案例代码：
from selenium import webdriver
import unittest
class BaiduTest(unittest.TestCase):
def setUp(self) -> None:
self.driver=webdriver.Chrome()
self.driver.get('http://www.baidu.com')
self.driver.maximize_window()
self.driver.implicitly_wait(30)
def tearDown(self) -> None:
self.driver.quit()
def test_baidu_title(self):
'''百度测试:验证百度首页的title'''
# assert self.driver.title=='百度一下，你就知道'
self.assertEqual(self.driver.title,'百度一下，你就知道')assertTrue
返回的是bool类型，也就是对被测试的对象进⾏验证，如果返回的是boolean类型并且是true，那么结果验证通过，那么⽅法assertFlase()验证的是被测试对象返回的内容是false,⻅案例代码：
from selenium import webdriver
import unittest
import time as t
class BaiduTest(unittest.TestCase):
def setUp(self) -> None: #前提
self.driver=webdriver.Chrome()
self.driver.get('https://mail.sina.com.cn/')
self.driver.maximize_window()
self.driver.implicitly_wait(30)
def tearDown(self) -> None: #清理
self.driver.quit()
def test_sina_isLogin(self):
'''sina邮箱验证，判断自动登录是否勾选'''
isLogin=self.driver.find_element_by_id('store1')
self.assertTrue(isLogin.is_selected())assertIn
assertIn()值的是⼀个值是否包含在另外⼀个值⾥⾯，在这⾥特别的强调⼀下，在assertIn()的⽅法⾥⾯，有两个参数，那么值的包含其实就是第⼆个实际参数包含第⼀个实际参数。与之相反的⽅法是assergNotIn(),⻅案例代码：
import unittest
from selenium import webdriver
class UiTest(unittest.TestCase):
def setUp(self) -> None:
self.driver=webdriver.Chrome()
self.driver.maximize_window()
self.driver.get('http://www.baidu.com')
self.driver.implicitly_wait(30)
def tearDown(self) -> None:
self.driver.quit()
def test_baidu_title_001(self):
self.assertIn('百度',self.driver.title)
def test_baidu_title_002(self):
self.assertIn('百度⼀下，你就知道',self.driver.title)
if __name__ == '__main__':
unittest.main()四、测试报告
在unittest的框架中，⽣成测试报告需要使⽤到HTMLTestRunner
import unittest
import os
from 单元测试框架.HTMLTestRunner import HTMLTestRunner #从HTMLTestRunner模块调用HTMLTestRunner类
def getTests():
'''加载所有的测试模块'''
suite=unittest.TestLoader().discover(
#找到被执行模块的路径
start_dir=os.path.dirname(__file__),
#加载路径下所有以test_开头的测试模块的文件
pattern='test_*.py' #正则表达式
)
return suite
def runSuite():
unittest.TextTestRunner().run(getTests())
def base_dir():
return os.path.dirname(os.path.dirname(__file__))#获取当前目录的上级目录
def run():
fp=open(os.path.join(base_dir(),'report','report.html'),'wb')#拼接report.html的路径 wb 二进制的方式写入
runner=HTMLTestRunner(
stream=fp, #流 执行一个写入一个
title='UI自动化测试报告',
description=''
)
runner.run(getTests())
if __name__ == '__main__':
run()
怎么解决每次生成的测试报告不覆盖之前的测试报告，同时被保留：
解决方法：引入时间库获取时间戳
代码如下：
import time
import unittest
import os
from HTMLTestRunner import HTMLTestRunner
def getTests():
'''加载所有的测试模块'''
suite=unittest.TestLoader().discover(
#找到被执行模块的路径
start_dir=os.path.dirname(__file__),
#加载路径下所有以test_开头的测试模块的文件
pattern='test_*.py' #正则表达式
)
return suite
def getNowTime():
return time.strftime('%y-%m-%d %H-%M-%S',time.localtime(time.time()))
def base_dir():
return os.path.dirname(os.path.dirname(__file__))
def run():
fp=open(os.path.join(base_dir(),'report',getNowTime()+'report.html'),'wb')
runner=HTMLTestRunner(
stream=fp,
title='UI自动化测试报告',
description=''
)
runner.run(getTests())
if __name__ == '__main__':
run()这样每次执行的都会生成一个测试报告：
最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走：
这些资料，对于【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴上万个测试工程师们走过最艰难的路程，希望也能帮助到你！
