UI测试时实现界面截图功能

selenium模块提供了截图函数get_screenshot_as_file()来截取当前窗口

# 定义截图函数 采取相对路径的方式将测试截图保存到.\reports\image\中
 from selenium import webdriver  #导入webdriver模块
 import os 
 def insert_img(driver,file_name):
       # 获取当前文件所在的绝对路径  E:/learn/erds_test/erds_ui/ui_test/models
    	base_dir=os.path.dirname(__file__) 
    	# 转换成str格式
    	base_dir=str(base_dir) 
    	# 以'/ui_test'将路径拆分两部分，获取第一部分：E:/learn/erds_test/erds_ui
    	base=base_dir.split('/ui_test')[0] 
    	# 设置截图保存的路径 E:/learn/erds_test/erds_ui/reports/image/+file_name
    	file_path=base+'/reports/image/'+file_name
    	# 截图
    	image=driver.get_screenshot_as_file(file_path) 
    
   if __name__ == '__main__':
   #打开浏览器，并截图
    	driver=webdriver.Chrome()
    	driver.get("https://rzrk.thinkfunds.cn:8001/login/") 
    	driver.maximize_window()
    	 #E:/learn/erds_test/erds_ui/reports/image/1.png
    	insert_img(driver,"1.png")
    	driver.quit()

运行程序：

打开1.png：