selenium+python做web端自動化測試框架與實例詳解教程

最近受到萬點暴擊，由於公司業務出現問題，工作任務沒那么繁重，有時間摸索selenium+python自動化測試，結合網上查到的資料自己編寫出適合web自動化測試的框架，由於本人也是剛剛開始學習python，這套自動化框架目前已經基本完成了所以總結下編寫的得失，便於以后回顧溫習，有許多不足的的地方，也遇到了各種奇葩問題，希望大神們多多指教。

首先我們要了解什么是自動化測試，簡單的說編寫代碼、腳本，讓軟件自動運行，發現缺陷，代替部分的手工測試。了解了自動化測試后，我們要清楚一個框架需要分那些模塊:

 

上圖的框架適合大多數的自動化測試，比如web UI  、接口自動化測試都可以采用，如大佬有好的方法請多多指教，簡單說明下每個模塊：

• common：存放一些共通的方法
• data：存放一些文件信息
• logs：存放程序中寫入的日志信息
• picture：存放程序中截圖文件信息
• report：存放測試報告
• test_case：存放編寫具體的測試用例
• conf.ini、readconf.py：存放編寫的配置信息

下面就具體介紹每個模塊的內容：conf.ini主要存放一些不會輕易改變的信息，編寫的代碼如下：

[DATABASE]

host = 127.0.0.1
username = root
password = root
port = 3306
database = cai_test

[HTTP]
# 接口的url
baseurl = http://xx.xxxx.xx
port = 8080
timeout = 1.0
readconf.py文件主要用於讀取conf.ini中的數據信息
# *_*coding:utf-8 *_*
__author__ = "Test Lu"
import os,codecs
import configparser

prodir = os.path.dirname(os.path.abspath(__file__))
conf_prodir = os.path.join(prodir,'conf.ini')
class Read_conf():
     def __init__(self):
         with open(conf_prodir) as fd:
             data = fd.read()
             #清空文件信息
             if data[:3] ==codecs.BOM_UTF8:
                 data = data[3:]
                 file = codecs.open(conf_prodir,'w')
                 file.write(data)
                 file.close()
         self.cf = configparser.ConfigParser()
         self.cf.read(conf_prodir)
     def  get_http(self,name):
         value = self.cf.get("HTTP",name)
         return value

     def get_db(self,name):
         return self.cf.get("DATABASE",name)
這里需要注意，python3.0以上版本與python2.7版本import configparser的方法有一些區別
讀取一些配置文集就介紹完了，下面就說說common包下的公共文件

現在就從上往下結束吧！common主要是封裝的一些定位元素的方法：

# *_*coding:utf-8 *_*
__author__ = "Test Lu"
from selenium import webdriver
import time,os
import common.config
# from common.logs import MyLog
project_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
class Comm(object):
    def __init__(self,driver):
        self.driver = driver
        # self.driver = webdriver.Firefox()
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
    def open_url(self,url):
        self.driver.get(url)
        self.driver.implicitly_wait(30)
    # selenium 定位方法
    def locate_element(self,loatetype,value):
        if (loatetype == 'id'):
            el = self.driver.find_element_by_id(value)
        if (loatetype == 'name'):
            el = self.driver.find_element_by_name(value)
        if (loatetype == 'class_name'):
            el = self.driver.find_element_by_class_name(value)
        if (loatetype == 'tag_name'):
            el = self.driver.find_elements_by_tag_name(value)
        if (loatetype == 'link'):
            el = self.driver.find_element_by_link_text(value)
        if (loatetype == 'css'):
            el = self.driver.find_element_by_css_selector(value)
        if (loatetype == 'partial_link'):
            el = self.driver.find_element_by_partial_link_text(value)
        if (loatetype == 'xpath'):
            el = self.driver.find_element_by_xpath(value)
        return el if el else None
    # selenium 點擊
    def click(self,loatetype,value):
        self.locate_element(loatetype,value).click()
    #selenium 輸入
    def input_data(self,loatetype,value,data):
        self.locate_element(loatetype,value).send_keys(data)
    #獲取定位到的指定元素
    def get_text(self, loatetype, value):
        return self.locate_element(loatetype, value).text
    # 獲取標簽屬性
    def get_attr(self, loatetype, value, attr):
        return self.locate_element(loatetype, value).get_attribute(attr)
    # 頁面截圖
    def sc_shot(self,id):
        for filename in os.listdir(os.path.dirname(os.getcwd())) :
            if filename == 'picture':
                break
        else:
            os.mkdir(os.path.dirname(os.getcwd()) + '/picture/')
        photo = self.driver.get_screenshot_as_file(project_dir +  '/picture/'
                                                   + str(id) + str('_') + time.strftime("%Y-%m-%d-%H-%M-%S") + '.png')
        return photo
    def __del__(self):
        time.sleep(2)
        self.driver.close()
        self.driver.quit()
下面介紹下，config文件主要用於讀取文件中的信息：

import os,xlrd
from common.logs import MyLog
from xml.etree import ElementTree as ElementTree
mylogger = MyLog.get_log()
project_dir = os.path.dirname(os.getcwd())

def user_Add():
    '''excel文件中讀取用戶登錄信息'''
    with xlrd.open_workbook(project_dir+'/data/test_data.xlsx') as files:
        table_user = files.sheet_by_name('userdata')
        try:
            username = str(int(table_user.cell(1,0).value))
        except:
            username = str(table_user.cell(1,0).value)
        try:
            passwd = str(int(table_user.cell(1,1).value))
        except:
            passwd = str(table_user.cell(1,1).value)
        try:
            check = str(int(table_user.cell(1, 2).value))
        except Exception:
            check = str(table_user.cell(1, 2).value)
        table_url = files.sheet_by_name('base_url')
        base_url = str(table_url.cell(1,0).value)
        return (username,passwd,base_url,check)

#從xml文件中讀取信息，定義全局一個字典來存取xml讀出的信息
database={}
def set_read_xml():
    sql_path = os.path.join(project_dir,'data','SQL.xml')
    data =ElementTree.parse(sql_path)
    for db in data.findall('database'):
        name = db.get('name')
        table = {}
        for tb in db.getchildren():
            table_name = tb.get("name")
            sql = {}
            for data in tb.getchildren():
                sql_id = data.get("id")
                sql[sql_id] = data.text
            table[table_name] = sql
        database[name] = table
        mylogger.info("讀取的xml文件的信息%s" %database)
def get_sql_sen(database_name,table_name,sql_id):
    set_read_xml()
    db = database.get(database_name).get(table_name)
    if db.get(sql_id):
        sql = db.get(sql_id).strip()
        mylogger.info("返回sql語句信息%s" % sql)
        return sql
    else:
        mylogger.info("查下的信息為空,傳遞的參數有誤！數據庫名稱：【%s】，表信息【%s】，查詢的id【%s】"
                      %(database_name,table_name,sql_id))

接着介紹最簡單的日志logs.py模塊：

# logging模塊支持我們自定義封裝一個新日志類
import logging,time
import os.path
class Logger(object):
    def __init__(self, logger,cases="./"):       
        self.logger = logging.getLogger(logger)
        self.logger.setLevel(logging.DEBUG)
        self.cases = cases
        # 創建一個handler，用於寫入日志文件
        for filename in os.listdir(os.path.dirname(os.getcwd())):
            if filename == "logs":
                break
        else:
            os.mkdir(os.path.dirname(os.getcwd())+'/logs')
        rq = time.strftime('%Y%m%d%H%M', time.localtime(time.time()))
        log_path = os.path.dirname(os.getcwd()) + '/logs/'  
        log_name = log_path + rq + '.log'  # 文件名
        # 將日志寫入磁盤
        fh = logging.FileHandler(log_name)
        fh.setLevel(logging.INFO)
        # 創建一個handler，用於輸出到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.INFO)
        # 定義handler的輸出格式
        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        fh.setFormatter(formatter)
        ch.setFormatter(formatter)
        # 給logger添加handler
        self.logger.addHandler(fh)
        self.logger.addHandler(ch)
    def getlog(self):
        return self.logger

common模塊最后一個是test_runner.py這個方法主要是用來執行全部的測試用例

import time,HTMLTestRunner
import unittest
from common.config import *
project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__),os.pardir))
class TestRunner(object):
    ''' 執行測試用例 '''
    def __init__(self, cases="../",title="Auto Test Report",description="Test case execution"):
        self.cases = cases
        self.title = title
        self.des = description
    def run(self):
        for filename in os.listdir(project_dir):
            if filename == "report":
                break
        else:
            os.mkdir(project_dir+'/report')
        # fp = open(project_dir+"/report/" + "report.html", 'wb')
        now = time.strftime("%Y-%m-%d_%H_%M_%S")
        # fp = open(project_dir+"/report/"+"result.html", 'wb')
        fp = open(project_dir+"/report/"+ now +"result.html", 'wb')
        tests =  unittest.defaultTestLoader.discover(self.cases,pattern='test*.py',top_level_dir=None)
        runner = HTMLTestRunner.HTMLTestRunner(stream=fp, title=self.title, description=self.des)
        runner.run(tests)
        fp.close()
以上就是common公共模塊所有的模塊，簡單說下在寫這些公共模塊時，出現了各種問題，特別是讀取xml文件的，唉！對於一個python的小白真是心酸啊！接着說下db模塊的內容，db模塊主要是讀取sql語句以及返回對應的值！

import pymysql
import readconf
import common.config as conf
readconf_conf = readconf.Read_conf()

host = readconf_conf.get_db("host")
username = readconf_conf.get_db("username")
password = readconf_conf.get_db("password")
port = readconf_conf.get_db("port")
database = readconf_conf.get_db("database")
config_db = {
    'host': str(host),
    'user': username,
    'password': password,
    'port': int(port),
    'db': database
}
class Mysql_DB():
    def __init__(self):
        '''初始化數據庫'''
        self.db = None
        self.cursor = None
    def connect_db(self):
        '''創建連接數據庫'''
        try:
            self.db = pymysql.connect(**config_db)
            #創建游標位置
            self.cursor = self.db.cursor()
            # print("鏈接數據庫成功")
            conf.mylogger.info("鏈接IP為%s的%s數據庫成功" %(host,database))
        except ConnectionError as ex:
            conf.mylogger.error(ex)
  
    def get_sql_result(self,sql,params,state):
        self.connect_db()
        try:
            self.cursor.execute(sql, params)
            self.db.commit()
            # return self.cursor
        except ConnectionError as ex:
            self.db.rollback()
        if state==0:
            return self.cursor.fetchone()
        else:
            return self.cursor.fetchall()
    def close_db(self):
        print("關閉數據庫")
        conf.mylogger.info("關閉數據庫")
        self.db.close()
剛開始寫db模塊是一直對字典模塊的信息怎樣傳遞到數據鏈接的模塊，進過網上查詢好些資料才徹底解決，對自己來說也是一種進步，哈哈，下面說下自己踩的坑，幫助自己以后學習**config_db把字典變成關鍵字參數傳遞，
下面舉例說明下：
如果kwargs=｛'a':1,'b':2,'c':3｝那么**kwargs這個等價為test(a=1,b=2,c=3)是不是很簡單！哈哈

以上就是框架的主要模塊，其他的模塊每個項目與每個系統都不一樣，在這里就是列舉出來了，因為就算寫出來大家也不能復用，下面就給大家看看小白還有哪些模塊

 

看下了下data模塊下的xml模塊大家可能用的到，就給大家貼出來吧！因為ui測試主要就用到select與delete語句，所以也沒有寫多么復雜的sql語句

<?xml version="1.0" encoding="utf-8" ?>
<data>

    <database name="database_member">
        <table name="table_member">
            <sql id="select_member">
                select * from user where real_name=%s
            </sql>
            <sql id="select_member_one">
                select mobile from user where mobile=%s
            </sql>
            <sql id="delete_member">
                delete from user where mobile=%s
            </sql>
            <sql id="insert_member">
                insert into user(id) value(%s)
            </sql>
            <sql id="update_member">
                uodate user set real_name = %s where uuid=%s
            </sql>
        </table>
    </database>
</data>

下面介紹下其他模塊的內容：test_data.xlsx文件主要是存放一些用戶信息，以及url信息，這樣修改用戶信息與url信息就不要修改代碼方便以后操作！logs是在代碼運行時候產生的日志信息，picture是存放圖片信息，report存放輸入的報告信息，
test_case是編寫用戶的模塊需要所有的用例名稱都要以test開頭來命名哦，這是因為unittest在進行測試時會自動匹配test_case文件夾下面所有test開頭的.py文件

以上就是小編寫的UI自動化框架，也是小編第一次寫這種博文，轉載請標明出處，謝謝。喜歡的朋友也可以給小編我點個贊吧，我會繼續努力學習，與大家共同成長噠！