Python中10个常用的接口自动化装饰器

2024软件测试面试刷题，这个小程序（永久刷题），靠它快速找到工作了！（刷题APP的天花板）-CSDN博客文章浏览阅读2.2k次，点赞85次，收藏11次。你知不知道有这么一个软件测试面试的刷题小程序。里面包含了面试常问的软件测试基础题，web自动化测试、app自动化测试、接口测试、性能测试、自动化测试、安全测试及一些常问到的人力资源题目。最主要的是他还收集了像阿里、华为这样的大厂面试真题，还有互动交流板块……https://blog.csdn.net/AI_Green/article/details/134931243?spm=1001.2014.3001.5502更多Python学习内容：ipengtao.com

装饰器（Decorators）是Python中一种强大的编程工具，它们用于修改或增强函数或方法的行为。在接口自动化测试中，装饰器可以起到简化代码、提高代码可维护性和可重用性的作用。本文将介绍Python中的10个常用接口自动化装饰器，并提供详细的示例代码。

1. 日志记录装饰器

日志记录装饰器用于记录函数或方法的调用信息，包括输入参数和返回结果。这对于调试和排查问题非常有帮助。

def log_decorator(func):
    def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        print(f"调用 {func.__name__}，输入参数：{args}, 返回结果：{result}")
        return result
    return wrapper

使用示例：

@log_decorator
def add(a, b):
    return a + b

result = add(1, 2)

2. 超时装饰器

超时装饰器用于设置函数或方法的最大执行时间，防止长时间运行的接口自动化测试导致阻塞。

import timeout_decorator

@timeout_decorator.timeout(5)  # 设置最大执行时间为5秒
def test_interface():
    # 执行接口测试的代码

3. 重试装饰器

重试装饰器用于在函数或方法失败时自动重试，以增加接口自动化测试的稳定性。

import retrying

@retrying.retry(stop_max_attempt_number=3)  # 最多重试3次
def test_interface():
    # 执行接口测试的代码

4. 数据驱动装饰器

数据驱动装饰器用于通过不同的输入数据多次运行同一个测试用例，以增加测试覆盖率。

def data_driven(data_list):
    def decorator(func):
        def wrapper(*args, **kwargs):
            for data in data_list:
                result = func(*args, **kwargs, data)
                print(f"输入数据：{data}, 返回结果：{result}")
        return wrapper
    return decorator

data_list = [1, 2, 3, 4, 5]

@data_driven(data_list)
def test_interface(data):
    # 执行接口测试的代码，使用输入数据data

5. 缓存装饰器

缓存装饰器用于缓存函数或方法的结果，以减少重复计算，提高性能。

import functools

@functools.lru_cache(maxsize=None)  # 设置缓存大小为无限
def get_data_from_db(query):
    # 从数据库中获取数据的代码

6. 参数检查装饰器

参数检查装饰器用于检查函数或方法的输入参数是否符合预期，以提前发现错误。

def check_input_args(args_check):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if args_check(*args, **kwargs):
                result = func(*args, **kwargs)
                return result
            else:
                raise ValueError("输入参数错误")
        return wrapper
    return decorator

def args_check(a, b):
    return isinstance(a, int) and isinstance(b, int)

@check_input_args(args_check)
def add(a, b):
    return a + b

7. 认证装饰器

认证装饰器用于在执行接口自动化测试前进行身份认证，以确保测试环境的安全性。

def authenticate(username, password):
    def decorator(func):
        def wrapper(*args, **kwargs):
            if check_credentials(username, password):
                result = func(*args, **kwargs)
                return result
            else:
                raise PermissionError("身份认证失败")
        return wrapper
    return decorator

def check_credentials(username, password):
    # 校验用户名和密码的代码

@authenticate("user", "password")
def test_interface():
    # 执行接口测试的代码

8. 计时装饰器

计时装饰器用于记录函数或方法的执行时间，以评估性能和优化代码。

import time

def timing_decorator(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"{func.__name__} 执行时间：{end_time - start_time}秒")
        return result
    return wrapper

使用示例：

@timing_decorator
def test_interface():
    # 执行接口测试的代码

9. 接口模拟装饰器

接口模拟装饰器用于模拟接口的返回数据，以便在开发和测试阶段使用。

def mock_interface(response_data):
    def decorator(func):
        def wrapper(*args, **kwargs):
            return response_data
        return wrapper
    return decorator

@mock_interface({"status": "success", "data": "mocked_data"})
def test_interface():
    # 执行接口测试的代码，将返回模拟的数据

10. 日志文件装饰器

日志文件装饰器用于将函数或方法的执行日志记录到文件中，以便后续分析和查看。

def log_to_file(log_file):
    def decorator(func):
        def wrapper(*args, **kwargs):
            result = func(*args, **kwargs)
            with open(log_file, "a") as file:
                file.write(f"{func.__name__} 执行结果：{result}\n")
            return result
        return wrapper
    return decorator

使用示例：

@log_to_file("test.log")
def test_interface():
    # 执行接口测试的代码，将执行结果记录到test.log文件中

总结

以上是Python中的10个常用接口自动化装饰器，它们可以简化接口测试代码、提高代码的可维护性和可重用性，并增加测试的稳定性和覆盖率。根据具体的需求和场景，可以选择合适的装饰器来优化接口自动化测试代码。

如果你觉得文章还不错，请大家 点赞、分享、留言 下，因为这将是我持续输出更多优质文章的最强动力！

行动吧，在路上总比一直观望的要好，未来的你肯定会感谢现在拼搏的自己！如果想学习提升找不到资料，没人答疑解惑时，请及时加入群： 786229024，里面有各种测试开发资料和技术可以一起交流哦。

最后： 下方这份完整的软件测试视频教程已经整理上传完成，需要的朋友们可以自行领取【保证100%免费】
软件测试面试文档
我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。