moto/tests/test_ssm/test_ssm_utils.py

from moto.ssm.utils import convert_to_tree, convert_to_params

SOURCE_PARAMS = [
    {
        "ARN": (
            "arn:aws:ssm:us-west-1::parameter/aws/service"
            "/global-infrastructure/regions/af-south-1"
        ),
        "DataType": "text",
        "Name": "/aws/service/global-infrastructure/regions/af-south-1",
        "Type": "String",
        "Value": "af-south-1",
        "Version": 1,
    },
    {
        "ARN": (
            "arn:aws:ssm:us-west-1::parameter/aws/service"
            "/global-infrastructure/regions/ap-northeast-2"
        ),
        "DataType": "text",
        "Name": "/aws/service/global-infrastructure/regions/ap-northeast-2",
        "Type": "String",
        "Value": "ap-northeast-2",
        "Version": 1,
    },
    {
        "ARN": (
            "arn:aws:ssm:us-west-1::parameter/aws/service"
            "/global-infrastructure/regions/cn-north-1"
        ),
        "DataType": "text",
        "Name": "/aws/service/global-infrastructure/regions/cn-north-1",
        "Type": "String",
        "Value": "cn-north-1",
        "Version": 1,
    },
    {
        "ARN": (
            "arn:aws:ssm:us-west-1::parameter/aws/service"
            "/global-infrastructure/regions/ap-northeast-2/services"
            "/codestar-notifications"
        ),
        "DataType": "text",
        "Name": (
            "/aws/service/global-infrastructure/regions"
            "/ap-northeast-2/services/codestar-notifications"
        ),
        "Type": "String",
        "Value": "codestar-notifications",
        "Version": 1,
    },
]

EXPECTED_TREE = {
    "aws": {
        "service": {
            "global-infrastructure": {
                "regions": {
                    "af-south-1": {"Value": "af-south-1"},
                    "cn-north-1": {"Value": "cn-north-1"},
                    "ap-northeast-2": {
                        "Value": "ap-northeast-2",
                        "services": {
                            "codestar-notifications": {
                                "Value": "codestar-notifications"
                            }
                        },
                    },
                }
            }
        }
    }
}

CONVERTED_PARAMS = [
    {
        "Name": "/aws/service/global-infrastructure/regions/af-south-1",
        "Value": "af-south-1",
    },
    {
        "Name": "/aws/service/global-infrastructure/regions/cn-north-1",
        "Value": "cn-north-1",
    },
    {
        "Name": "/aws/service/global-infrastructure/regions/ap-northeast-2",
        "Value": "ap-northeast-2",
    },
    {
        "Name": (
            "/aws/service/global-infrastructure/regions/ap-northeast-2"
            "/services/codestar-notifications"
        ),
        "Value": "codestar-notifications",
    },
]


def test_convert_to_tree():
    tree = convert_to_tree(SOURCE_PARAMS)

    assert tree == EXPECTED_TREE


def test_convert_to_params():
    actual = convert_to_params(EXPECTED_TREE)
    assert len(actual) == len(CONVERTED_PARAMS)
    for param in CONVERTED_PARAMS:
        assert param in actual


def test_input_is_correct():
    """Test input should match."""
    for src in SOURCE_PARAMS:
        minimized_src = {"Name": src["Name"], "Value": src["Value"]}
        assert minimized_src in CONVERTED_PARAMS