前一篇文章我們知道了,如何去封裝幾個簡單的Selenium方法到我們自定義的類,這次我們編寫一個類,叫瀏覽器引擎類,通過更改一個字符串的值,利用if語句去判斷和控制啟動那個瀏覽器。這里我們暫時,支持三大瀏覽器(IE,Chrome,Firefox)。這里有一個前提條件,在基礎(chǔ)篇中,啟動三大瀏覽器的driver文件,檢查下你的Python安裝路徑下有沒有這三個driver插件,如果沒有,請回到基礎(chǔ)篇的如何啟動火狐和IE瀏覽器文章去看看如何做。
我們繼續(xù)在test1這個包下新建一個browser_engine.py文件,然后在另外一個包下新建一個test.py文件去測試這個瀏覽器引擎類是否工作正常。這個瀏覽器引擎類,我們一開始寫簡單一點,只寫啟動瀏覽器。
先看看browser_engine.py中的代碼:
# coding=utf-8
from selenium import webdriver
class BrowserEngine(object):
'''
定義一個瀏覽器引擎類,根據(jù)browser_type的值去,控制啟動不同的瀏覽器,這里主要是IE,F(xiàn)irefox, Chrome
'''
def __init__(self, driver):
self.driver = driver
browser_type = 'IE' # maybe Firefox, Chrome, IE
def get_browser(self):
'''
通過if語句,來控制初始化不同瀏覽器的啟動,默認(rèn)是啟動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的值??雌饋砉δ芎唵?,但是我們只是需要學(xué)習(xí)這種做事的方式和思維,在下一個部分,框架設(shè)計的時候,我會再告訴大家如何去加強這個引擎類的功能,到時候去修改配置文件中的瀏覽器類型,而不是修改代碼中的字段。通過修改配置文件,從而去打開不同瀏覽器,并開始測試相關(guān)腳本。