python unittest requests框架代码实现
时间: 2023-08-16 08:08:58 浏览: 102
python+unittest+requests实现接口自动化的方法
以下是一个使用python unittest和requests库进行接口自动化测试的示例代码:
```python
import unittest
import requests
class TestApi(unittest.TestCase):
def setUp(self):
self.url = 'http://example.com/api'
self.headers = {'Content-Type': 'application/json'}
def test_api_response(self):
data = {'param1': 'value1', 'param2': 'value2'}
response = requests.post(self.url, headers=self.headers, json=data)
self.assertEqual(response.status_code, 200)
self.assertIn('success', response.json())
def test_api_result(self):
data = {'param1': 'value1', 'param2': 'value2'}
response = requests.post(self.url, headers=self.headers, json=data)
self.assertEqual(response.status_code, 200)
result = response.json()['result']
self.assertEqual(result, 'expected_result')
def tearDown(self):
pass
if __name__ == '__main__':
unittest.main()
```
在该示例代码中,我们定义了一个TestApi类来执行测试。在setUp()方法中,我们设置了API的URL和请求头。在test_api_response()方法中,我们发送了一个POST请求并断言响应的状态码和返回结果中是否包含“success”字符串。在test_api_result()方法中,我们发送了一个POST请求并断言响应的状态码和返回结果中的“result”值是否等于预期值。最后,我们使用unittest.main()方法运行测试。
阅读全文