一个数据驱动的API测试框架

```python
import requests
import json
import csv
import unittest
from HTMLTestRunner import HTMLTestRunner
# 1. 从单独文件中读取配置数据
def read_test_data_from_csv(csv_file):
test_data = []
with open(csv_file, 'r', newline='') as file:
reader = csv.DictReader(file)
for row in reader:
test_data.append(row)
return test_data
# 2. 接口输入参数使用json或者csv
def run_api_test(data):
test_results = []
for test_case in data:
url = test_case['url']
method = test_case['method']
headers = json.loads(test_case['headers'])
payload = json.loads(test_case['payload'])
expected_response_code = int(test_case['expected_response_code'])
response = None
try:
if method == 'GET':
response = requests.get(url, headers=headers)
elif method == 'POST':
response = requests.post(url, headers=headers, json=payload)
# 检查响应状态码
if response.status_code == expected_response_code:
test_results.append({'test_case': test_case, 'result': 'Pass'})
else:
test_results.append({'test_case': test_case, 'result': 'Fail', 'reason': f'Expected {expected_response_code}, but got {response.status_code}'})
except Exception as e:
test_results.append({'test_case': test_case, 'result': 'Error', 'reason': str(e)})
return test_results
# 3. 输出结果使用html方式
def generate_html_report(test_results):
with open('test_report.html', 'w', encoding='utf-8') as file:
runner = HTMLTestRunner(stream=file, title='API Test Report', description='Test results for API tests')
runner.run(unittest.TestLoader().loadTestsFromTestCase(ApiTest))
# 4. 测试报告，要有统计测试用例失败以及失败原因
class ApiTest(unittest.TestCase):
def test_api(self):
test_data = read_test_data_from_csv('test_data.csv')
results = run_api_test(test_data)
for result in results:
self.assertEqual(result['result'], 'Pass', result.get('reason', ''))
if __name__ == '__main__':
unittest.main()
generate_html_report()
```