Web自動化測試項目（二）BasePage實現

一、BasePage介紹

1. 創建一個BasePage類，對Selenium Api進行二次封裝
2. 爲了快速創建項目並投產，用到的Selenium Api才進行封裝，沒用到的則不封裝
3. 優先封裝最重要的幾個方法：
   • 初始化webdriver
   • 元素定位（不封裝元素操作方法，例如send_keys）
   • 瀏覽器加載頁面

二、Selenium Api 二次封裝

├── pages
│   ├── __init__.py
│   └── base_page.py
└── utils
    └── __init__.py

base_page.py

from selenium import webdriver
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

DOMAIN = 'https://www.tapd.cn/'
TIMEOUT = 10
POLL_FREQUENCY = 0.5


class BasePage():

    def __init__(self, driver=None, path=None):
        self.driver = driver if driver != None else webdriver.Chrome()
        self.get_url(path)

    def find_element(self, locator):
        WebDriverWait(driver=self.driver, timeout=TIMEOUT, poll_frequency=POLL_FREQUENCY).until(
            EC.visibility_of_element_located(locator))
        return self.driver.find_element(*locator)

    def get_url(self, path=None):
        if path != None:
            url = URL + path
        else:
            url = None

        if url != None:
            self.driver.get(url)

三、優化base_page

1. DOMAIN,TIMEOUT,POLL_FREQUENCY這幾個全局變量是隨時可以改變的，我們稱爲魔法變量，會特意建一個文件存起來，命名爲: constants.py ，本項目存放在utils目錄 (所有的url也可放在單獨的一個文件叫urls.py，區別於constants.py)。
2. 編寫一個dirver工具類（單例模式），實現只打開和關閉一次瀏覽器，存放在se_utils.py，用於管理啓動不同的瀏覽器的開關。

├── pages
│   ├── __init__.py
│   └── base_page.py
├── run_case
│   └── __init__.py
└── utils
    ├── __init__.py
    ├── constants.py
    └── se_utils.py

constants.py

DOMAIN = 'https://www.tapd.cn/'
LOGIN_URL = '/cloud_logins/login'
TIMEOUT = 10
POLL_FREQUENCY = 0.5

se_utils.py

from selenium import webdriver

class Driver():
    _driver = None

    @classmethod
    def get_driver(cls, browser_name='Chrome'):
        if cls._driver == None:
            if browser_name == 'Chrome':
                cls._driver = webdriver.Chrome()
            elif browser_name == 'Firefox':
                cls._driver = webdriver.Firefox()
            elif browser_name == 'Safari':
                cls._driver == webdriver.Safari()
            elif browser_name == 'Opera':
                cls._driver == webdriver.Opera()
            elif browser_name == 'edge':
                cls._driver == webdriver.Edge()
            elif browser_name == 'Ie':
                cls._driver == webdriver.Ie()
            else:
                raise NameError(
                    "Not found %s browser,You can enter 'Chrome', 'Firefox', 'Ie', 'Edge', 'Safari',Opera" % browser_name)
        return cls._driver

    @classmethod
    def quit_driver(cls):
        if cls._driver:
            cls._driver.quit()
            cls._driver = None

base_page.py

from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

from utils.constants import TIMEOUT, POLL_FREQUENCY, DOMAIN
from utils.se_utils import Driver

class BasePage():

    def __init__(self, path=None):
        self.driver = Driver.get_driver()
        self.driver.maximize_window()
        self.get_url(path)

    def find_element(self, locator):
        WebDriverWait(driver=self.driver, timeout=TIMEOUT, poll_frequency=POLL_FREQUENCY).until(
            EC.visibility_of_element_located(locator))
        return self.driver.find_element(*locator)

    def get_url(self, path=None):
        if path != None:
            url = DOMAIN + path
        else:
            url = None

        if url != None:
            self.driver.get(url)

四、其他頁面繼承BasePage

1. 編寫兩個頁面類，登錄頁和登陸成功後的首頁
2. 頁面類裏包含元素定位和用例

login_page.py

from base_page import BasePage, By
from home_page import HomePageAction

class LoginPage(BasePage):
    username_locator = (By.CSS_SELECTOR, '#username')
    password_locator = (By.CSS_SELECTOR, '#password_input')
    login_button = (By.CSS_SELECTOR, '#tcloud_login_button')

    def username_input_box(self):
        return self.find_element(self.username_locator)

    def password_input_box(self):
        return self.find_element(self.password_locator)

    def login_btn(self):
        return self.find_element(self.login_button)


class LoginPageAction(LoginPage):

    def login(self, username, password):
        self.username_input_box().clear()
        self.username_input_box().send_keys(username)
        self.password_input_box().clear()
        self.password_input_box().send_keys(password)
        self.login_btn().click()
        return HomePageAction()

home_page.py

from base_page import BasePage, By

class HomePage(BasePage):
    user_title_locator = (By.CSS_SELECTOR, '.avatar-text-default.avatar-Q')

    def user_title(self):
        return self.find_element(self.user_title_locator)


class HomePageAction(HomePage):

    @property
    def get_user_title(self):
        return self.user_title().get_attribute('title')