自动化测试-16.selenium数据的分离之Excel的使用

前言

　　在自动化测试中，单纯写得测试脚本，数据往往是写死的，这样会给程序的可扩展性降低，以及成本的增加，将程序和数据进行剥离有利于提高代码的执行效率，提升工作量，因此将数据写入到Excel文件中，前置条件已在本地搭建BugFree，若没有搭建可参考BugFree本地搭建。

　　下列代码实现创建Excel文件的数据，和从Excel中读取数据，并将数据传值给脚本：

#encoding=utf-8

from selenium import webdriver

import xlsxwriter #往Excel表格写数据 安装cmd输入： pip install XlsxWriter，前置条件已安装pip

import xlrd,os # xlrd模块用于读EXCEL文档数据，安装xlrd在cmd中输入pip install xlrd，前置条件已安装pip

# 新建Excel表格
def write_content():
    # 新建Excel表格
    xl = xlsxwriter.Workbook(os.getcwd()+'\\testData\\testResult.xlsx') # 新建Excel表格，注意需要在当前目录下新建testData文件夹
    table = xl.add_worksheet('result') # 新建一个名为result的页签
    # 设置格式：行高、列宽、底色、字号等格式
    blue = xl.add_format({'font_name': 'Arial', 'font_size': 10,'color': 'blue'})
    red  = xl.add_format({'font_name': 'Arial', 'font_size': 10,'color': 'red'})
    # 设置第一行的行高为30
    table.set_row(0,15)
    # 设置第1列到第3列的列宽为50
    table.set_column(0,2,50)
    # 写数据
    table.write_string(0,0,'url',blue)
    table.write_string(0,1,u'用户名',blue) # 往第一行第一列写入字符串数据
    table.write_string(0,2,u'密码',red)
    table.write_string(1,0,u'http://localhost/index.php/site/login',blue) # 往第二行第一列写入数字,本地搭建的bugfree地址
    table.write_string(1,1,'admin',blue)
    table.write_string(1,2,'123456',red)
    # 关闭Excel表格
    xl.close()

# 读取数据
def read_content():
    # 一下脚本实现bugfree登录，测试数据从Excel读取
    data = xlrd.open_workbook(os.getcwd()+'\\testData\\testResult.xlsx')
    tableGet = data.sheet_by_name('result')
    testData = tableGet.row_values(1) #读取第二行数据
    url,username,password = testData[0],testData[1],testData[2]
    print url,username,password
    return url,username,password

# 定义测试脚本
def bugFreeLogin(driver,url,username,password):
    driver.get(url)
    driver.find_element_by_id('LoginForm_username').send_keys(username)
    driver.find_element_by_id('LoginForm_password').send_keys(password)
    driver.find_element_by_id('SubmitLoginBTN').click()

# 启动脚本
if __name__ == '__main__':
    # 数据的写入
    write_content()
    # 数据的读取
    tup_result = read_content()
    # 启动脚本去执行
    driver = webdriver.Firefox()
    bugFreeLogin(driver,tup_result[0],tup_result[1],tup_result[2])