selenium基础框架的封装(Python版)

一、常用函数的封装

在使用selenium做web自动化测试的过程中，经常会碰到各种各样的问题，比如：

　　1、页面加载比较慢时，selenium查找元素抛出异常，导致脚本运行中止

　　2、写完脚本后发现代码逻辑不够清晰，维护困难，降低测试效率

所以很有必要对selenium进行封装使框架更适合公司的需求，下面是基类的封装，主要使用单例模式，可以将业务逻辑代码存放在不同的Python文件中

注：BASE_URL为博客园的登录地址

 # -*- coding:utf-8 -*-
 from selenium import webdriver
 import time

 OVER_TIME = 5
 BASE_URL = "https://passport.cnblogs.com/user/signin?ReturnUrl=https%3A%2F%2Fwww.cnblogs.com%2F"

 class Driver(object):

     def __new__(cls, *args, **kw):
         """
         使用单例模式将类设置为运行时只有一个实例，在其他Python类中使用基类时，
         可以创建多个对象，保证所有的对象都是基于一个浏览器
         """
         if not hasattr(cls, '_instance'):
             orig = super(Driver, cls)
             cls._instance = orig.__new__(cls, *args, **kw)
         return cls._instance

     def start(self, url=BASE_URL, driver_name="Chrome"):
         """
         启动浏览器
         :param url: 测试地址
         :param driver_name: 在启动时设置浏览器的类型
         :return:
         """
         if driver_name == "Firefox":
             self.driver = webdriver.Firefox()
         elif driver_name == "Ie":
             self.driver = webdriver.Ie()
         else:
             self.driver = webdriver.Chrome()
         self.driver.implicitly_wait(OVER_TIME)
         self.driver.get(url)
         self.driver.maximize_window()

     def get_url(self):
         """返回浏览器的地址"""
         return BASE_URL

     def find_element(self, by, value):
         """
         这里添加了一个OVER_TIME作为查找元素的超时次数，根据系统的实际情况设置OVER_TIME的大小
         """
         for i in range(OVER_TIME):
             try:
                 return self.driver.find_element(by=by, value=value)
             except Exception, e:
                 print e

     def find_elements(self, by, value):
         """与find_element一致"""
         for i in range(OVER_TIME):
             try:
                 return self.driver.find_elements(by=by, value=value)
             except Exception, e:
                 print e

     def find_display_elements(self, by, value):
         """
         查找状态为displayed的元素集合，当查找一类元素时，
         经常出现有些元素是不可见的情况，此函数屏蔽那些不可见的元素
         """
         for i in range(OVER_TIME):
             try:
                 elements = self.driver.find_elements(by=by, value=value)
                 num = elements.__len__()
             except Exception, e:
                 print e
                 time.sleep(1)
             if num >= 1:
                 break
         display_element = []
         # 将可见的元素放到列表中， 并返回
         for j in range(num):
             element = elements.__getitem__(j)
             if element.is_displayed():
                 display_element.append(element)
         return display_element

     def is_element_present(self, By, Value):
         """判断元素是否存在"""
         try:
             self.driver.find_element(by=By, value=Value)
             return True
         except Exception, e:
             print e
             return False

     def close(self):
         self.driver.close()

     def quit(self):
         u"""退出浏览器"""
         self.driver.quit()

 if __name__ == "__main__":
     page = Driver()
     page.start()

 二、使用封装过的框架和pageobject模式设计登录脚本

 # -*- coding:utf-8 -*-
 from webdriver import Driver
 from selenium.webdriver.common.by import By

 class Login(object):

     def __init__(self, user_name="xxx", pwd="xxx"):
         self.page = Driver()
         page = self.page
         page.start(url=page.get_url())
         self.user_name = page.find_element(By.ID, "input1")
         self.pwd = page.find_element(By.ID, "input2")
         self.login_btn = page.find_element(By.ID, "signin")
         self.login(user_name, pwd)

     def login(self, user_name, pwd):
         self.user_name.clear()
         self.user_name.send_keys(user_name)
         self.pwd.clear()
         self.pwd.send_keys(pwd)
         self.login_btn.click()

 if __name__ == "__main__":
     Login("xxx", "xxx")

 三、打开博客园新闻、博问等子页面脚本设计

 # -*- coding:utf-8 -*-
 from webdriver import Driver
 from login import Login
 from selenium.webdriver.common.by import By

 class SubPage(object):

     def __init__(self):
         self.page = Driver()
         page = self.page
         self.home_page = page.find_element(By.LINK_TEXT, u"园子")
         self.news_page = page.find_element(By.LINK_TEXT, u"新闻")
         self.q_page = page.find_element(By.LINK_TEXT, u"博问")
         self.ing_page = page.find_element(By.LINK_TEXT, u"闪存")

     def open_home_page(self):
         self.home_page.click()

     def open_news_page(self):
         self.news_page.click()

     def open_q_page(self):
         self.q_page.click()

     def open_ing_page(self):
         self.ing_page.click()

 if __name__ == '__main__':
     Login("xxx", "xxx")
     sub = SubPage()
     sub.open_q_page()

四、其他

　　按照此模式博友可根据公司具体的业务来实现自动化测试脚本，本文抛砖引玉，如有其他更好的想法可以一起探讨

　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　　Thanks!