Python unittest装饰器实现原理及代码

时间:2021-05-22

1、前言

前面讲到unittest里面setUp可以在每次执行用例前执行,这样有效的减少了代码量,但是有个弊端,比如打开浏览器操作,每次执行用例时候都会重新打开,这样就会浪费很多时间。

于是就想是不是可以只打开一次浏览器,执行完用例再关闭呢?这就需要用到装饰器(@classmethod)来解决了。

2、@classmethod 装饰器中的 setUpClass() 和 tearDownClass() 与 setUp() 和 tearDown() 的区别

  • setUp():每个测试 case 运行之前运行
  • tearDown():每个测试 case 运行完之后执行
  • setUpClass():必须使用 @classmethod 装饰器, 所有 case 运行之前只运行一次
    • @是修饰符,classmethod 是 python 里的类方法
  • tearDownClass():必须使用 @classmethod 装饰器, 所有 case 运行完之后只运行一次
import unittestclass Test(unittest.TestCase): @classmethod def setUpClass(cls) -> None: print("-----开始执行用例-----") def setUp(self) -> None: print("执行用例准备动作...") def test01(self): print("第一条用例正在执行...") def test02(self): print("第二条用例正在执行...") def tearDown(self) -> None: print("用例执行完结动作...") @classmethod def tearDownClass(cls) -> None: print("-----用例执行结束-----") if __name__ == '__main__': unittest.main()#执行结果如下:"""-----开始执行用例-----执行用例准备动作...第一条用例正在执行...用例执行完结动作...执行用例准备动作...第二条用例正在执行...用例执行完结动作...-----用例执行结束-----"""
from selenium import webdriverimport unittestfrom time import sleepclass DymLogin(unittest.TestCase): @classmethod def setUpClass(cls) -> None: """ 所有的用例只需要执行一次打开浏览器操作 """ print("Test start...") cls.dr = webdriver.Chrome() cls.dr.get("http:///") cls.dr.implicitly_wait(10) cls.dr.maximize_window() cls.dr.find_element_by_class_name("login").click() sleep(1) @classmethod def tearDownClass(cls) -> None: sleep(2) cls.dr.quit() print("Test end...") def test_login01(self): """ 账号密码都为空 """ self.dr.find_element_by_name("loginName").clear() self.dr.find_element_by_name("loginPwd").clear() self.dr.find_element_by_xpath('//*[@id="loginForm"]/div/div/div/div[2]/div[1]/div[2]').click() #登录操作 TipsMsg = self.dr.find_element_by_class_name("promptLogin").text self.assertEqual(TipsMsg,"-请输入登录账号!",msg="Test_login01 login exception") def test_login02(self): """ 账号正确,密码为空 """ self.dr.find_element_by_name("loginName").clear() self.dr.find_element_by_name("loginPwd").clear() self.dr.find_element_by_name("loginName").send_keys("17779828887") self.dr.find_element_by_xpath('//*[@id="loginForm"]/div/div/div/div[2]/div[1]/div[2]').click() # 登录操作 TipsMsg = self.dr.find_element_by_class_name("promptLogin").text self.assertEqual(TipsMsg, "-请输入密码!", msg="Test_login02 login exception") def test_login03(self): """ 账号密码都正确 """ self.dr.find_element_by_name("loginName").clear() self.dr.find_element_by_name("loginPwd").clear() self.dr.find_element_by_name("loginName").send_keys("17779828887") self.dr.find_element_by_name("loginPwd").send_keys("zy295240???") self.dr.find_element_by_xpath('//*[@id="loginForm"]/div/div/div/div[2]/div[1]/div[2]').click() #登录操作 sleep(2) TipsMsg = self.dr.find_element_by_css_selector("#w_nav_temp_1000 > div.fr.dl > a:nth-child(4)").text self.assertEqual(TipsMsg,"[退出]",msg="Test_login03 login exception")if __name__ == '__main__': unittest.main()#执行结果如下"""Test start......Test end...----------------------------------------------------------------------Ran 3 tests in 25.253sOK"""

3、通过装饰器跳过某条用例

添加装饰器(@unittest.skip(""))

import unittestclass Test(unittest.TestCase): def setUp(self) -> None: print("---start---") def tearDown(self) -> None: print("---end---") def testbbb(self): print("case testbbb") @unittest.skip("testaaa 被跳过") def testaaa(self): print("case testaaa")if __name__ == '__main__': unittest.main()# 结果如下s.----------------------------------------------------------------------Ran 2 tests in 0.000sOK (skipped=1)---start---case testbbb---end---

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章