如何使用Moto正确Mock类中的AWS SSM/Secrets Manager客户端

以下是针对你遇到问题的具体排查和解决步骤：

核心问题分析 #

你遇到的错误大概率是Lambda初始化阶段的ParamsGetter实例化早于Moto Mock生效，导致代码调用了真实的AWS API而非Mock服务；或者是Mock环境中没有预先创建对应的参数/密钥。

具体解决方案 #

1. 调整ParamsGetter的初始化时机 #

不要在模块级别提前实例化ParamsGetter，改为在Lambda Handler内部延迟初始化，确保Moto Mock已经生效后才创建客户端和获取参数：

# 你的Lambda代码文件，比如my_lambda.py
import boto3

class ParamsGetter:
    def __init__(self):
        self.ssm_client = boto3.client("ssm", region_name="us-east-1")
        self.secrets_client = boto3.client("secretsmanager", region_name="us-east-1")
        self.cache = {}

    def get_ssm_param(self, name):
        if name not in self.cache:
            resp = self.ssm_client.get_parameter(Name=name, WithDecryption=True)
            self.cache[name] = resp["Parameter"]["Value"]
        return self.cache[name]

    def get_secret(self, name):
        if name not in self.cache:
            resp = self.secrets_client.get_secret_value(SecretId=name)
            self.cache[name] = resp["SecretString"]
        return self.cache[name]

# Lambda Handler：在这里才初始化ParamsGetter
def lambda_handler(event, context):
    param_getter = ParamsGetter()
    db_url = param_getter.get_ssm_param("/lambda/db/url")
    db_cred = param_getter.get_secret("lambda/db/credentials")
    return {"status": "ok", "db_url": db_url}

2. 正确配置Moto Mock并预置测试数据 #

在conftest.py中创建作用域合适的fixture，激活Mock并提前创建需要的SSM参数和Secrets Manager密钥：

# conftest.py
import pytest
import boto3
from moto import mock_ssm, mock_secretsmanager

@pytest.fixture(scope="module")
def mock_aws_resources():
    # 同时激活SSM和Secrets Manager的Mock
    with mock_ssm(), mock_secretsmanager():
        # 初始化Mock客户端，预置测试参数
        ssm_client = boto3.client("ssm", region_name="us-east-1")
        ssm_client.put_parameter(
            Name="/lambda/db/url",
            Value="mysql://test:test@localhost:3306/db",
            Type="SecureString"
        )

        # 预置测试密钥
        secrets_client = boto3.client("secretsmanager", region_name="us-east-1")
        secrets_client.create_secret(
            Name="lambda/db/credentials",
            SecretString='{"user":"admin","pass":"test123"}'
        )

        yield  # 执行测试用例

3. 编写测试用例并绑定Mock Fixture #

确保测试用例依赖上述fixture，并且在Mock生效后再导入Lambda模块：

# test_lambda.py
def test_lambda_param_getter(mock_aws_resources):
    # 确保Mock激活后再导入Lambda模块
    from my_lambda import lambda_handler
    result = lambda_handler({}, None)
    
    assert result["db_url"] == "mysql://test:test@localhost:3306/db"
    assert "admin" in result["db_cred"]

特殊场景处理：必须模块级初始化ParamsGetter #

如果业务需求要求在模块级别初始化ParamsGetter，需要在测试时先激活Mock再重新加载模块，确保初始化时使用的是Mock客户端：

def test_module_level_init(mock_aws_resources):
    import importlib
    import my_lambda

    # 重新加载模块，触发模块级的ParamsGetter初始化
    importlib.reload(my_lambda)
    
    result = my_lambda.lambda_handler({}, None)
    assert result["db_url"] == "mysql://test:test@localhost:3306/db"

额外排查点 #

• 检查Moto版本：确保使用最新稳定版，旧版本可能存在SSM/Secrets Manager的Mock兼容问题
• 确认参数名称完全匹配：包括前缀、大小写，Moto的Mock会严格校验参数名称
• 检查boto3客户端的区域：Lambda代码中的客户端区域要和Mock中创建客户端的区域一致

内容的提问来源于stack exchange，提问作者Alyx