python自动化测试截图装饰器的使用

最新推荐文章于 2024-07-29 19:27:43 发布

转载最新推荐文章于 2024-07-29 19:27:43 发布 · 1k 阅读

文章标签：

#python #自动化测试 #截图 #装饰器

python自动化测试专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种在Python单元测试中使用装饰器来捕获错误并自动生成截图的方法，通过装饰器增强测试用例，当测试失败时会自动保存当前页面的截图，便于后续的问题定位。

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

装饰器：

def getImage(function):
    @wraps(function)
    def get_ErrImage(self,*args, **kwargs):      
        try:
            result = function(self,*args, **kwargs)
        except:
            timestr = time.strftime("%Y-%m-%d_%H_%M_%S")
            self.dr.get_screenshot_as_file('E:\\Appium\\Mailchat\\image\\getscreen\\%s %s.png' %
                                           (function.__name__timestr))
        else:
            logging.info (" %s 脚本运行正常" %
                (function.__name__)
                )
        return result
    return get_ErrImage

如何使用：

class myInitiate_addOppionTest(unittest.TestCase):

    def setUp(self):
        public.setUp.setUp(self)

    def tearDown(self):
        self.dr.quit()

    @getImage
    def test_myInitiate_addOppion(self):
        go_loginPage(self)   #访问登录页

if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(myInitiate_addOppionTest)
    unittest.TextTestRunner(verbosity=2).run(suite)