python利用Excel读取和存储测试数据完成接口自动化教程

#读取多条测试用例
#1、导入requests模块
import requests
#从 class_12_19.do_excel1导入read_data函数
from do_excel2 import read_data
from do_excel2 import write_data
from do_excel2 import count_case
#定义http请求函数

COOKIE=None
def http_request2(method,url,data):

 if method=='get':
  print('发起一个get请求')
  result=requests.get(url,data,cookies=COOKIE)
 else:
  print('发起一个post请求')
  result=requests.post(url,data,cookies=COOKIE)
 return result #返回响应体
 # return result.json() #返回响应结果：结果是字典类型：{'status': 1, 'code': '10001', 'data': None, 'msg': '登录成功'}


#从Excel读取到多条测试数据
sheets=['login','recharge','withdraw']
for sheet1 in sheets:
 max_row=count_case(sheet1)
 print(max_row)
 for case_id in range(1,max_row):
  data=read_data(sheet1,case_id)
  print('读取到第{}条测试用例:'.format(data[0]))
  print('测试数据 ',data)
  #print(type(data[2]))
  #调用函数发起http请求
  result=http_request2(data[4],data[2],eval(data[3]))
  print('响应结果为 ',result.json())
  if result.cookies:
    COOKIE=result.cookies

   #将测试实际结果写入excel
   #write_data(case_id+1,6,result['code'])
  write_data(sheet1,case_id+1,7,str(result.json()))
  #对比测试结果和期望结果
  if result.json()['code']==str(data[5]):
   print('测试通过')
   #将用例执行结果写入Excel
   write_data(sheet1,case_id+1,8,'Pass')
  else:
   write_data(sheet1,case_id+1,8,'Fail')
   print('测试失败')

# 导入load_workbook
from openpyxl import load_workbook

#读取测试数据
#将excel中每一条测试用例读取到一个列表中

#读取一条测试用例——写到一个函数中
def read_data(sheet_name,case_id):
 # 打开excel
 workbook1=load_workbook('test_case2.xlsx')
 # 定位表单（test_data）
 sheet1=workbook1[sheet_name]
 print(sheet1)
 test_case=[] #用来存储每一行数据，也就是一条测试用例
 test_case.append(sheet1.cell(case_id+1,1).value)
 test_case.append(sheet1.cell(case_id+1,2).value)
 test_case.append(sheet1.cell(case_id+1,3).value)
 test_case.append(sheet1.cell(case_id+1,4).value)
 test_case.append(sheet1.cell(case_id+1,5).value)
 test_case.append(sheet1.cell(case_id+1,6).value)
 return test_case #将读取到的用例返回

#调用函数读取第1条测试用例，并将返回结果保存在data中
# data=read_data(1)
# print(data)

#将测试结果写会excel
def write_data(sheet_name,row,col,value):
 workbook1=load_workbook('test_case2.xlsx')
 sheet=workbook1[sheet_name]
 sheet.cell(row,col).value=value
 workbook1.save('test_case2.xlsx')


#统计测试用例的行数
def count_case(sheet_name):
  workbook1=load_workbook('test_case2.xlsx')
  sheet=workbook1[sheet_name]
  max_row=sheet.max_row #统计测试用例的行数
  return max_row

import csv # 导入scv库，可以读取csv文件
from selenium import webdriver
import unittest
from time import sleep
import time
import os
import HTMLTestRunner
import codecs
import sys
dr = webdriver.Chrome()
class testLo(unittest.TestCase):
def setUp(self):
pass
def test_login(self):
'''登陆测试'''
path = 'F:\\Python_test\\'
# 要读取的scv文件路径
my_file = 'F:\\pythonproject\\interfaceTest\\testFile\\ss.csv'
# csv.reader()读取csv文件，
# Python3.X用open，Python2.X用file，'r'为读取
# open(file,'r')中'r'为读取权限，w为写入，还有rb，wd等涉及到编码的读写属性
#data = csv.reader(codecs.open(my_file, 'r', encoding='UTF-8',errors= 'ignore'))
with codecs.open(my_file, 'r', encoding='UTF-8',errors= 'ignore') as f:
data=csv.reader((line.replace('\x00','') for line in f))
# for循环将读取到的csv文件的内容一行行循环，这里定义了user变量(可自定义)
# user[0]表示csv文件的第一列，user[1]表示第二列，user[N]表示第N列
# for循环有个缺点，就是一旦遇到错误，循环就停止，所以用try，except保证循环执行完
print(my_file)
for user in data:
print(user)
dr.get('https://passport.cnblogs.com/user/signin')
# dr.find_element_by_id('input1').clear()
dr.find_element_by_id('input1').send_keys(user[0])
# dr.find_element_by_id('input2').clear()
dr.find_element_by_id('input2').send_keys(user[1])
dr.find_element_by_id('signin').click()
sleep(1)
print('\n' + '测试项：' + user[2])
dr.get_screenshot_as_file(path + user[3] + ".jpg")
try:
assert dr.find_element_by_id(user[4]).text
try:
error_message = dr.find_element_by_id(user[4]).text
self.assertEqual(error_message, user[5])
print('提示信息正确！预期值与实际值一致:')
print('预期值：' + user[5])
print('实际值:' + error_message)
except:
print('提示信息错误！预期值与实际值不符：')
print('预期值：' + user[5])
print('实际值:' + error_message)
except:
print('提示信息类型错误,请确认元素名称是否正确！')
def tearDown(self):
dr.refresh()
# 关闭浏览器
dr.quit()
if __name__ == "__main__":
# 定义脚本标题，加u为了防止中文乱码
report_title = u'登陆模块测试报告'
# 定义脚本内容，加u为了防止中文乱码
desc = u'登陆模块测试报告详情：'
# 定义date为日期，time为时间
date = time.strftime("%Y%m%d")
time = time.strftime("%Y%m%d%H%M%S")
# 定义path为文件路径，目录级别，可根据实际情况自定义修改
path = 'F:\\Python_test\\' + date + "\\login\\" + time + "\\"
# 定义报告文件路径和名字，路径为前面定义的path，名字为report（可自定义），格式为.html
report_path = path + "report.html"
# 判断是否定义的路径目录存在，不能存在则创建
if not os.path.exists(path):
os.makedirs(path)
else:
pass
# 定义一个测试容器
testsuite = unittest.TestSuite()
# 将测试用例添加到容器
testsuite.addTest(testLo("test_login"))
# 将运行结果保存到report，名字为定义的路径和文件名，运行脚本
report = open(report_path, 'wb')
#with open(report_path, 'wb') as report:
runner = HTMLTestRunner.HTMLTestRunner(stream=report, title=report_title, description=desc)
runner.run(testsuite)
# 关闭report，脚本结束
report.close()

''''' 
将结果导出到result.csv中,以UTF_8 with BOM编码（微软产品能正确识别UTF_8 with BOM存储的中文文件）存储 
''' 
#data.to_csv('result_utf8_no_bom.csv',encoding='utf_8')#导出的结果不能别excel正确识别 
data.to_csv('result_utf8_with_bom.csv',encoding='utf_8_sig')