python+unittest+xlrd+request搭建API测试框架

 1 #!/usr/bin/env python
 2 # -*- coding: utf-8 -*-
 3 # @Time    : 2017-07-28 18:07
 4 import xlrd
 5 import requests
 6 class APiTool:
 7 
 8     # 调取表格中用例方法
 9     def xlsee(self, xlsFile):
10         sheetlist = []  # 用来保存表格所有数据
11         rqapi = xlrd.open_workbook(xlsFile)   # 获得文件对象
12         sheet_name = rqapi.sheet_names()[0]  # 获取第一个sheet名称
13         sheet = rqapi.sheet_by_name(sheet_name)  # 获取第一个sheet对象
14         nrow = sheet.nrows   # 获取行总数
15         for i in range(1,nrow):
16             sheetlist.append(sheet.row_values(i))
17         return sheetlist
18 
19     # 请求方法
20     def request(self, rqtype, rqurl, paramete, headers):
21         if rqtype == "get":
22             apiresult = requests.get(url=rqurl, params=paramete, headers=headers)  # 发送请求
23             return apiresult
24         if rqtype == "post":
25             apiresult = requests.post(url=rqurl, data=paramete, headers=headers)
26             return apiresult
27         else:
28             print("请求参数错误，请求类型只支持get+post，请求地址支持string，参数支持dict")
29 
30 
31     # 对返回的json值进行深度断言
32 
33     def compare_json_data(self,A, B, L = [], xpath = ‘.‘):
34         if isinstance(A, list) and isinstance(B, list):
35             for i in range(len(A)):
36                 try:
37                     self.compare_json_data(A[i], B[i], L, xpath + ‘[%s]‘ % str(i))
38                 except:
39                     L.append(‘▇▇▇ A中的key %s[%s]未在B中找到\n‘ % (xpath, i))
40         if isinstance(A, dict) and isinstance(B, dict):
41             for i in A:
42                 try:
43                     B[i]
44                 except:
45                     L.append(‘▇▇▇ A中的key %s/%s 未在B中找到\n‘ % (xpath, i))
46                     continue
47                 if not (isinstance(A.get(i), (list, dict)) or isinstance(B.get(i), (list, dict))):
48                     if type(A.get(i)) != type(B.get(i)):
49                         L.append(‘▇▇▇ 类型不同参数在[A]中的绝对路径:  %s/%s  ??? A is %s, B is %s \n‘ % (xpath, i, type(A.get(i)), type(B.get(i))))
50                     elif A.get(i) != B.get(i):
51                         L.append(‘▇▇▇ 仅内容不同参数在[A]中的绝对路径:  %s/%s  ??? A is %s, B is %s \n‘ % (xpath, i, A.get(i), B.get(i)))
52                     continue
53                 self.compare_json_data(A.get(i), B.get(i), L, xpath + ‘/‘ + str(i))
54             return
55         if type(A) != type(B):
56             L.append(‘▇▇▇ 类型不同参数在[A]中的绝对路径:  %s  ??? A is %s, B is %s \n‘ % (xpath, type(A), type(B)))
57         elif A != B and type(A) is not list:
58             L.append(‘▇▇▇ 仅内容不同参数在[A]中的绝对路径:  %s  ??? A is %s, B is %s \n‘ % (xpath, A, B))
59         return L
60 
61     def Assert(self,A,B):
62         C = []
63         self.compare_json_data(A, B, C)
64         assert len(C) == 0, "\n"+"".join(C)