我一直在思考第一个博客应该写什么,然后我就解决了开通博客后解决的第一个问题,择题不如撞题;
如果大多数人和我一样,接触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里的代码进行这样改(请看注释):
1 dr = webdriver.Firefox()#定义全局的dr,不在setUp函数里,实现了只启动一次firefox
2 def setUp(self,driver= dr):
3 self.driver = driver#初始化数据中,将全局变量dr赋值给self.driver,这样每次测试开始都只会锁定dr,而不是新创建一个
4 self.driver.implicitly_wait(30)
5 self.base_url = "http://192.168.1.207:8081"
6 self.verificationErrors = []
7 self.accept_next_alert = True
8
9 def test_login(self):
10 driver = self.driver
11 driver.get(self.base_url + "/RchWebSchool/")
12 driver.find_element_by_css_selector("span").click()
13 driver.find_element_by_name("j_password").clear()
14 driver.find_element_by_name("j_password").send_keys("admin")
15 driver.find_element_by_name("j_username").clear()
16 driver.find_element_by_name("j_username").send_keys("admin")
17 driver.find_element_by_xpath("//button[@type='submit']").click()
18
19 def test_login_reg(self):
20 driver = self.driver
21 #driver.get(self.base_url + "/RchWebSchool/")
22 ...
23 #修改个人信息自动化代码
24 ...
25
26 def test_zzz_quit(self):#最后一个测试用例,实现了只进行一次退出浏览器
27 self.driver.quit()
28
29
30 def tearDown(self):
31 try:
32 self.driver.refresh()#将退出浏览器的操作变成刷新浏览器,用于不同用例之间的接洽操作
33 except ConnectionRefusedError as e:
34 print(e)
35 finally:
36 self.assertEqual([], self.verificationErrors)
大家应该看到,阳阳把tearown(self)里面的代码已经改的面目全非了,这是因为我们每个测试用例(test_xxx)执行之后,都会走一次tearown,最后一个退出浏览器的test_zzz_quit当然也不例外,所以在走完test_zzz_quit之后,一定会再次执行刷新浏览器的操作,但是此前浏览器已经被关闭了,刷新浏览器一定会报错,所以把异常ConnectionRefusedError抛出,不再进行处理,视为测试用例通过;
当然,如果自动化只到这里的话,大家也可把teardown干掉,把每次刷新浏览器的操作放在每个测试用例的最后面,都是可以的,我只是习惯性保持unittest的完整性;