selenium中的备胎
漫漫人生路，总会错几步。在跑selenium的时候，或多或少，总有那么几次会fail掉。如同我们汽车有备胎，selenium中我们也可以弄个备胎。
在日常的selenium的自动化测试中，我们总能得到这样的error.
raise TimeoutException(message, screen, stacktrace)
selenium.common.exceptions.TimeoutException: Message: None基本上就是超时的error. 如果页面超时，会导致后面的所有都不执行。 selenium是单进程在执行，我们考虑搞一个备用的进程，如果主进程死了，就可以切换过来，如同我们生活中的备胎。 这样就不需要重新启动selenium。
前面我们用过retry, 是需要重新启动的。这个不需要重启，就是耗费了一点点资源，可以忽略不计。 原理很简单，就是我们备用一个windows handle, 如果异常了，我们就切换过去，重来一遍。
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import traceback
url = "http://new/"
chrome_options = Options()
prefs = {
'profile.default_content_setting_values': {
'images': 2, # 禁用图片的加载
'javascript': 2 ##禁用js，可能会导致通过js加载的互动数抓取失效
}
}
chrome_options.add_experimental_option("prefs", prefs)
# chrome_options.add_argument("--headless") # 不弹出浏览器
browser = webdriver.Chrome(chrome_options=chrome_options)
browser.implicitly_wait(5) # 操作、获取元素时的隐式等待时间
browser.set_page_load_timeout(10) # 页面加载超时等待时间
main_win = browser.current_window_handle # 记录当前窗口的句柄
all_win = browser.window_handles
try:
if len(all_win) == 1:
print('开启备胎')
js = 'window.open("https://www.baidu.com");'
browser.execute_script(js) # 开启保护
all_win = browser.window_handles
browser.get(url) # 此处访问你需要的URL
except:
# 超时
print('Time out')
if 'time' in str(traceback.format_exc()):
# 切换新的浏览器窗口
for win in all_win:
if main_win != win:
print('WIN', win, 'Main', main_win)
print('切换到备胎')
browser.close()
browser.switch_to.window(win)
main_win = win
browser.get(url) # 此处访问你需要的URL
browser.find_element_by_id('new-activate-btn').click() #后续操作测试: 我们可以把需要打开的页面，用抓包工具给断点着，制造超时的场景。 测试通过。
这如同我们生活中，如果上天再给一次机会，我一定会好好争取。
这样我们的测试就稳定多了，如果2次都失败，也就是备胎都不起作用，那也就无力回天，得考虑优化代码了。
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2019-04-22，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读