您现在的位置是：首页 >技术杂谈 >利用Python如何实现数据驱动的接口自动化测试网站首页技术杂谈

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

import csv

import json

import requests

import time

import unittest

def readCSV(self,filename):

 '''

 :param filename: 需要读取的数据文件

 :return: [{data1},{data2}...]

 '''

 datas = []

 try:

 #以DictReader的方式读取数据文件，方便与json互做转换

 with open(filename,'r') as csvfile :

 #从文件里读取到的数据转换成字典列表的格式

 reader = csv.DictReader(csvfile)

 for row in reader:

 data = {}

 data['id'] = row['id']

 data['url'] = row['url']

 data['token'] = str(row['token'])

 data['mobile'] = row['mobile']

 data['email'] = row['email']

 data['expect'] = json.dumps(row['expect'])

 if isinstance(row['expect'],dict)

 else row['expect'] #如果expect读取出来的不是json则取其原值，否则转为json格式保存到result里

 datas.append(data)

 return datas

 #如果文件找不到，返回空的datas

 except FileNotFoundError:

 print("文件不存在",filename)

 return datas

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

def get_request(self,url,params):

 '''

 通用的调用GET接口方法

 :param url:string 接口路径

 :param params:{"":"","":""} 需要传入的参数

 :return: response响应体

 '''

 print("调用API...")

 r = requests.get(url,params=params)

 print(r.text)

 return r

def post_request(self,url,params):

 '''

 通用的调用POST接口方法

 :param url: string 接口路径

 :param params: {"":"","":""} 需要传入的参数

 :return:response响应体

 '''

 print("调用API...")

 r = requests.post(url,params=json.dumps(params)) #post的方法必须用json.dumps()转化成json格式

 print(r.text)

 return r

1

2

3

4

5

6

7

8

9

def assertResult(self,except_value,real_value):

 '''

 校验样本字符串中是否包含指定字符串

 :param except_value: string 指定字符串

 :param real_value: string 样本字符串

 :return: Boolean 样本中包含指定字符串返回True,否则返回False

 '''

 ifsuccess = except_value in str(real_value)

 return ifsuccess

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

def writeCSV(self,filename,results):

 '''

 写入csv文件指定内容

 :param filename: string 需要写入的文件名称

 :param results: [{data1},{data2},...] 写入的内容

 :return: 无

 '''

 print("写文件:",filename)

 #以DictWriter的方式写文件

 with open(filename,'w+') as csvfile:

 headers="id,url,token,mobile,email,expect,real_value,assert_value".split(",")

 writer = csv.DictWriter(csvfile,fieldnames=headers)

 #写表头

 writer.writeheader()

 #写数据

 if results.__len__() > 0 :

 for result in results:

 writer.writerow(result)

 csvfile.close()

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

def test_interface1(self):

 #指定读取的数据文件名称

 data_file = "../data/data.csv"

 #指定最终结果生成的数据文件名称

 result_file = "../data/result_{}.csv".format(str(time.time()).split(".")[0])

 #读取指定文件的数据

 datas = self.readCSV(data_file)

 #数据文件有内容则调用接口，否则直接测试结束

 if datas.__len__() > 0:

 results =[]

 #获取数据文件里的每一行

 for testcase in datas :

 result = {}

 result["id"] = testcase["id"]

 result["url"] = testcase["url"]

 result["token"] = testcase["token"]

 result["mobile"] = testcase["mobile"]

 result["email"] = testcase["email"]

 result["expect"] = testcase["expect"]

 #组装参数

 params = {

 "token":result["token"],

 "mobile":result["mobile"],

 "email":result["email"]

 }

 #调用API接口，获取响应结果

 real_value = self.get_request(result["url"],params)

 #调用assert方法，检查预期结果是否在响应结果中存在

 assert_value = self.assertResult(result["expect"],real_value.text)

 result["real_value"] = real_value.text

 result["assert_value"] = assert_value

 #获取每一行里的所有字段以及实际结果和验证结果

 results.append(result)

 #执行完所有记录后，将所有结果写入result.csv

 self.writeCSV(result_file,results) #写入csv文件

 print("测试结束")