Function test_create

python/example_code/ssm/test/test_document.py:16–65

(make_stubber, capsys, error_code)

@pytest.mark.parametrize("error_code", [None, "TestException", "DocumentAlreadyExists"])
def test_create(make_stubber, capsys, error_code):
    ssm_client = boto3.client("ssm")
    ssm_stubber = make_stubber(ssm_client)
    document_wrapper = DocumentWrapper(ssm_client)
    name = "python-test"
    content = """
{
    "schemaVersion": "2.2",
    "description": "Run a simple shell command",
    "mainSteps": [
        {
            "action": "aws:runShellScript",
            "name": "runEchoCommand",
            "inputs": {
              "runCommand": [
                "echo 'Hello, world!'"
              ]
            }
        }
    ]
}
            """

    ssm_stubber.stub_create_document(
        content,
        name,
        error_code=error_code,
    )

    if error_code is None:
        document_wrapper.create(
            content,
            name,
        )
        assert document_wrapper.name == name
    elif error_code == "DocumentAlreadyExists":
        document_wrapper.create(
            content,
            name,
        )
        assert document_wrapper.name == name
        capt = capsys.readouterr()
        assert "already exists" in capt.out
    else:
        with pytest.raises(ClientError) as exc_info:
            document_wrapper.create(
                content,
                name,
            )
        assert exc_info.value.response["Error"]["Code"] == error_code


@pytest.mark.parametrize("error_code", [None, "TestException"])