前一篇文章我们知道了,如何去封装几个简单的Selenium方法到我们自定义的类,这次我们编写一个类,叫浏览器引擎类,通过更改一个字符串的值,利用if语句去判断和控制启动那个浏览器。这里我们暂时,支持三大浏览器(IE,Chrome,Firefox)。这里有一个前提条件,在基础篇中,启动三大浏览器的driver文件,检查下你的Python安装路径下有没有这三个driver插件,如果没有,请回到基础篇的如何启动火狐和IE浏览器文章去看看如何做。 我们继续在test1这个包下新建一个browser_engine.py文件,然后在另外一个包下新建一个test.py文件去测试这个浏览器引擎类是否工作正常。这个浏览器引擎类,我们一开始写简单一点,只写启动浏览器。 先看看browser_engine.py中的代码: # coding=utf-8 from selenium import webdriver class BrowserEngine(object): ''' 定义一个浏览器引擎类,根据browser_type的值去,控制启动不同的浏览器,这里主要是IE,Firefox, Chrome ''' def __init__(self, driver): self.driver = driver browser_type = 'IE' # maybe Firefox, Chrome, IE def get_browser(self): ''' 通过if语句,来控制初始化不同浏览器的启动,默认是启动Chrome :return: driver ''' if self.browser_type == 'Firefox': driver = webdriver.Firefox elif self.browser_type == 'Chrome': driver = webdriver.Chrome elif self.browser_type == 'IE': driver = webdriver.Ie else: driver = webdriver.Chrome driver.maximize_window driver.implicitly_wait(10) return driver 再看看test.py代码,进行测试,更改browser_engine.py中browser_type的值,去测试三大浏览器是否启动正常。 # coding=utf-8 import time from test1.browser_engine import BrowserEngine class TestBrowserEngine(object): def open_browser(self): browserengine = BrowserEngine(self) driver = browserengine.get_browser tbe = TestBrowserEngine tbe.open_browser 目前,自定义的浏览器引擎类到这里就封装好了,只支持打开不同浏览器,需要手动修改,引擎类中browser_type的值。看起来功能简单,但是我们只是需要学习这种做事的方式和思维,在下一个部分,框架设计的时候,我会再告诉大家如何去加强这个引擎类的功能,到时候去修改配置文件中的浏览器类型,而不是修改代码中的字段。通过修改配置文件,从而去打开不同浏览器,并开始测试相关脚本。 |
|
来自: 昵称16619343 > 《办公技能》