Python 模拟 AWS SSM

Question

我已经编写了一个代码来为我获取 SSM 参数

import boto3
    
client = boto3.client('ssm')
    
def lambda_handler(event, context):
    return client.get_parameter(Name=event["param"], WithDecryption=True)

if __name__ == '__main__':
    print(lambda_handler({"param": "/mypath/password"}, ""))

但是，我无法为它编写测试用例，我曾尝试使用 moto但出于某种原因，它仍然给了我 SSM 商店的实际值(value)

import os

import boto3
from moto import mock_ssm
import pytest

from handler import lambda_handler

@pytest.fixture
def aws_credentials():
    os.environ["AWS_ACCESS_KEY_ID"] = "testing"
    os.environ["AWS_SECRET_ACCESS_KEY"] = "testing"
    os.environ["AWS_SECURITY_TOKEN"] = "testing"
    os.environ["AWS_SESSION_TOKEN"] = "testing"

@mock_ssm
def test_ssm():
    ssm = boto3.client('ssm')
    ssm.put_parameter(
        Name="/mypath/password",
        Description="A test parameter",
        Value="this is it!",
        Type="SecureString"
    )
    resp = lambda_handler({"param": "/mypath/password"}, "")
    assert resp["Parameter"]["Value"] == "this is it!"

我是否遗漏了一些无意中听到的东西，我应该怎么做才能使它工作，或者是否有另一种方法可以在 python 中模拟 SSM。

Answer 1

当您打补丁时 ssm来自 @mock_ssm ，您已经实例化了变量 handler.client在 handler.py 中作为 boto3 的真实实例客户端，因此它不是修补版本。
解决方案1:
仅在 lambda 处理程序期间初始化客户端，以便补丁在创建时已经生效。
handler.py

import boto3

# client = boto3.client('ssm')  # Remove this instantiation

def lambda_handler(event, context):
    client = boto3.client('ssm')  # Move it here
    return client.get_parameter(Name=event["param"], WithDecryption=True)

if __name__ == '__main__':
    print(lambda_handler({"param": "/mypath/password"}, ""))

解决方案2:
仅在补丁生效后才导入/执行 handler.py 文件。
test_handler.py

...
# from handler import lambda_handler  # Remove this import
...
@mock_ssm
def test_ssm():
    from handler import lambda_handler  # Move it here
    ...
...

解决方案3
补丁生效后重新加载 handler.py。

...
@mock_ssm
def test_ssm():
    from importlib import reload
    import sys
    reload(sys.modules['handler'])
    ...
...

输出

$ pytest -q -rP
============================================================================================ PASSES ============================================================================================
1 passed in 1.35s