Appium python+appium 在报错或者断言失败时怎么自动截图？

usky · 2017年06月01日 · 最后由哇了个哇回复于 2021年08月20日 · 2617 次阅读

在做自动化脚本开发，使用的 unittest 框架，只想在断言失败的时候截图。下面这个例子无论成功或失败都会触发指定的函数，请问怎么修改，只在失败的时候触发函数呢？

import unittest
class loginTests(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        pass

    @classmethod
    def tearDownClass(cls):
        pass

    def aaa(self):
        print('触发函数')

    def test1(self):
        i =2
        self.assertEqual(i,1,self.aaa())

if __name__ == '__main__':
    unittest.main()

2 个赞

共收到 6 条回复时间点赞

匿名 #7 · 2017年06月02日 1 个赞

try:
     self.assertTrue(self.driver.find_element_by_name(u'xxxxxx'))
except Exception as msg:
      print (msg)
      public.screenshot("error.png")

其中断言部分可根据实际情况修改
截图部分也需要自己修改，我这段是调用的公共方法

五棵木 #2 · 2017年06月02日 3 个赞

可以使用装饰器：

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)