WebUI自动化中截图的使用

前言

做UI自动化时，有时候可能因为需要把某个步骤的界面显示截图保存，这里我用到的是webdriver的截图功能。

环境搭建

这里说的UI自动化指的是 web的UI自动化，使用的是Python+Selenium+webdriver写的，用Unittest框架来组织用例和脚本。具体环境搭建可以参考之前的环境搭建详细文章：
https://blog.csdn.net/NoamaNelson/article/details/102971936

封装截图功能creenShot.py

driver：指的是使用哪个浏览器的驱动，我后边使用的是Chrome

# coding=utf-8

import time
from selenium import webdriver

def sav_creenshot(driver):
    now=time.strftime("%Y-%m-%d-%H_%M_%S",time.localtime(time.time())) # 截图保存的文件名格式
    pic_path = "./"+now+'_screen.png' # 截图保存的路径
    # print(pic_path)
    driver.save_screenshot(pic_path) # 调用Driver的截图保存功能

示例test_baidu.py

写一个百度搜索hello的实例，test_baidu.py

# coding=utf-8
from selenium import webdriver
import time
import unittest
import logging
from creenShot import sav_creenshot # 导入刚刚封装的截图方法

class Testbaidu(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome() # 指定浏览器驱动
        self.url = "https://www.baidu.com/" # URL地址
        self.driver.maximize_window()  # 最大化窗口
        self.driver.get(self.url) # 获取访问网址
        self.log = logging.getLogger()  # 记录日志

    def tearDown(self):
        self.driver.close() # 关闭浏览器

    def test_search(self):
        # 搜索框输入“hello”
        self.log.info("输入框输入hello")
        self.driver.find_element_by_id("kw").send_keys("hello")
        self.driver.find_element_by_id("su").click()
        time.sleep(2)
        sav_creenshot(self.driver)  # 截图
        self.assertIn("hello", self.driver.page_source)
        self.log.info("搜索成功")

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