csv 文件读写

https://docs.python.org/zh-cn/3/library/csv.html
可以使用csv模块进行csv文件数据的读写。 csv模块是python内置的，所以不需要安装，可以直接使用。

csv文件写入

csv_demo.py

  1. # 导入csv模块
  2. import csv
  3. # 打开csv文件，权限设置为写入 设置不要换行
  4. with open('data.csv',mode='w',encoding='utf8',newline='') as file:
  5.     # 使用csv模块往文件中写入内容
  6.     wr = csv.writer(file)
  7.     # 写入一行内容 使用列表格式数据
  8.     wr.writerow(["用户名","密码","期望状态码"])
  9.     for i in range(10):
  10.         wr.writerow([f'test{i}',"123456",200])

csv文件读取

csv_demo.py

  1. # 读取csv文件  文件权限为 r 读取
  2. with open('data.csv',mode='r',encoding='utf8') as file:
  3.     # 读取文件
  4.     rd = csv.reader(file)
  5.     # 循环读取
  6.     for line in rd:
  7.         print(line)

接口测试CSV数据驱动

也可以将测试数据放到csv文件中，通过读取csv文件的内容实现接口的自动化。

准备测试数据

data.csv

  1. 用户名,密码,期望状态码,期望的message
  2. ,123456,510,登录名不能为空
  3. 1321111,123456,500,请输入正确的手机号！
  4. 13213331333,,510,密码不能为空
  5. ,,510,密码不能为空
  6. 13213331333,123456,500,登录失败！

读取csv文件进行接口自动化

test_csv_data.py

  1. """
  2. 读取csv文件中的内容，进行数据驱动接口测试
  3. """
  5. import requests
  6. import pytest
  7. import csv
  8. def read_csv():
  9.     testdata = []
  10.     with open('data.csv',mode='r',encoding='utf8') as file:
  11.         rd = csv.reader(file)
  12.         # 去掉第一行数据
  13.         next(rd)
  14.         for line in rd:
  15.             testdata.append(tuple(line))
  17.     return testdata
  19. # 读取csv文件中的数据
  20. testdata = read_csv()
  21. print(testdata)
  24. @pytest.mark.parametrize("username,passwd,excpet_code,excpet_msg",testdata)
  25. def test_login(username,passwd,excpet_code,excpet_msg):
  26.     url = "http://49.233.108.117:28019/api/v1/user/login"
  27.     body = {
  28.         "loginName": username,
  29.         "passwordMd5": passwd
  30.     }
  31.     # 发送请求
  32.     r = requests.post(url,json=body)
  33.     print(r.json())
  34.     # 添加断言 因为从csv文件中读取出来的数字，会自动被转换为字符串
  35.     assert r.json()["resultCode"] == int(excpet_code)  # 将csv文件中的字符串转为数字
  36.     assert r.json()["message"] == excpet_msg

运行

  1. pytest -s -v test_csv_data.py

总结

从csv文件中读取数据，使用csv模块，csv文件中读取的数据默认为字符串类型，在加断言的时候，针对状态码进行断言，断言的结果中需要将字符串再转换为数字。

csv数据驱动参数化，本质上还是从 将数据从文件中读取出来，将数据转换为列表格式， 使用pytest的参数化功能进行自动化。

csv参数化

topic.csv

  1. token,title,tab,content,期望的状态码,期望的错误提示信息
  2. ,helloworld,ask,helloworld,401,错误的accessToken
  3. 999900,helloworld,ask,helloworld,401,错误的accessToken
  4. 6c231bd4-1180-466e-9c01-dfc83eaddbc8,,ask,helloworld,400,标题不能为空
  5. 6c231bd4-1180-466e-9c01-dfc83eaddbc8,helloworld,,helloworkld,400,必须选择一个版块
  6. 6c231bd4-1180-466e-9c01-dfc83eaddbc8,helloworld,ask,,400,内容不可为空

test_csv_data.py

  1. """
  2. 读取csv文件中的内容，进行数据驱动接口测试
  3. """
  5. import requests
  6. import pytest
  7. import csv
  8. def read_csv(filename):
  9.     testdata = []
  10.     with open(filename,mode='r',encoding='utf8') as file:
  11.         rd = csv.reader(file)
  12.         # 去掉第一行数据
  13.         next(rd)
  14.         for line in rd:
  15.             testdata.append(tuple(line))
  17.     return testdata
  19. # 读取csv文件中的数据
  20. testdata = read_csv('data.csv')
  21. print(testdata)
  24. @pytest.mark.parametrize("username,passwd,excpet_code,excpet_msg",testdata)
  25. def test_login(username,passwd,excpet_code,excpet_msg):
  26.     url = "http://49.233.108.117:28019/api/v1/user/login"
  27.     body = {
  28.         "loginName": username,
  29.         "passwordMd5": passwd
  30.     }
  31.     # 发送请求
  32.     r = requests.post(url,json=body)
  33.     print(r.json())
  34.     # 添加断言 因为从csv文件中读取出来的数字，会自动被转换为字符串
  35.     assert r.json()["resultCode"] == int(excpet_code)  # 将csv文件中的字符串转为数字
  36.     assert r.json()["message"] == excpet_msg
  38. test_topic_data = read_csv('topic.csv')
  39. @pytest.mark.parametrize('token,title,tab,content,expect_code,expect_msg',test_topic_data)
  40. def test_topics(token,title,tab,content,expect_code,expect_msg):
  41.     url = "http://47.100.175.62:3000/api/v1/topics"
  42.     body = {
  43.         "accesstoken":token,
  44.         "title":title,
  45.         "tab":tab,
  46.         "content":content
  47.     }
  48.     r = requests.post(url,json=body)
  49.     print(r.status_code,r.json())
  51.     # 状态码断言
  52.     assert r.status_code == int(expect_code)
  54.     # 服务器返回结果断言
  55.     assert r.json()["error_msg"] == expect_msg