首页 > 代码库 > Selenium2+python自动化60-异常后截图(screenshot)

Selenium2+python自动化60-异常后截图(screenshot)

前言

在执行用例过程中由于是无人值守的,用例运行报错的时候,我们希望能对当前屏幕截图,留下证据。

在写用例的时候,最后一步是断言,可以把截图的动作放在断言这里,那么如何在断言失败后截图呢?

 

一、截图方法

1.get_screenshot_as_file(self, filename)

--这个方法是获取当前window的截图,出现IOError时候返回False,截图成功返回True。

filename参数是保存文件的路径。
Usage:
driver.get_screenshot_as_file(‘/Screenshots/foo.png‘)

 

2.get_screenshot_as_base64(self)

--这个方法也是获取屏幕截图,保存的是base64的编码格式,在HTML界面输出截图的时候,会用到。

比如,想把截图放到html测试报告里。
Usage:
driver.get_screenshot_as_base64()

 

3.get_screenshot_as_png(self)
--这个是获取屏幕截图,保存的是二进制数据,很少用到.
Usage:
driver.get_screenshot_as_png()

 

二、异常后截图

1.为了能抛异常,把定位登录按钮的id换了个错的id。

2.给图片命名时候加个时间戳,避免同一个文件名称被覆盖掉。

3.文件路径,这里直接写的文件名称,就是跟当前的脚本同一个路径。如果图片输出到其它文件路径,需要些文件的绝对路径了。

4.截图的结果,如果没截到图返回False,截图成功会返回True。

技术分享

 

三、selenium实例

1.在unittest框架里写用例的时候,我们希望在断言失败的时候,对当前屏幕截图。

2.如果加try...except捕获异常后结果,此时所有的测试用例都是通过的了,会影响测试结果。解决办法其实很简单,再把异常抛出来就行了。

3.参考代码:

# coding:utf-8
from selenium import webdriver
import time,unittest
from selenium.webdriver.support import expected_conditions as EC
class Login(unittest.TestCase):
    def setUp(self):
        url_login = "https://passport.cnblogs.com/user/signin"
        self.driver = webdriver.Firefox()
        self.driver.get(url_login)

    def test_01(self):
        ‘‘‘前面输入账号密码,让正确运行到assert这一步,断言故意设置为False不成功‘‘‘
        try:
            self.driver.find_element_by_id("input1").send_keys(u"上海-悠悠")
            self.driver.find_element_by_id("input2").send_keys("xxx")
            # 登录id是错的,定位会抛异常
            self.driver.find_element_by_id("signin").click()
            # 判断登录成功页面是否有账号:"上海-悠悠"
            time.sleep(3)
            locator = ("id", "lnk_current_user")
            result = EC.text_to_be_present_in_element(locator,u"上海-悠悠")(self.driver)
            self.assertFalse(result)
        except Exception as msg:
            print(u"异常原因%s"%msg)
            # 图片名称可以加个时间戳
            nowTime = time.strftime("%Y%m%d.%H.%M.%S")
            self.driver.get_screenshot_as_file(‘%s.jpg‘ % nowTime)
            raise

    def tearDown(self):
        pass

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

4.运行结果:

异常原因True is not false

Failure
Traceback (most recent call last):
  File "D:\test\yoyot\ketang\test01.py", line 22, in test_01
    self.assertFalse(result)
AssertionError: True is not false

 

学习过程中有遇到疑问的,可以加selenium(python+java) QQ群交流:232607095

觉得对你有帮助,就在右下角点个赞吧,感谢支持!

Selenium2+python自动化60-异常后截图(screenshot)