unittest 装饰器 @classmethod

最新推荐文章于 2024-02-18 17:16:22 发布

原创最新推荐文章于 2024-02-18 17:16:22 发布 · 929 阅读

1 ·

CC 4.0 BY-SA版权

selenium基础专栏收录该内容

9 篇文章

订阅专栏

本文讲解了如何使用Python的unittest框架中的装饰器（@classmethod）来优化测试用例的执行效率，通过只在所有测试用例开始前和结束后各执行一次setUpClass和tearDownClass方法，避免了重复打开和关闭浏览器等耗时操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前面讲到unittest里面setUp可以在每次执行用例前执行，这样有效的减少了代码量，但是有个弊端，比如打开浏览器操作，每次执行用例时候都会重新打开，这样就会浪费很多时间。
于是就想是不是可以只打开一次浏览器，执行完用例再关闭呢？这就需要用到装饰器（@classmethod）来解决了。
一、装饰器
1.用setUp与setUpClass区别

setup():每个测试case运行前运行
teardown():每个测试case运行完后执行
setUpClass():必须使用@classmethod 装饰器,所有case运行前只运行一次
tearDownClass():必须使用@classmethod装饰器,所有case运行完后只运行一次

2.@是修饰符，classmethod是python里的类方法

import unittest
import time

class Test(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        print("start!")

    @classmethod
    def tearDownClass(cls):
        time.sleep(1)
        print("end!")

    def testA(self):
        print("测试用例1")
    def testC(self):
        print("测试用例3")
    def testB(self):
        print("测试用例2")
if __name__ == "__main__":
    unittest.main()