1 #encoding=utf-8
2 import xlrd,time,os
3 from xlutils.copy import copy
4 from selenium import webdriver
5
6 def getTestData(excelPath,sheetName):
7 '''
8 定义从excel文件中获取数据
9 :param excelPath: excel文件路径
10 :param sheetName: 表格中的页签名
11 :return: 返回测试数据字段
12 '''
13 # 打开存有测试数据的excel
14 xl = xlrd.open_workbook(excelPath)
15 # 指定打开某个页签的table
16 table = xl.sheet_by_name(sheetName)
17 # 获取表中的第一行，用来做字典的key进行存储,返回的是列表从第一列到最后一列
18 listKey = table.row_values(0)
19 # 用于存放测试数据
20 listTestData = []
21 # 外围for循环取到的是行
22 for i in range(1,table.nrows):
23 # 将存取的每次数据存在空字典内
24 dicData = {}
25 # 内部for循环取到的是列
26 for j in range(0,table.ncols):
27 # key为第一行的每列，value为每行的列值
28 dicData.update({listKey[j]:table.cell_value(i,j)})
29 listTestData.append(dicData)
30 print('取到的数据为：')
31 # 遍历取到的每行数据
32 for rowData in listTestData:
33 print(rowData)
34 # 返回测试数据
35 return listTestData
36
37 def writeResultToExcel(excelPath,sheetName,listRe,reResultFiles):
38 '''
39 定义函数，先复制测试用例表格模板，再往里面写入实际结果，测试结果以及截图路径
40 :param excelPath: 需要复制的excel模板路径
41 :param sheetName: 被复制末班中的页签名
42 :param listRe: 列表类型，其内部数据为字典类型，列表中包含实际结果，测试结果，截图路径的存取数据字典
43 :param reResultFiles: 写入完成后保存的excel路径
44 :return:
45 '''
46 # 打开指定路径的excel文件
47 xl = xlrd.open_workbook(excelPath,encoding_override='utf-8')
48 # 打开对应页签的表
49 table = xl.sheet_by_name(sheetName)
50 # 复制模板
51 excelMudle = copy(xl)
52 # 获取到sheetName对应的表
53 mudleTable = excelMudle.get_sheet(sheetName)
54 # 获取模板表中有多少列数
55 cols = table.ncols
56 # 获取模板表中有多少行数
57 rows = table.nrows
58 for rowsName in listRe[0].keys():
59 # 每次往第一行最新的列写入表头
60 mudleTable.write(0,cols,rowsName)
61 # 列索引往后移动一位
62 cols+=1
63 #逐行写数据，从第2行开始写，写到最后一行，每行是一次测试结果，每次测试结果放在字典中
64 for i in range(1,rows):
65 '''
66 listRe[0]是第二行索引为1的测试结果，listRe[1]是第三行索引为2的测试结果,行值和数据存储存字典索引相差1
67 每次往第i行写数据 写 listRe[i-1]的数据
68 实际结果,i是行索引,cols-3是列的倒数第3个列，cols-2是列的倒数第2个列，cols-1是列的倒数第1个列
69 '''
70 mudleTable.write(i,cols-3,listRe[i-1]['realyResult'])
71 # 测试结果
72 mudleTable.write(i,cols-2,listRe[i-1]['testResult'])
73 # 截图路径
74 mudleTable.write(i,cols-1,listRe[i-1]['pngPath'])
75 # 保存测试结果
76 excelMudle.save(reResultFiles)
77
78 def removeDir(dirPath):
79 '''
80 删除Windows下文件夹里面的所有文件和子文件夹
81 :param dirPath: 要删除的文件夹路径
82 :return: None
83 '''
84 # 读取要删除的目录下所有文件和子目录
85 listTopDir = os.listdir(dirPath)
86 for item in listTopDir:
87 # 如果是子目录则递归删除
88 if os.path.isdir(dirPath+'\\'+item):
89 # 删除子目录中的文件
90 removeDir(dirPath+'\\'+item)
91 # 删除子目录
92 os.rmdir(dirPath+'\\'+item)
93 # 如果是文件则直接删除
94 else:
95 os.remove(dirPath+'\\'+item)
96
97 def bugfreeLogin(driver,dic):
98 '''
99 bugfree登录功能
100 :param driver: 浏览器对象
101 :param dic: dic数据字典
102 :return: 返回dicRe，字典内的键值对与
103 '''
104 dicRe = {}
105 realyResult,testResult,pngPath='','',''
106 # 打开登录页面
107 driver.get(str(dic['url']))
108 driver.find_element_by_id('LoginForm_username').clear()
109 # 输入用户名
110 driver.find_element_by_id('LoginForm_username').send_keys(str(dic['username']))
111 driver.find_element_by_id('LoginForm_password').clear()
112 # 输入密码
113 driver.find_element_by_id('LoginForm_password').send_keys(str(dic['password']))
114 # 点击登录按钮
115 driver.find_element_by_id('SubmitLoginBTN').click()
116 time.sleep(3)
117 #获取提示信息用于判断测试结果，如果登录成功则判断页面展示"欢迎, 系统管理员"，如果登录失败则判断提示信息
118 try:
119 # 登录成功
120 realyResult = driver.find_element_by_css_selector('.user-info').text
121 except:
122 # 登录失败，取提示信息
123 realyResult = driver.find_element_by_id('login-error-div').text
124 else:
125 #登录成功，退出系统
126 driver.find_element_by_link_text('退出').click()
127
128 #判断结果
129 if (dic['yuqiResult'] in realyResult): # 预期结果与实际结果一致
130 # 记录测试结果
131 testResult = 'PASS'
132 else: ##预期结果与实际结果不一致
133 # 记录测试结果
134 testResult = 'FAIL'
135 # 保存路径
136 pngPath = os.getcwd() + '\\errorPng\\' + time.strftime('%Y-%m-%d %H-%M-%S', time.localtime()) + '.png'
137 # 截图保存在pngPath变量存储的路径中
138 driver.get_screenshot_as_file(pngPath)
139 # 往测试结果字典中插入一条测试数据，存储的内容以realyResult、testResult、pngPath为键值
140 dicRe.update({'realyResult': realyResult, 'testResult': testResult, 'pngPath': pngPath})
141 return dicRe
142
143 if __name__ == '__main__':
144 # 调用removeDir删除report目录下的所有文件或目录
145 removeDir(os.getcwd()+'\\report')
146 # 调用removeDir删除errorpng目录下的所有文件或目录
147 removeDir(os.getcwd()+'\\errorpng')
148 # 启动Firefox驱动
149 driver = webdriver.Firefox()
150 # 设置driver驱动的全局定时等待最长时间为5秒
151 driver.implicitly_wait(5)
152 # 定义空的列表，用于存储测试字典数据
153 listRe= []
154 # 取到测试数据的excel文档路径
155 fpath = os.getcwd()+'\\testData\\bugfree_login.xls'
156 # 取到excel表中页签为login的表
157 sheetName = 'login'
158 # 结果文档的路径
159 resultPath= os.getcwd()+'\\report\\test_result '+time.strftime('%Y-%m-%d %H-%M-%S',time.localtime())+'.xls'
160 # 读取测试数据
161 list1 = getTestData(fpath,sheetName)
162 # 此循环用于遍历测试用例中的用例数
163 for i in list1:
164 # 每一条测试数据，执行一次登录
165 dicResult = bugfreeLogin(driver,i)
166 # 往列表中存储我们测试结果中的字典数据
167 listRe.append(dicResult)
168 print listRe
169 #往测试结果文档写数据：复制测试数据文档+写入：实际结果、测试结果、截图路径
170 writeResultToExcel(fpath, sheetName, listRe, resultPath)
171 # 关闭所有浏览器
172 driver.quit()
