接口自动化框架篇：自定义断言封装逻辑！_自定义断言要不要学-CSDN博客

在接口自动化测试中，断言是非常重要的一环，用于验证接口返回结果是否符合预期。在编写自动化测试脚本时，通常会使用一些断言库

来进行断言操作，比如assert库、unittest库等。但是，有些情况下这些库提供的断言方法可能不够灵活，无法满足我们的需求，这时就需要我们自定义断言封装逻辑。

下面就来详细介绍一下如何从零开始规范地封装自定义断言逻辑。

首先，我们需要创建一个新的断言库，可以将其命名为custom_assert。在这个库中，我们可以定义一些常用的断言方法，以提供给自动化测试脚本使用。

1. 创建custom_assert.py文件，并打开编辑。

2. 在custom_assert.py文件中，导入所需的模块或库，比如json、re等。

3. 在文件中定义自定义断言方法

。这些方法需要满足以下几个条件：

- 方法名要以assert

开头，以明确表示这是一个断言方法。
- 方法的参数需要包括被断言的实际值和预期值。
- 方法的返回值可以是布尔类型

，用于表示断言结果的成功或失败。

例如，我们可以定义一个断言方法来验证接口返回的状态码

是否是200：

def assert_status_code(actual, expected=200):
    """
    验证接口返回的状态码是否是200
    :param actual: 实际状态码
    :param expected: 预期状态码，默认为200
    :return: 断言结果
    """
    return actual == expected

再例如，我们可以定义一个断言方法来验证接口返回的json数据中是否包含指定字段：

def assert_json_contains(actual, field):
    """
    验证接口返回的json数据中是否包含指定字段
    :param actual: 实际返回的json数据
    :param field: 指定字段
    :return: 断言结果
    """
    return field in actual

4. 编写完自定义断言方法后，可以进行一些测试来验证方法的准确性和可用性

。可以使用unittest库中的TestCase类来编写测试用例，并在测试用例中使用自定义断言方法进行断言操作。

import unittest

class CustomAssertTest(unittest.TestCase):
    def test_assert_status_code(self):
        actual = 200
        expected = 200
        self.assertTrue(assert_status_code(actual, expected))

    def test_assert_json_contains(self):
        actual = {"name": "John"}
        field = "name"
        self.assertTrue(assert_json_contains(actual, field))

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

运行测试用例，如果所有测试用例都通过，说明自定义断言方法的逻辑是正确的。

5. 将custom_assert.py文件上传到测试项目的代码仓库中，并在自动化测试脚本中导入custom_assert库，即可使用自定义断言方法。

import custom_assert

def test_api():
    response = requests.get("http://example.com/api")
    assert custom_assert.assert_status_code(response.status_code)
    assert custom_assert.assert_json_contains(response.json(), "name")

if __name__ == "__main__":
    test_api()

这样，我们就完成了自定义断言的封装逻辑。通过将常用的断言方法封装为自定义断言库，可以提高测试脚本的可读性和可维护性，并使断言操作更加灵活和符合需求。同时，自定义断言方法也可以帮助我们快速定位测试用例失败的原因，提高排查问题的效率。

希望这篇文章能够帮助你理解和规范自定义断言封装逻辑的过程，让你的接口自动化测试更加高效和准确！

最后感谢每一个认真阅读我文章的人，看着粉丝一路的上涨和关注，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走！