这篇文章主要介绍了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()

总结

• 测试函数、测试类/测试方法的封装，其实不管是什么单元测试框架，遵循的方式都一样。

• 而在命名方式上各有自己的要求，比如pytest与unittest中测试命名方法有一定的区别。

• 把一个有自己断言的函数或方法看成是一条测试用例，那么测试类其实就是一个含有一条或者多条测试用例的测试用例集，类中的每个方法对应一条测试用例。

• 一个测试类中放置哪些测试方法，换句话说一个测试用例集中应该包含哪些测试用例，这个可以按照项目自身情况而定，也可按照测试人员自己的想法而定，主旨就是要清晰明了