编写一个基于unittest的接口测试框架
时间: 2024-06-08 11:05:44 浏览: 11
下面是一个基于unittest的接口测试框架的示例代码:
```python
import unittest
import requests
class TestAPI(unittest.TestCase):
def setUp(self):
self.base_url = 'http://api.example.com/'
def test_api_1(self):
url = self.base_url + 'endpoint1'
data = {'param1': 'value1', 'param2': 'value2'}
response = requests.post(url, data=data)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json()['result'], 'success')
def test_api_2(self):
url = self.base_url + 'endpoint2'
params = {'param1': 'value1', 'param2': 'value2'}
headers = {'Authorization': 'Bearer token'}
response = requests.get(url, params=params, headers=headers)
self.assertEqual(response.status_code, 200)
self.assertIn('data', response.json())
def tearDown(self):
pass
if __name__ == '__main__':
unittest.main()
```
在这个示例中,我们首先导入了unittest和requests库。然后定义了一个TestAPI类,继承了unittest.TestCase。在setUp方法中,我们定义了接口的基础URL。接下来,我们定义了两个测试用例test_api_1和test_api_2。在每个测试用例中,我们定义了接口的URL、参数、请求头,并使用requests库发送请求,然后使用断言来验证返回结果是否符合预期。最后,在tearDown方法中,我们可以清理测试环境。
运行这个测试框架,可以使用命令行运行:
```
python test_api.py
```
这个测试框架可以自动运行所有的测试用例,并输出结果。如果有测试用例失败,会在控制台输出详细的错误信息。这个测试框架可以很方便地扩展,添加更多的测试用例和断言。