Ui功能自动化测试：重新封装 Webdriver Api

selenium 做自动化测试过程中发现，有些元素定位比较麻烦，有些元素定位写脚本特别的繁琐，所以重新封装了下Api 。

• find 寻找单个元素
• find_elements 寻找多个元素
• clear_keys 清除按钮
• send_keys 输入
• click_button 点击按钮
• script 输入js脚本
• wait 智能等待
• set_window 设置window大小

def find(self,loc):
‘’’
find element.
Usage:
driver.find((By.XPATH,"//a"))
‘’’
try:
WebDriverWait(self.driver,self.timeout).until(lambda drive:drive.find_element(*loc).is_displayed())
return self.driver.find_element(*loc)
except Exception,e:
print u"%s 页面中超时%ds未能找到 %s 元素%s" %(self,self.timeout,loc,e)

def find_elements(self,loc):
    '''
    find elements.
    Usage:
    driver.find_elements((By.XPATH,"//a"))
    '''
    try:
        WebDriverWait(self.driver,self.timeout).until(lambda drive:drive.find_elements(*loc).is_displayed())
        return self.driver.find_element(*loc)
    except Exception,e:
        print u"%s 页面中超时%ds未能找到 %s 元素%s" %(self,self.timeout,loc,e)


def click_keys(self,loc):
    '''
    click keys
    Usage:
    driver.click_keys((By.XPATH,"//a"))
    '''
    self.find(loc).click()


def clear_keys(self,loc):
    '''
    clear keys
    Usage:
    driver.clear_keys((By.XPATH,"//a"))
    '''
    self.find(loc).clear()


def send_keys(self,loc,value):
    '''
    send keys
    Usage:
    send_keys((By.XPATH,"//a"),'a')
    '''
    sleep(3)
    self.clear_keys(loc)
    self.find(loc).send_keys(value)


def click_button(self,loc):
    '''
    click button
    Usage:
    click_button((By.XPATH,"//a"))
    '''
    sleep(3)
    self.find(loc).click()


def script(self,src):
    '''
    execute_script
    Usage:
    script(src)
    '''
    return self.driver.execute_script(src)


def wait(self, secs):
    '''
    Implicitly wait.All elements on the page.
    Usage:
    driver.wait(10)
    '''
    self.driver.implicitly_wait(secs)


def set_window(self, wide, high):
    '''
    Set browser window wide and high.
    Usage:
    driver.set_window(wide,high)
    '''
    self.driver.set_window_size(wide, high)


def right_click(self, loc):
    '''
    Right click element.
    Usage:
    driver.right_click((By.XPATH,"//a"))
    '''
    el = self.find(loc)
    ActionChains(self.driver).context_click(el).perform()


def move_to_element(self, loc):
    '''
    Mouse over the element.
    Usage:
    driver.move_to_element((By.XPATH,"//a"))
    '''
    el = self.find(loc)
    ActionChains(self.driver).move_to_element(el).perform()


def double_click(self, loc):
    '''
    Double click element.
    Usage:
    driver.double_click((By.XPATH,"//a"))
    '''
    el = self.find(loc)
    ActionChains(self.driver).double_click(el).perform()


def drag_and_drop(self, loc1, loc2):
    '''
    Drags an element a certain distance and then drops it.
    Usage:
    driver.drag_and_drop((By.XPATH,"//a"),(By.XPATH,"//b"))
    '''
    element  = self.find(loc1)
    target  = self.find(loc2)
    ActionChains(self.driver).drag_and_drop(element , target).perform()


def get_display(self, loc):
    '''
    Gets the element to display,The return result is true or false.
    Usage:
    driver.get_display(By.XPATH,"//a")
    '''
    el = self.find(loc)
    return el.is_displayed()


def isElement(self,identifyBy,c):
    '''
    Determine whether elements exist
    Usage:
    isElement(By.XPATH,"//a")
    '''
    sleep(1)
    flag=None
    try:
        if identifyBy == "id":
            #self.driver.implicitly_wait(60)
            self.driver.find_element_by_id(c)
        elif identifyBy == "xpath":
            #self.driver.implicitly_wait(60)
            self.driver.find_element_by_xpath(c)
        elif identifyBy == "class":
            self.driver.find_element_by_class_name(c)
        elif identifyBy == "link text":
            self.driver.find_element_by_link_text(c)
        elif identifyBy == "partial link text":
            self.driver.find_element_by_partial_link_text(c)
        elif identifyBy == "name":
            self.driver.find_element_by_name(c)
        elif identifyBy == "tag name":
            self.driver.find_element_by_tag_name(c)
        elif identifyBy == "css selector":
            self.driver.find_element_by_css_selector(c)
        flag = True
    except NoSuchElementException,e:
        flag = False
    finally:
        return flag


def refresh(self):
    '''
    refresh  page
    Usage:
    driver.refresh
    '''
    self.driver.implicitly_wait(10)
    self.driver.refresh()


def list_to_str(self,string):
    '''
    string to unicode
    Usage:
    list_to_str(str)
    '''
    str_symptom = str(string).replace('u\'','\'')
    return str_symptom.decode("unicode-escape")


def get_text(self,loc):
    '''
    get text
    Usage:
    get_text(By.XPATH,"//a")
    '''
    sleep(3)
    return self.find(loc).text


def get_url(self,url):
    '''
    get url
    Usage:
    get_url(url)
    '''
    sleep(3)
    return self.driver.get(url)


def get_title(self,url):
    '''
    get title
    Usage:
    get_title(url):
    '''
    self.driver.get(url)
    return self.driver.title

• right_click 右击
• move_to_element 移动到某个元素
• double_click 双击
• drag_and_drop 拖动
• get_display 获取元素显示属性
• isElement 判断元素是否存在
• refresh 刷新页面
• list_to_str str转换到unicode
• get_text 获取元素text
• get_url 切换url
• get_title 获取当前title