【selenium】python+selenium+unittest，关于每次执行完一个测试用例都关闭浏览器等时间较长的问题之解决方案

　我一直在思考第一个博客应该写什么，然后我就解决了开通博客后解决的第一个问题，择题不如撞题；

　　如果大多数人和我一样，接触python+selenium+unittest是从selenium IDE开始的话，你也一定会遇到这样的问题：

　　我们写了5个，10个，甚至20个测试用例，放在一个py脚本里，每个测试用例执行完毕之后，都会走一遍退出浏览器的操作，然后再启动浏览器，再退出，如此反复，浪费了大量的时间，今天就说说我研究解决这个问题的历程；

　　我们第一个录制并导出的python selenium代码，万能的“登录功能”，是这个样子的：

# -*- coding: utf-8 -*-
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re

class Login(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://192.168.1.207:8080"
        self.verificationErrors = []
        self.accept_next_alert = True

    def test_login(self):
        driver = self.driver
        driver.get(self.base_url + "/QAS/RchWebSchool/")
        driver.find_element_by_css_selector("span").click()
        driver.find_element_by_name("j_password").clear()
        driver.find_element_by_name("j_password").send_keys("admin")
        driver.find_element_by_name("j_username").clear()
        driver.find_element_by_name("j_username").send_keys("admin")
        driver.find_element_by_xpath("//button[@type='submit']").click()

    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException as e: return False
        return True

    def is_alert_present(self):
        try: self.driver.switch_to_alert()
        except NoAlertPresentException as e: return False
        return True

    def close_alert_and_get_its_text(self):
        try:
            alert = self.driver.switch_to_alert()
            alert_text = alert.text
            if self.accept_next_alert:
                alert.accept()
            else:
                alert.dismiss()
            return alert_text
        finally: self.accept_next_alert = True

    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

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

　　往往我们很容易读懂这三个def（如果读不懂，请看虫师先生的博客：selenium-webdriver(python) (十六) --unittest 框架）：

　　　　setUp(self)

　　　　test_login(self)

　　　　tearDown(self)

　　读懂之后，我们就可以写另外一个功能“修改个人信息”的自动化脚本了，假设我们在class：Login 里是这样写的：

    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://192.168.1.207:8080"
        self.verificationErrors = []
        self.accept_next_alert = True

    def test_login(self):
        driver = self.driver
        driver.get(self.base_url + "/QAS/RchWebSchool/")
        driver.find_element_by_css_selector("span").click()
        driver.find_element_by_name("j_password").clear()
        driver.find_element_by_name("j_password").send_keys("admin")
        driver.find_element_by_name("j_username").clear()
        driver.find_element_by_name("j_username").send_keys("admin")
        driver.find_element_by_xpath("//button[@type='submit']").click()
　　　　
    def test_login_reg(self):
        driver = self.driver
        driver.get(self.base_url + "/QAS/RchWebSchool/")

        #登录代码
        ...
        #修改个人信息代码
        ...


    def tearDown(self):
        self.driver.quit()
        self.assertEqual([], self.verificationErrors)

　　我们发现，每增加一个测试用例，都增加一次操作：启动浏览器（setUp）→登录测试网址（#登录代码）→关闭浏览器（teardown），而当用例过多的时候，如此反复的操作会浪费大量的时间；

　　怎么办呢？　　

　　如果我们能只走一次启动浏览器和退出浏览器，那该有多好啊；

　　我们尝试将所有的功能的操作都写在一个test里面，我们的test_login(self)代码就变成了这样：

    def test_login(self):
        driver = self.driver
        driver.get(self.base_url + "/QAS/RchWebSchool/")
        driver.find_element_by_css_selector("span").click()
        driver.find_element_by_name("j_password").clear()
        driver.find_element_by_name("j_password").send_keys("admin")
        driver.find_element_by_name("j_username").clear()
        driver.find_element_by_name("j_username").send_keys("admin")
        driver.find_element_by_xpath("//button[@type='submit']").click()
        
        
        #修改个人信息
        ...
        #功能续3
        ...
        #功能续4
        ...

　　但是，这样就了一个新的问题，当我们只测试其中一个功能，或者用到其中某一个功能的时候，没有办法单独调用某一个功能的代码，失去了原本的灵活性；

　　所以阳阳针对class：Login里的代码进行这样改（请看注释）：

    dr = webdriver.Firefox()#定义全局的dr，不在setUp函数里，实现了只启动一次firefox
    def setUp(self,driver= dr):
        self.driver = driver#初始化数据中，将全局变量dr赋值给self.driver,这样每次测试开始都只会锁定dr，而不是新创建一个
        self.driver.implicitly_wait(30)
        self.base_url = "http://192.168.1.207:8081"
        self.verificationErrors = []
        self.accept_next_alert = True

    def test_login(self):
        driver = self.driver
        driver.get(self.base_url + "/RchWebSchool/")
        driver.find_element_by_css_selector("span").click()
        driver.find_element_by_name("j_password").clear()
        driver.find_element_by_name("j_password").send_keys("admin")
        driver.find_element_by_name("j_username").clear()
        driver.find_element_by_name("j_username").send_keys("admin")
        driver.find_element_by_xpath("//button[@type='submit']").click()

    def test_login_reg(self):
        driver = self.driver
        #driver.get(self.base_url + "/RchWebSchool/")
        ...
        #修改个人信息自动化代码
        ...

    def test_zzz_quit(self):#最后一个测试用例，实现了只进行一次退出浏览器
        self.driver.quit()


    def tearDown(self):
        try:
            self.driver.refresh()#将退出浏览器的操作变成刷新浏览器，用于不同用例之间的接洽操作
        except ConnectionRefusedError as e:
            print(e)
        finally:
            self.assertEqual([], self.verificationErrors)

　　大家应该看到，阳阳把tearown(self)里面的代码已经改的面目全非了，这是因为我们每个测试用例（test_xxx）执行之后，都会走一次tearown，最后一个退出浏览器的test_zzz_quit当然也不例外，所以在走完test_zzz_quit之后，一定会再次执行刷新浏览器的操作，但是此前浏览器已经被关闭了，刷新浏览器一定会报错，所以把异常ConnectionRefusedError抛出，不再进行处理，视为测试用例通过；

　　当然，如果自动化只到这里的话，大家也可把teardown干掉，把每次刷新浏览器的操作放在每个测试用例的最后面，都是可以的，我只是习惯性保持unittest的完整性；