写在前面:经过了实践总结一下经验,心得进行一个分享。
首先driver是可以单独抽出来的,变成一个driver函数放在driver.py。
from selenium import webdriver from selenium.webdriver.chrome.service import Service from selenium.webdriver.chrome.options import Options def get_chrome_driver(): # 启动driver options = Options() # 替换为你本地的Chromed路径 options.binary_location = r"D:\software\Chrome-bin\chrome.exe" # 替换为你本地的Chromedriver路径 chromedriver_path = r'D:\software\Chrome-bin\chromedriver.exe' # 使用Service对象指定Chromedriver路径 service = Service(chromedriver_path) driver = webdriver.Chrome(service=service, options=options) return driver
然后我们新建page.py,在这里我们只处理页面的对象,因此我们创建页面类在此,当然这个页面类需要和HTML元素和驱动进行组合才能形成一个页面。
我们创建BasePage,以后的页面类都用它作为基类。首先是传一个驱动,组合起来
class BasePage(object): """Base class to initialize the base page that will be called from all pages""" def __init__(self, driver): self.driver = driver class LoginPage(BasePage): def __init__(self, driver): super().__init__(driver)
接着我们在element.py创建元素的基类,以后的元素类都继承自它
from selenium.webdriver.support.ui import WebDriverWait class BaseElement(object): """Base page class that is initialized on every page object class.""" locator = None def __set__(self, obj, value): """Sets the text to the value supplied""" driver = obj.driver WebDriverWait(driver, 5).until(lambda dri: dri.find_element(*self.locator)) driver.find_element(*self.locator).clear() driver.find_element(*self.locator).send_keys(value) def __get__(self, obj, owner): """Gets the specified element object""" driver = obj.driver WebDriverWait(driver, 5).until(lambda dri: dri.find_element(*self.locator)) self.element = driver.find_element(*self.locator) return self.element def click(self): self.element.click() def get_text(self): return self.element.get_attribute("value")
接着我们在locator.py创建定位器类,没什么特别,就是作为一个类似字典的静态类存放元素的位置
from selenium.webdriver.common.by import By class LonginPageLocators(object): """A class for login page locators. All login page locators should come here""" basic_username = (By.ID, 'basic_username') basic_password = (By.ID, 'basic_password') details_button = (By.ID, 'details-button') proceed_link = (By.ID, 'proceed-link') login_button = (By.ID, 'submitBtn')
准备工作就绪,那么我们来编写LoginPage吧~
首先组合我们需要的HTML元素,然后编写一个login业务函数进行业务处理。
class BasePage(object): """Base class to initialize the base page that will be called from all pages""" def __init__(self, driver): self.driver = driver class LoginPage(BasePage): def __init__(self, driver): super().__init__(driver) class DetailButtonElement(BaseElement): locator = LonginPageLocators.details_button class ProceedLinkElement(BaseElement): locator = LonginPageLocators.proceed_link class UsernameTextElement(BaseElement): locator = LonginPageLocators.basic_username class PasswordTextElement(BaseElement): locator = LonginPageLocators.basic_password class LoginButtonElement(BaseElement): locator = LonginPageLocators.login_button class RankBoxElement(BaseElement): locator = LonginPageLocators.rank_box detail_button_ele = DetailButtonElement() proceed_link_ele = ProceedLinkElement() username_text_ele = UsernameTextElement() password_text_ele = PasswordTextElement() login_button_ele = LoginButtonElement() rank_box_ele = RankBoxElement() def login(self): self.detail_button_ele.click() self.proceed_link_ele.click() self.username_text_ele = 'admin' self.password_text_ele = '123456' self.login_button_ele.click() _ = self.rank_box_ele # just test web is complete
_ = self.rank_box_ele小细节是用来测试是否进入主页面的。
回到单元测试.py添加测试函数,debug运行。
def test_login(self): login_page = page.LoginPage(self.driver) login_page.login()
当我们登录成功后,我们就可以跳转到我们需要页面,进行脚本处理。当然了进入了新的页面,别忘了创建新的页面类。在我的例子中就是创建了新的页面类ResourceDetailPage以便完成后续的业务。
def test_turn_to_resource(self): self.test_login() # 页面跳转 self.driver.get("https://127.0.0.1:38000/#/resource/detail/3668") def test_resource_delete(self): self.test_turn_to_resource() # 创建新的page对象 resource_detail_page = page.ResourceDetailPage(self.driver) # 在新的page处理业务... resource_detail_page.while_delete()