查找元素并不总是成功。这可以改善吗？

时间：2017-07-13 10:06:37

标签： python python-3.x selenium selenium-webdriver automated-tests

我有一个特定的脚本，它引用了我测试的UI中的某些元素。一些元素按其ID定位，一些元素按其Xpath定位。在某个场景之后，此脚本会从我的UI收集一些信息。我重复调整这个脚本以找到特定的结果。

大部分时间都找到所有元素。碰巧在大约100次（有时是几十次循环）循环之后，脚本停止并且大部分时间它与未找到的元素相关。 我知道这种自动化故障很常见，我还假设有时它可能与我的UI（甚至是由UI管理的设备）相关的某个问题有关。

是否有可能/建议以更能容忍某些情况的方式设计我的自动化（通过我使用内置隐式等待的方式）？

这些测试之王是否建议创建重试机制？

这是脚本：

from selenium import webdriver
from datetime import datetime
import time


def main():
    i = 0
    while True:
        i = i +1
        execute_code(i)

        if i == 500:
            break

def Cold_Restart(browser):

    browser.switch_to.default_content()

    browser.switch_to.frame('box_menu')
    browser.switch_to.frame('box_menu')
    SysBtn = browser.find_element_by_id('System')
    SysBtn.click()

    browser.switch_to.default_content()

    browser.switch_to.frame('main_menu')
    Mainten_Btn = browser.find_element_by_id('Maintenance')
    Mainten_Btn.click()

    browser.switch_to.default_content()

    browser.switch_to.frame('main_body')
    Mntn_Rst_Tab = browser.find_element_by_id('tab_restart')
    Mntn_Rst_Tab.click()

    browser.switch_to.frame('maint_sys')
    Cold_Rst_Btn = browser.find_element_by_id('cold_restart')
    Cold_Rst_Btn.click()

    #In order to confirm the Alert Message I first need to switch to the alert pop-up message and then accept it
    alertDialog = browser.switch_to_alert()
    alertDialog.accept()

    time.sleep(205)

    return


def Port_Admin_Down(browser):

    browser.switch_to.default_content()

    browser.switch_to.frame('box_menu')
    browser.switch_to.frame('box_menu')
    Port_19 = browser.find_element_by_id('Port-19')
    Port_19.click()


    browser.switch_to.default_content()

    # Show the Port Configuration TAB
    browser.switch_to.frame('main_body')
    CFP2_Info = browser.find_element_by_id('tab_general')
    CFP2_Info.click()

    browser.switch_to.frame('config_port') # Move to the inner frame that holds the configuration fields and buttons

    Admin_Down_Btn = browser.find_element_by_id('red_button')
    Admin_Down_Btn.click()

    #In order to confirm the Alert Message I first need to switch to the alert pop-up message and then accept it
    alertDialog = browser.switch_to_alert()
    alertDialog.accept()

    time.sleep(5)
    return


def Port_Admin_Up(browser):

    browser.switch_to.default_content()

    browser.switch_to.default_content()

    browser.switch_to.frame('box_menu')
    browser.switch_to.frame('box_menu')
    Port_19 = browser.find_element_by_id('Port-19')
    Port_19.click()
    browser.switch_to.default_content()

    # Show the Port Configuration TAB
    browser.switch_to.frame('main_body')
    CFP2_Info = browser.find_element_by_id('tab_general')
    CFP2_Info.click()

    browser.switch_to.frame('config_port') # Move to the inner frame that holds the configuration fields and buttons

    Admin_Down_Btn = browser.find_element_by_id('green_button')
    Admin_Down_Btn.click()

    #In order to confirm the Alert Message I first need to switch to the alert pop-up message and then accept it
    alertDialog = browser.switch_to_alert()
    alertDialog.accept()    

    time.sleep(5)
    return

def Gui_Login(browser, URL):
    browser.get(URL)
#    browser.implicitly_wait(20) # Implicit wait


    browser.maximize_window()

    # Find the User Name text box and fill the User name
    user_name_box = browser.find_element_by_id('u_name_box')
    user_name_box.click()
    user_name_box.send_keys('admin')

    # Find the Password text box and fill the Password
    user_pass_box = browser.find_element_by_id('u_pass_box')
    user_pass_box.click()
    user_pass_box.send_keys('admin')

    #webdriver.ActionChains(driver).send_keys(Keys.ESCAPE).perform()

    login_button = browser.find_element_by_id('login_but')
    login_button.click()

    return

def Gui_Logout(browser):

    browser.switch_to.default_content() # Get back to the default starting point


    # In order to click the Logout button I needed to pass through two frames
    browser.switch_to.frame('box_menu')
    browser.switch_to.frame('box_menu')
    logout_btn = browser.find_element_by_id('Logout')
    logout_btn.click()

    return

def Sample_Uplink(browser):

    browser.switch_to.default_content()

    # Go to the Configuration Pane (main_menu)
    browser.switch_to.frame('main_menu')
    Configuration_Btn = browser.find_element_by_id('Configuration')
    Configuration_Btn.click()

    browser.switch_to.default_content()

    # Go to the Uplink 1 CFP2 information and take the Rx Pwr
    browser.switch_to.frame('box_menu')
    browser.switch_to.frame('box_menu')
    Port_19 = browser.find_element_by_id('Port-19')
    Port_19.click()

    browser.switch_to.default_content()

    # Show the Optic Module information TAB
    browser.switch_to.frame('main_body')
    CFP2_Info = browser.find_element_by_id('tab_XFP')
    CFP2_Info.click()

    # Collect the Rx Pwr from the CFP2 Info screen
    browser.switch_to.frame('config_port') # Move to the inner frame that holds all the tables
    #browser.find_element_by_class_name('table_round_corner')
    Rx_Pwr = browser.find_element_by_xpath('html/body/form/div[1]/div/table/tbody/tr[2]/td[2]') # Take the Rx Pwr according to its Xpath
    # print (Rx_Pwr.text) # print the Rx Pwr result to screen
    RcvPwr = Rx_Pwr.text

    # Collect the OSNR measurement from the CFP2 Info screen
    OSNR = browser.find_element_by_xpath('html/body/form/div[1]/div/table/tbody/tr[4]/td[2]')
    OSNR_Lvl = OSNR.text


#    time.sleep(5)
    return (RcvPwr, OSNR_Lvl)



def Save_2_File(Rx_Pwr, OSNR_Lvl, i):
    file = open("test_results.txt", "a")
    file.write("%i. " %i)
    file.write(datetime.now().strftime('%H:%M:%S %d-%m-%Y    ')) # Print Time & Date to the text file 
    file.write(Rx_Pwr) # Print the Rx_Pwr to the text file
    file.write('%10s' %(OSNR_Lvl)) # Format the placement of the OSNR value
    file.write('\n') # Make sure that the next iteration will write the results in the next line
    file.close() # Closing the file
    return


def execute_code(i):
    profile = webdriver.FirefoxProfile()
    profile.accept_untrusted_certs = True

    browser = webdriver.Firefox(firefox_profile = profile)
    browser.implicitly_wait(20) # Implicit wait


    URL1 = 'http://10.0.1.131' #  First device that will be Cold Restarted
    Gui_Login(browser, URL1)

    Cold_Restart(browser)
    (RcvPwr, OSNR_Lvl) = Sample_Uplink(browser)
    Save_2_File(RcvPwr, OSNR_Lvl, i)

    URL2 = 'http://10.0.1.134'
    Gui_Login(browser, URL2)

    Cold_Restart(browser)
    (RcvPwr, OSNR_Lvl) = Sample_Uplink(browser)
    Save_2_File(RcvPwr, OSNR_Lvl, i)


    browser.get(URL1)

    Port_Admin_Down(browser)
    Port_Admin_Up(browser)

    time.sleep(5)

    browser.get(URL2)   

    #Get Rx Pwr and OSNR and save in file
    (RcvPwr, OSNR_Lvl) = Sample_Uplink(browser)
    Save_2_File(RcvPwr, OSNR_Lvl, i)


    Port_Admin_Down(browser)
    Port_Admin_Up(browser)

    time.sleep(5)

    browser.get(URL1)   

    #Get Rx Pwr and OSNR and save in file
    (RcvPwr, OSNR_Lvl) = Sample_Uplink(browser)
    Save_2_File(RcvPwr, OSNR_Lvl, i)


    browser.quit()


if __name__ == '__main__':
    main()

2 个答案:

答案 0 :(得分：1)

如果你想在某些循环中避免错误。 try / catch会有所帮助。

首先：将一些代码从execute_code（）移动到init / quit webdriver，然后尝试/捕获execute_code（）。

def main():
    i = 0
    while True:
        i = i +1
        profile = webdriver.FirefoxProfile()
        profile.accept_untrusted_certs = True
        browser = webdriver.Firefox(firefox_profile = profile)
        browser.implicitly_wait(20) # Implicit wait
        try:
            execute_code(i, browser)
            browser.quit()
            if i == 500:
                break
        except:
            browser.quit()

第二：修改execute_code（）。添加浏览器作为参数。删除init / quit webdriver的代码。

def execute_code(i, browser):
    URL1 = 'http://10.0.1.131' #  First device that will be Cold Restarted
    Gui_Login(browser, URL1)

    Cold_Restart(browser)
    (RcvPwr, OSNR_Lvl) = Sample_Uplink(browser)
    Save_2_File(RcvPwr, OSNR_Lvl, i)

    URL2 = 'http://10.0.1.134'
    Gui_Login(browser, URL2)

    Cold_Restart(browser)
    (RcvPwr, OSNR_Lvl) = Sample_Uplink(browser)
    Save_2_File(RcvPwr, OSNR_Lvl, i)


    browser.get(URL1)

    Port_Admin_Down(browser)
    Port_Admin_Up(browser)

    time.sleep(5)

    browser.get(URL2)   

    #Get Rx Pwr and OSNR and save in file
    (RcvPwr, OSNR_Lvl) = Sample_Uplink(browser)
    Save_2_File(RcvPwr, OSNR_Lvl, i)


    Port_Admin_Down(browser)
    Port_Admin_Up(browser)

    time.sleep(5)

    browser.get(URL1)   

    #Get Rx Pwr and OSNR and save in file
    (RcvPwr, OSNR_Lvl) = Sample_Uplink(browser)
    Save_2_File(RcvPwr, OSNR_Lvl, i)

答案 1 :(得分：1)

如果失败，我会wait except NoSuchElementException try: element = WebDriverWait(driver, 10).until( EC.presence_of_element_located((By.ID, "Maintenance")) ) except NoSuchElementException: # Handle your error case here else: element.click() 。

def wait_for_func(driver, element_tuple, func_success, func_failed, time=10):
    try:
        element = WebDriverWait(driver, time).until(
            EC.presence_of_element_located(element_tuple)
        )
    except NoSuchElementException:
        return None, func_failed(element)
    else:
        return element, func_success(element)

def do_click(element):
    return element.click()

def do_print():
     return 'Something gone wrong'

element, result = wait_for_func(driver, (By.ID, 'Maintenance'), func_success=do_click, func_failed=do_print)

此时你可以优雅地停止，重试最后一次通话，只是跳过这个测试等等......

检查python selenium waits doc

编辑评论

我不是很喜欢这个，但如果添加一些线条打扰你，你可以随时为它做一个包装：

wait_for_func(driver, (By.ID, 'Maintenance'), lambda el: el.click(), do_print)

这样可以避免重复尝试/除了所有地方，但你肯定知道Zen of Python：

  

明确比隐含更好。   简单比复杂更好。   复杂比复杂更好。

无论如何，如果你做了类似的事情，你会发现定义一个函数只是为了点击一个元素并不是那么好，所以lambda可以帮助你：

lambda

我发现char C = '#'; byte B = (byte)C; Console.WriteLine(B.GetType()); // Outputs "Byte" 不是那么漂亮，但是如你所愿。

相关问题
最新问题