python接口自动化框架

# Python接口自动化框架示例代码

import requests
import unittest

class TestAPI(unittest.TestCase):
    def setUp(self):
        # 初始化测试环境，比如设置基础URL等
        self.base_url = "https://api.example.com"

    def test_get_user(self):
        # 测试获取用户信息的接口
        endpoint = "/users/123"
        url = self.base_url + endpoint

        # 发送GET请求
        response = requests.get(url)

        # 断言状态码为200
        self.assertEqual(response.status_code, 200)

        # 获取响应数据
        data = response.json()

        # 断言响应数据中的某些字段
        self.assertEqual(data['id'], 123)
        self.assertEqual(data['name'], 'John Doe')

    def test_create_user(self):
        # 测试创建用户的接口
        endpoint = "/users"
        url = self.base_url + endpoint

        # 准备POST请求的数据
        user_data = {
            "name": "Jane Doe",
            "email": "jane.doe@example.com"
        }

        # 发送POST请求
        response = requests.post(url, json=user_data)

        # 断言状态码为201（创建成功）
        self.assertEqual(response.status_code, 201)

        # 获取响应数据
        data = response.json()

        # 断言响应数据中的某些字段
        self.assertEqual(data['name'], 'Jane Doe')
        self.assertEqual(data['email'], 'jane.doe@example.com')

if __name__ == "__main__":
    unittest.main()

解释说明：

1. 导入库：

   • requests：用于发送HTTP请求。
   • unittest：Python自带的单元测试框架，用于编写和运行测试用例。

2. 类定义：

   • TestAPI：继承自unittest.TestCase，表示这是一个测试类。
   • setUp方法：在每个测试方法执行前都会调用，用于初始化测试环境。这里设置了API的基础URL。

3. 测试方法：

   • test_get_user：测试获取用户信息的接口。
     • 使用requests.get发送GET请求。
     • 使用self.assertEqual进行断言，确保返回的状态码和响应数据符合预期。
   • test_create_user：测试创建用户的接口。
     • 使用requests.post发送POST请求，并传递JSON格式的数据。
     • 同样使用断言确保创建操作成功，并验证返回的数据。

4. 运行测试：

   • 如果直接运行这个脚本，unittest.main()会自动发现并执行所有以test_开头的方法。

这个示例展示了如何使用Python编写一个简单的接口自动化测试框架。你可以根据实际需求扩展更多的测试用例和功能。