自动化测试-18.selenium之bugFree代码注释

 #encoding=utf-8
 import xlrd,time,os
 from xlutils.copy import copy
 from selenium import webdriver

 def getTestData(excelPath,sheetName):
     '''
     定义从excel文件中获取数据
     :param excelPath: excel文件路径
     :param sheetName: 表格中的页签名
     :return: 返回测试数据字段
     '''
     # 打开存有测试数据的excel
     xl = xlrd.open_workbook(excelPath)
     # 指定打开某个页签的table
     table = xl.sheet_by_name(sheetName)
     # 获取表中的第一行，用来做字典的key进行存储,返回的是列表从第一列到最后一列
     listKey = table.row_values(0)
     # 用于存放测试数据
     listTestData = []
     # 外围for循环取到的是行
     for i in range(1,table.nrows):
         # 将存取的每次数据存在空字典内
         dicData = {}
         # 内部for循环取到的是列
         for j in range(0,table.ncols):
             # key为第一行的每列，value为每行的列值
             dicData.update({listKey[j]:table.cell_value(i,j)})
         listTestData.append(dicData)
     print('取到的数据为：')
     # 遍历取到的每行数据
     for rowData in listTestData:
         print(rowData)
     # 返回测试数据
     return listTestData

 def writeResultToExcel(excelPath,sheetName,listRe,reResultFiles):
     '''
     定义函数，先复制测试用例表格模板，再往里面写入实际结果，测试结果以及截图路径
     :param excelPath: 需要复制的excel模板路径
     :param sheetName: 被复制末班中的页签名
     :param listRe: 列表类型，其内部数据为字典类型，列表中包含实际结果，测试结果，截图路径的存取数据字典
     :param reResultFiles: 写入完成后保存的excel路径
     :return:
     '''
     # 打开指定路径的excel文件
     xl = xlrd.open_workbook(excelPath,encoding_override='utf-8')
     # 打开对应页签的表
     table = xl.sheet_by_name(sheetName)
     # 复制模板
     excelMudle = copy(xl)
     # 获取到sheetName对应的表
     mudleTable = excelMudle.get_sheet(sheetName)
     # 获取模板表中有多少列数
     cols = table.ncols
     # 获取模板表中有多少行数
     rows = table.nrows
     for rowsName in listRe[0].keys():
         # 每次往第一行最新的列写入表头
         mudleTable.write(0,cols,rowsName)
         # 列索引往后移动一位
         cols+=1
     #逐行写数据，从第2行开始写，写到最后一行，每行是一次测试结果，每次测试结果放在字典中
     for i  in  range(1,rows):
         '''
         listRe[0]是第二行索引为1的测试结果，listRe[1]是第三行索引为2的测试结果,行值和数据存储存字典索引相差1
         每次往第i行写数据  写 listRe[i-1]的数据
         实际结果,i是行索引,cols-3是列的倒数第3个列，cols-2是列的倒数第2个列，cols-1是列的倒数第1个列
         '''
         mudleTable.write(i,cols-3,listRe[i-1]['realyResult'])
         # 测试结果
         mudleTable.write(i,cols-2,listRe[i-1]['testResult'])
         # 截图路径
         mudleTable.write(i,cols-1,listRe[i-1]['pngPath'])
         # 保存测试结果
         excelMudle.save(reResultFiles)

 def removeDir(dirPath):
     '''
     删除Windows下文件夹里面的所有文件和子文件夹
     :param dirPath: 要删除的文件夹路径
     :return: None
     '''
     # 读取要删除的目录下所有文件和子目录
     listTopDir = os.listdir(dirPath)
     for item in listTopDir:
         # 如果是子目录则递归删除
         if os.path.isdir(dirPath+'\\'+item):
             # 删除子目录中的文件
             removeDir(dirPath+'\\'+item)
             # 删除子目录
             os.rmdir(dirPath+'\\'+item)
         # 如果是文件则直接删除
         else:
             os.remove(dirPath+'\\'+item)

 def bugfreeLogin(driver,dic):
     '''
     bugfree登录功能
     :param driver: 浏览器对象
     :param dic: dic数据字典
     :return: 返回dicRe，字典内的键值对与
     '''
     dicRe = {}
     realyResult,testResult,pngPath='','',''
     # 打开登录页面
     driver.get(str(dic['url']))
     driver.find_element_by_id('LoginForm_username').clear()
     # 输入用户名
     driver.find_element_by_id('LoginForm_username').send_keys(str(dic['username']))
     driver.find_element_by_id('LoginForm_password').clear()
     # 输入密码
     driver.find_element_by_id('LoginForm_password').send_keys(str(dic['password']))
     # 点击登录按钮
     driver.find_element_by_id('SubmitLoginBTN').click()
     time.sleep(3)
     #获取提示信息用于判断测试结果，如果登录成功则判断页面展示"欢迎,  系统管理员"，如果登录失败则判断提示信息
     try:
         # 登录成功
         realyResult = driver.find_element_by_css_selector('.user-info').text
     except:
         # 登录失败，取提示信息
         realyResult = driver.find_element_by_id('login-error-div').text
     else:
         #登录成功，退出系统
         driver.find_element_by_link_text('退出').click()

     #判断结果
     if (dic['yuqiResult'] in realyResult):  # 预期结果与实际结果一致
         # 记录测试结果
         testResult = 'PASS'
     else:  ##预期结果与实际结果不一致
         # 记录测试结果
         testResult = 'FAIL'
         # 保存路径
         pngPath = os.getcwd() + '\\errorPng\\' + time.strftime('%Y-%m-%d %H-%M-%S', time.localtime()) + '.png'
         # 截图保存在pngPath变量存储的路径中
         driver.get_screenshot_as_file(pngPath)
     # 往测试结果字典中插入一条测试数据，存储的内容以realyResult、testResult、pngPath为键值
     dicRe.update({'realyResult': realyResult, 'testResult': testResult, 'pngPath': pngPath})
     return dicRe

 if __name__ == '__main__':
     # 调用removeDir删除report目录下的所有文件或目录
     removeDir(os.getcwd()+'\\report')
     # 调用removeDir删除errorpng目录下的所有文件或目录
     removeDir(os.getcwd()+'\\errorpng')
     # 启动Firefox驱动
     driver = webdriver.Firefox()
     # 设置driver驱动的全局定时等待最长时间为5秒
     driver.implicitly_wait(5)
     # 定义空的列表，用于存储测试字典数据
     listRe= []
     # 取到测试数据的excel文档路径
     fpath = os.getcwd()+'\\testData\\bugfree_login.xls'
     # 取到excel表中页签为login的表
     sheetName = 'login'
     # 结果文档的路径
     resultPath= os.getcwd()+'\\report\\test_result '+time.strftime('%Y-%m-%d %H-%M-%S',time.localtime())+'.xls'
     # 读取测试数据
     list1 = getTestData(fpath,sheetName)
     # 此循环用于遍历测试用例中的用例数
     for i in list1:
         # 每一条测试数据，执行一次登录
         dicResult = bugfreeLogin(driver,i)
         # 往列表中存储我们测试结果中的字典数据
         listRe.append(dicResult)
     print listRe
     #往测试结果文档写数据：复制测试数据文档+写入：实际结果、测试结果、截图路径
     writeResultToExcel(fpath, sheetName, listRe, resultPath)
     # 关闭所有浏览器
     driver.quit()