Python+Appium自動化測試(6)-元素等待方法與重新封裝元素定位方法

在appium自動化測試腳本運行的過程中，因為網絡不穩定、測試機或模擬器卡頓等原因，有時候會出現頁面元素加載超時元素定位失敗的情況，但實際這又不是bug，只是元素加載較慢，這個時候我們就會使用元素等待的方法來避免這種情況，增加代碼的健壯性。

一，元素等待方法

1，強制等待

import time

# 強制等待5s
time.sleep(5)

2，隱式等待

implicitly_wait()是由webdriver提供的隱式等待方法，它不是針對某一個元素，而是針對當前session（即當前driver對象的生命周期）的全部元素，所以只需要在構造driver對象時設置一次即可。隱式等待在定位元素時，需等待該頁面全部元素加載完成，才會執行下一步操作（即下一條語句），如果超過設定時間未加載完成則拋出異常。

from appium import webdriver

def android_driver():
    desired_caps = {
        "platformName": "Android",
        "platformVersion": "10",
        "deviceName": "PCT_AL10",
        "appPackage": "com.ss.android.article.news",
        "appActivity": ".activity.MainActivity",
        "automationName": "uiautomator2",
        "unicodeKeyboard": True,
        "resetKeyboard": True,
    }
    # 啟動app
    driver = webdriver.Remote('http://127.0.0.1:4723/wd/hub', desired_caps)
    # 隱式等待8s
    driver.implicitly_wait(8)
    return driver

3，顯式等待

3.1，webDriverWait()是由webdriver提供的顯示等待方法。與隱式等待不一樣的是，顯示等待是針對單個元素定位進行等待，每隔一段時間檢查需要定位的元素是否加載完成，超過參數規定的時間仍未定位到該元素，則定位該元素失敗，拋出異常。

from selenium.webdriver.support.ui import WebDriverWait

WebDriverWait(driver,timeout,poll_frequency=0.5,ignored_exceptions=None)
# 參數說明：
# driver，上面代碼返回的driver對象
# timeout，最長等待時間，使用時要考慮隱式等待的時間（假如有設置隱式等待的話）
# poll_frequency，檢查元素的時間間隔，默認是0.5s即每隔0.5秒查找一次
# ignored_exceptions，超時后拋出的異常信息，默認NoSuchElementExeception

3.2，WedDriverWait()需要與unit()或until_not()方法結合使用。

until(method, message='')
# 源碼說明：Calls the method provided with the driver as an argument until the return value is not False.
# 調用driver提供的方法作為參數，直到返回值不是False。

until_not(method, message='')
# 源碼說明：Calls the method provided with the driver as an argument until the return value is False.
# 調用driver提供的方法作為參數，直到返回值為False

自定義等待時間，使用find_element_by_*()方法定位元素，如下：

# 設置等待，最長等待時間為5s，每0.5秒檢查一次
wait = WebDriverWait(driver, 5, 0.5)
# 使用匿名函數定位元素
wait.until(lambda diver:driver.find_element_by_id("android:id/button1"))

3.3，WebDriverWait()與expected_conditions結合使用。

expected_conditions是webdriver.support提供的一個類，這個類里面提供了比較多的預期條件判斷的方法，但在我們定位元素過程中常用以下三種方法

presence_of_element_located	判斷某個元素是否被加載到 dom 樹里，但該元素不一定可見
visibility_of_element_located	判斷元素是否可見（可見代表元素非隱藏，並且元素寬和高都不等於 0）
element_to_be_clickable	判斷某個元素中是否可見並且可點擊

from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

wait = WebDriverWait(driver, 5, 0.5)
element = waite.until(EC.presence_of_element_located((By.ID, "android:id/button1"), message="")
# message=""可以省略，注意此時By.ID有兩層（）
# element = waite.until(EC.presence_of_element_located((By.ID, "android:id/button1"))

二，重新封裝元素定位方法

在腳本編寫的過程中，為了增加腳本的健壯性，排除非bug因素導致的腳本運行失敗，我們可以在定位元素時加入顯示等待，封裝成新的元素定位方法。

# /basePage.py

from selenium.webdriver.support import expected_conditions as ec
from selenium.webdriver.support.ui import WebDriverWait
from appium.webdriver.common.mobileby import MobileBy as By

class BasePage:
    def __init__(self, driver):
        self.driver = driver

    def get_visible_element(self, locator, timeout=20):
        '''
        獲取可視元素
        param loctor: By方法定位元素，如(By.XPATH, "//*[@text='照片']")
        return：返回可見元素
        '''
        try:
            return WebDriverWait(self.driver, timeout).until(
                ec.visibility_of_element_located(locator)
            )
        except Exception as e:
            # 截圖、日志
            Screenshots(self.driver, "獲取可視元素失敗").screen_shot()
            log.error("獲取可視元素失敗：{}".format(e))

    def get_presence_element(self, locator, timeout=20):
        '''獲取存在元素'''
        try:
            return WebDriverWait(self.driver, timeout).until(
                ec.presence_of_element_located(locator)
            )
        except Exception as e:
            Screenshots(self.driver, "獲取存在元素失敗").screen_shot()
            log.error("獲取存在元素失敗：{}".format(e))

    def get_clickable_element(self, locator, timeout=20):
        '''獲取可點擊元素'''
        try:
            return WebDriverWait(self.driver, timeout).until(
                ec.element_to_be_clickable(locator)
            )
        except Exception as e:
            Screenshots(self.driver, "獲取可點擊元素失敗").screen_shot()
            log.error("可點擊元素獲取失敗：{}".format(e))

這樣就可以調用新的方法來進行元素定位

# /homePage.py

from basePage import BasePage

class HomePage(BasePage):
    i_know_btn = (By.ID, 'com.ss.android.article.news:id/ciy')
    jurisdiction_btn = (By.ID, 'android:id/button1')
    no_login_btn = (By.XPATH, '//android.widget.TabWidget/android.widget.RelativeLayout[@index=3]')

    def enter_to_login_page(self):
        '''首頁進入未登錄頁面'''
        get_visible_element(self.i_know_btn).click()
        get_presence_element(self.jurisdiction_btn).click()
        get_clickable_element(self.no_login_btn).click()