当前位置：首页 > article >正文

接口自动化框架篇：Pytest中的接口请求封装！

article 2025/2/23 21:07:31

在进行接口自动化测试时，我们需要一个稳定、易用且可扩展的框架来封装接口请求。Pytest是一种非常流行的Python测试框架，它提供了很多方便的功能和插件，可以帮助我们轻松进行接口自动化测试。

接口请求封装是框架的核心部分，它负责发送HTTP请求、处理响应和断言结果。以下是一个从0到1的规划，详细介绍如何在Pytest中实现接口请求封装。

一、安装Pytest和相关插件

首先，我们需要安装Pytest和相关的插件。可以通过以下命令来安装：

pip install pytest
pip install requests

二、创建一个基础的请求类

接口请求封装可以通过一个基础的请求类来实现。这个类可以包含发送请求、处理响应和断言结果的方法。以下是一个基础请求类的示例：

import requests

class BaseRequest:
    def __init__(self):
        self.session = requests.Session()

    def send_request(self, method, url, params=None, data=None, headers=None):
        # 发送请求
        response = self.session.request(method, url, params=params, data=data, headers=headers)
        return response

    def assert_response(self, response, status_code=200, data=None):
        # 断言响应结果
        assert response.status_code == status_code
        if data is not None:
            assert response.json() == data

在这个基础请求类中，我们使用了requests库来发送请求，并使用assert语句来断言响应结果。这个类还使用了Session对象来保持请求的会话，以便在多个请求之间共享cookie和其他会话信息。

三、创建一个接口类

接口类是基于基础请求类的封装，它可以定义具体的接口和参数。以下是一个接口类的示例：

class TestAPI:
    def __init__(self):
        self.base_request = BaseRequest()

    def login(self, username, password):
        # 登录接口
        url = 'http://api.example.com/login'
        data = {
            'username': username,
            'password': password
        }
        response = self.base_request.send_request('POST', url, data=data)
        self.base_request.assert_response(response, data={'success': True})

    def get_user_info(self, user_id):
        # 获取用户信息接口
        url = 'http://api.example.com/user_info'
        params = {
            'user_id': user_id
        }
        response = self.base_request.send_request('GET', url, params=params)
        self.base_request.assert_response(response)
        return response.json()

在这个接口类中，我们使用了基础请求类来发送请求和断言响应结果。接口类封装了具体的接口和参数，并返回响应结果供后续使用。

四、编写测试用例

有了接口类，我们就可以编写测试用例来进行接口自动化测试了。以下是一个测试用例的示例：

class TestAPI:
    def test_login(self):
        # 登录接口测试用例
        api = TestAPI()
        api.login('admin', '123456')

    def test_get_user_info(self):
        # 获取用户信息接口测试用例
        api = TestAPI()
        user_info = api.get_user_info(1)
        assert user_info['id'] == 1

在这个测试用例中，我们创建了一个TestAPI对象，并调用了其中的方法来测试接口。我们可以使用assert语句来断言测试结果是否符合预期。