pytest并发执行时token异常处理问题

接前面加入钩子函数处理token复用的问题,只保证了用例的串联执行,我的部分测试用例中接入了通义千问的部分接口生成测试数据,七八个场景跑完差不多快要10分钟。考虑使用并发执行。

http://t.csdnimg.cn/ACexL

使用多线程和不使用耗时差距很大,建议还是开启

前置条件:

下载pytest-xdist 这个包 ,直接在终端pip install  pytest-xdist即可

更改pytest.ini 命令行配置 

更改token处理方式,加入锁机制

代码是由之前的代码使用gpt生成的 ,针对token.json文件的锁好像无用,还是会导致登录频繁报错。这里改为对username的锁,保证同一时间同一账号只登录一次。另外,避免出现其他不稳定的情况,建议加上重试 --reruns 2。

# 全局文件锁
token_file_lock = threading.Lock()

# 用户级别的锁字典,用于控制对每个用户名的并发访问
user_token_locks = {}

def load_tokens_from_file(file_path):
    with token_file_lock:
        with open(file_path, 'r') as file:
            return json.load(file)

def save_tokens_to_file(tokens, file_path):
    with token_file_lock:
        with open(file_path, 'w') as file:
            json.dump(tokens, file)

def get_user_lock(username):
    """
    获取或创建针对特定username的锁。
    """
    if username not in user_token_locks:
        user_token_locks[username] = threading.Lock()
    return user_token_locks[username]

def synchronized_per_user(func):
    """装饰器,用于同步函数,确保对每个username的操作是线程安全的"""
    @wraps(func)
    def wrapper(username, *args, **kwargs):
        user_lock = get_user_lock(username)
        with user_lock:
            return func(username, *args, **kwargs)
    return wrapper

@pytest.fixture(scope="module")
def auth_tokens(refresh_threshold=600, file_path='tokens.json'):
    tokens = load_tokens_from_file(file_path)

    @synchronized_per_user
    def get_or_refresh_token(username):
        nonlocal tokens
        token_info = tokens.get(username)
        if not token_info or (time.time() - token_info['timestamp']) > refresh_threshold:
            logger.info(f"没有获取到该账号{username}的token,或者token已过期,重新获取token中  当前时间为{time.time()}  时间间隔为{time.time() - token_info['timestamp']}")
            # 假设login_user是一个模拟的函数,实际应用中需要替换为真实的认证逻辑
            token = login_user(username, '222222').response.json()['data']['accessToken']
            tokens[username] = {'token': token, 'timestamp': time.time()}
            save_tokens_to_file(tokens, file_path)
        logger.info(f"{username}的token为{tokens[username]['token']}")
        return tokens[username]['token']

    return get_or_refresh_token

这里还有个疑问,加上并发执行后,控制台打印的中文日志全变成了乱码,不明所以,后续有时间再看看怎么解决

相关推荐

  1. 代码扫描常见问题盘点-并发处理类/异常

    2024-06-14 17:02:06       20 阅读
  2. pytest中fixture异常处理

    2024-06-14 17:02:06       33 阅读
  3. pytest并发执行用例方案

    2024-06-14 17:02:06       35 阅读
  4. pytesttoken的一种处理方法

    2024-06-14 17:02:06       30 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-14 17:02:06       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-14 17:02:06       106 阅读
  3. 在Django里面运行非项目文件

    2024-06-14 17:02:06       87 阅读
  4. Python语言-面向对象

    2024-06-14 17:02:06       96 阅读

热门阅读

  1. mysql定时任务事件清理单表数据

    2024-06-14 17:02:06       26 阅读
  2. 修改容器内部时间

    2024-06-14 17:02:06       26 阅读
  3. 死锁的产生预防处理解锁

    2024-06-14 17:02:06       26 阅读
  4. 电压kV为什么k要小写,原因你知道吗?

    2024-06-14 17:02:06       31 阅读
  5. 男人圣经 14

    2024-06-14 17:02:06       25 阅读
  6. AI 到底是什么?

    2024-06-14 17:02:06       30 阅读
  7. ApplicationContextAware使用【工具类】

    2024-06-14 17:02:06       37 阅读
  8. TF-IDF(Term Frequency-Inverse Document Frequency)算法

    2024-06-14 17:02:06       26 阅读
  9. 详解 Flink Table API 和 Flink SQL 之函数

    2024-06-14 17:02:06       23 阅读
  10. 测评要求+基本措施+对应产品

    2024-06-14 17:02:06       29 阅读
  11. 保险丝的选型

    2024-06-14 17:02:06       29 阅读
  12. Django的‘通用视图TemplateView’

    2024-06-14 17:02:06       26 阅读
  13. RSA学习

    RSA学习

    2024-06-14 17:02:06      22 阅读