Python3 測試報告BeautifulReport中添加截圖

在測試類中，添加save_img方法，在測試過程中出現錯誤時，自動截圖並返回失敗

默認存放的圖片路徑是img

def save_img(self, img_name):
    """
        傳入一個img_name, 並存儲到默認的文件路徑下
    :param img_name:
    :return:
    """
    self.driver.get_screenshot_as_file('{}/{}.png'.format(os.path.abspath(self.img_path), img_name))

調用時，使用如下：

@BeautifulReport.add_test_img('test_errors_save_imgs')
def test_errors_save_imgs(self):
    """
        如果在測試過程中, 出現不確定的錯誤, 程序會自動截圖, 並返回失敗, 如果你需要程序自動截圖, 則需要咋測試類中定義 save_img方法
    """
    self.driver.find_element_by_xpath('//abc')

 

效果：