python+pytest接口自动化之测试函数、测试类/测试方法的封装-Toy模板网

这篇具有很好参考价值的文章主要介绍了python+pytest接口自动化之测试函数、测试类/测试方法的封装。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

前言

今天呢，笔者想和大家聊聊python+pytest接口自动化中将代码进行封装，只有将测试代码进行封装，才能被测试框架识别执行。

例如单个接口的请求代码如下：

import requests

headers = {
    "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36"
}

url = "https://mp.toutiao.com/profile_v4/"
res = requests.get(url=h_url, headers=headers)

假设我们需要将上面这段代码编写成测试框架能执行的测试用例，仅仅只是这样写代码显然是不够的，还需要进行如下补充：

需要将代码封装成单元测试框架 (pytest或unittest) 能识别的测试函数或测试类，否则将不会被识别执行。
需要加上断言，即结果与期望之间的对比，单元测试框架才能判定该用例执行结果是否通过，结果==期望则说明通过，否则失败。

python中函数以及类的封装这里不做过多说明，这篇文章的目的是让大家明白在接口自动化测试中一般怎样封装测试代码。

一、测试用例封装的一般规则

测试用例的封装有两种，测试函数和测试类，封装的一般规则如下：

一个测试函数对应一条测试用例。
测试类中可定义多个测试方法，一个测试方法对应一条测试用例，测试类可以看作是一个测试用例集。
pytest中测试函数或测试方法的命名必须以test开头，测试类名必须以Test开头。具体命名规则可以参考我之前的文章pytest(3)-测试命名规则。
对于单接口的测试校验，一个单接口的测试用例只包含一个接口请求，即将一个接口请求封装成一个测试函数或测试方法。
对于场景(多接口) 的测试校验，一条场景测试用例需请求多个接口，因此需要将多个接口请求封装在同一个测试函数或方法中。
一般封装一个接口的正向校验、异常校验封装成不同的方法，并封装在同一个测试类中。如定义一个登陆的测试类，正确用户名、密码请求封装成一个方法 (即一条测试用例)，正确用户名、错误密码请求封装成另一个方法 (即另一条测试用例)。
也可以将某个功能点或功能相关联的接口用例封装在同一个测试类中。比如个人中心涉及到的接口，可以封装在同一个测试类中

二、测试函数的封装

一般而言，一个测试函数对应一条用例。上面的代码编写成一条测试用例，示例如下：

强调，pytest中测试函数命名必须以test开头，如test_get_home。

三、测试类/方法的封装

一个测试类相当于一个测试用例集，类中的每个方法对应一条测试用例。以登录接口为例，封装成测试类，示例如下：

强调，pytest中测试类命名需要以Test开头，如TestLogin，且测试类中不能有init方法。测试类中测试方法必须以test开头，如test_login_normal。

四、示例代码

pytest中可以使用命令行或者使用代码方式即 pytest.main() 执行用例，具体可参考文章pytest(1)-简介。

完整的示例代码如下：

import requests
import pytest
import json

def test_get_home():
    '''
    请求首页接口
    :return:
    '''
    headers = {
        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36"
    }

    url = "https://www.cnblogs.com/lfr0123/"
    res = requests.get(url=url, headers=headers)
    # 断言，判断返回结果的code是否等于200，当然实际接口测试中一般返回结果中还会有别的字段需要断言
    assert res.status_code == 200

class TestLogin:
    '''
    登录接口校验
    '''
    url = "http://127.0.0.1:5000/login"
    headers = {"Content-Type": "application/json;charset=utf8"}

    def test_login_normal(self):
        '''正确用户名、正确密码登录'''
        data = {
            "username": "AndyLiu",
            "password": "123456"
        }
        res = requests.post(url=self.url, json=data, headers=self.headers)
        # 断言
        assert res.status_code == 200
        assert json.loads(res.text)["token"]

    def test_login_error(self):
        '''正确用户名、错误密码登录'''
        data = {
            "username": "AndyLiu",
            "password": "111111"
        }
        res = requests.post(url=self.url, json=data, headers=self.headers)
        # 断言
        assert res.status_code == 200
        assert not json.loads(res.text)["token"]

if __name__ == '__main__':
    pytest.main()