写个python+selenium 的自动化测试批量执行多个测试case文件用例截屏

最近一直面试换工作，闲来无聊写写。

思路：

1、查找目标目录下的以test* 为开头的case文件

2、通过run_all_case.py 来执行全部测试用例

3、每一个用例执行完成后截屏一个图片文件保存在文件夹中

具体实现，多个测试用例文件，举例：login，personal（登录case，和 登录后进入个人中心）

新建三个文件分别为：test_login_case.py    test_personalCenter_case.py  run_all_case

好，下面分别写下三个文件的code， 俩个测试case我都是随便写写 没写那么完整的case其实跟功能设计用例一个道理。

self.web.get_screenshot_as_file("D:\\result\\login\\test_login_user_null.png") # 注意文件保存格式为 png

代码中我会用到这个函数来截屏并保存到相应目录下作为提bug的依据。

路径你们可以用相对路径用os包 os.path 自己去查，我写的绝对路径。

截屏很重要，我觉得比输出纯文字的html报告要好点，一般跑完后我还是会在缺陷管理工具中提交bug，看个人喜好。

第一个：test_login_case.py

#coding:utf-8
from selenium import webdriver
import unittest
import time

class Login_case(unittest.TestCase):
    def setUp(self):
        print('开始测试')
    def tearDown(self):
        print('完成测试')
        self.web.quit()
        # 定义登录方法 主体
    def login(self, username, password):
        self.web = webdriver.Chrome()
        self.web.get('https://')  # 登录页面 例如https://www.baidu.com
        self.web.maximize_window()
        time.sleep(5)
        # 定位元素 打开登录页根据网站实际操作去定位登录元素
        self.web.find_element_by_class_name('register_login_btn').click() 
        self.web.fi