快速搞定selenium grid分布式
写这篇文章,似乎有点重复造轮子的嫌疑。当看了几篇相关文章后,我还是决定把半年前的半成品给完成了。
以传统的方式部署分布式Selenium Grid集群需要耗费大量时间和机器成本来准备测试环境。
Snake用docker来搭建Selenium grid环境,用ptest框架来写代码,尝试简化这个过程。
NO.1 搭环境
首先你得把docker加个阿里云加速器,不然那个等待是漫长的。
写个docker-compose.yml
hub:
image: selenium/hub
ports:
- "4444:4444"
firefox:
image: selenium/node-firefox
ports:
- 5901:5900
links:
- hub
chrome:
image: selenium/node-chrome
ports:
- 5902:5900
links:
- hub
启动起来:
docker-compose up
看起来是这样的
启动成功。
环境部分,就大功告成了。So easy!
NO.2 写代码
写了一个很简单的例子,用了同事写的ptest框架。
这个框架改进了很多市面上框架用起来不爽的地方,比如:报告,并发,写log, 传参等。
__author__ = 'anderson'
# coding=utf-8
from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
from selenium.webdriver.support.ui import WebDriverWait
from ptest.decorator import TestClass, Test, BeforeMethod, AfterMethod
@TestClass(run_mode="parallel") # the test cases in this class will be executed by multiple threads
class SeleniumClass:
@BeforeMethod(description="Prepare test Enviroment.")
def before(self):
preporter.info("set up driver")
browser = config.get_property("browser")
if browser == "firefox":
self.driver = webdriver.Remote(command_executor='http://127.0.0.1:4444/wd/hub',
desired_capabilities=DesiredCapabilities.FIREFOX)
elif browser == "chrome":
self.driver = webdriver.Remote(command_executor='http://127.0.0.1:4444/wd/hub',
desired_capabilities=DesiredCapabilities.CHROME)
@Test(tags=["regression", "smoke"])
def test1(self):
preporter.info("start to test")
self.driver.get("http://www.baidu.com")
inputElement = self.driver.find_element_by_name("wd")
inputElement.send_keys("docker")
inputElement.submit()
WebDriverWait(self.driver, 20).until(lambda driver: driver.title.startswith("docker"))
assert_true(self.driver.title.startswith("docker"))
@AfterMethod(always_run=True, description="Clean up")
def after(self):
preporter.info("cleaning up")
self.driver.quit()
运行,得到结果:
Oopos, 出错了!
仔细看log和截图,原来是没有支持中文。
解决方案是: 换docker image 源,搜到了支持中文的。
重启后再运行。
Nice!
NO.3 并发
其实第二步里面已经打开了并发。运行的时候,输入并发数就可以了。
ptest3 -t test -n 2 -Dbrowser=firefox
这里的n是并发数,参数是firefox. 让firefox做为浏览器。(可以根据case 数量来设定。也可以设置多种浏览器)
当然,也可以在yml里面多注册几个hub也能实现并发, up to you!
感想
其实在这里,我都没做啥,只是做了个搬运工而已,报告,截图,并发,重跑等等,我都不用操心,只要设置几个参数就行。好的框架,能事半功倍。
更多精彩,请关注微信公众号: python爱好部落