接口调用单元测试
时间: 2023-09-03 20:12:49 浏览: 46
接口调用单元测试是指对某个接口进行测试,以确保接口的功能正常、输入输出符合预期、能够正确处理各种异常情况等。下面是一个基本的接口调用单元测试的示例:
```python
import unittest
import requests
class TestAPI(unittest.TestCase):
def setUp(self):
self.base_url = 'http://localhost:8000/api/'
def test_get_data(self):
response = requests.get(self.base_url + 'data/')
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json(), {'data': 'Hello World'})
def test_post_data(self):
data = {'message': 'Hello World'}
response = requests.post(self.base_url + 'data/', data=data)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json(), {'status': 'success'})
def test_invalid_input(self):
data = {'message': ''}
response = requests.post(self.base_url + 'data/', data=data)
self.assertEqual(response.status_code, 400)
self.assertEqual(response.json(), {'error': 'Invalid input'})
if __name__ == '__main__':
unittest.main()
```
这个示例中,我们定义了三个测试用例,分别测试了获取数据、提交数据和输入无效数据的情况。在每个测试用例中,我们使用 requests 库发起 HTTP 请求,并检查响应的状态码和内容是否符合预期。最后,在 `if __name__ == '__main__':` 中调用 `unittest.main()` 来运行测试用例。