catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Use centralized ACCOUNT_ID (#4029)

Browse source
This commit is contained in:
Bert Blommers 2021-06-23 18:03:11 +01:00 • committed by GitHub
commit 5e4bccc22d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 30 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/test_ssm/test_ssm_boto3.py
View file

@ -12,6 +12,7 @@ from botocore.exceptions import ClientError
import pytest
from moto import mock_ec2, mock_ssm
from moto.core import ACCOUNT_ID
from moto.ssm.models import PARAMETER_VERSION_LIMIT, PARAMETER_HISTORY_MAX_RESULTS
from tests import EXAMPLE_AMI_ID
@ -120,8 +121,8 @@ def test_get_parameters_by_path():
{p["ARN"] for p in response["Parameters"]}.should.equal(
set(
[
"arn:aws:ssm:us-east-1:1234567890:parameter/foo",
"arn:aws:ssm:us-east-1:1234567890:parameter/baz",
"arn:aws:ssm:us-east-1:{}:parameter/foo".format(ACCOUNT_ID),
"arn:aws:ssm:us-east-1:{}:parameter/baz".format(ACCOUNT_ID),
]
)
)
@ -247,7 +248,7 @@ def test_put_parameter(name):
response["Parameters"][0]["Version"].should.equal(1)
response["Parameters"][0]["LastModifiedDate"].should.be.a(datetime.datetime)
response["Parameters"][0]["ARN"].should.equal(
"arn:aws:ssm:us-east-1:1234567890:parameter/{}".format(name)
"arn:aws:ssm:us-east-1:{}:parameter/{}".format(ACCOUNT_ID, name)
)
initial_modification_date = response["Parameters"][0]["LastModifiedDate"]
@ -274,7 +275,7 @@ def test_put_parameter(name):
initial_modification_date
)
response["Parameters"][0]["ARN"].should.equal(
"arn:aws:ssm:us-east-1:1234567890:parameter/{}".format(name)
"arn:aws:ssm:us-east-1:{}:parameter/{}".format(ACCOUNT_ID, name)
)
response = client.put_parameter(
@ -295,7 +296,7 @@ def test_put_parameter(name):
initial_modification_date
)
response["Parameters"][0]["ARN"].should.equal(
"arn:aws:ssm:us-east-1:1234567890:parameter/{}".format(name)
"arn:aws:ssm:us-east-1:{}:parameter/{}".format(ACCOUNT_ID, name)
)
@ -408,7 +409,7 @@ def test_get_parameter():
response["Parameter"]["Type"].should.equal("String")
response["Parameter"]["LastModifiedDate"].should.be.a(datetime.datetime)
response["Parameter"]["ARN"].should.equal(
"arn:aws:ssm:us-east-1:1234567890:parameter/test"
"arn:aws:ssm:us-east-1:{}:parameter/test".format(ACCOUNT_ID)
)
@ -434,7 +435,7 @@ def test_get_parameter_with_version_and_labels():
response["Parameter"]["Type"].should.equal("String")
response["Parameter"]["LastModifiedDate"].should.be.a(datetime.datetime)
response["Parameter"]["ARN"].should.equal(
"arn:aws:ssm:us-east-1:1234567890:parameter/test-1"
"arn:aws:ssm:us-east-1:{}:parameter/test-1".format(ACCOUNT_ID)
)
response = client.get_parameter(Name="test-2:1", WithDecryption=False)
@ -443,7 +444,7 @@ def test_get_parameter_with_version_and_labels():
response["Parameter"]["Type"].should.equal("String")
response["Parameter"]["LastModifiedDate"].should.be.a(datetime.datetime)
response["Parameter"]["ARN"].should.equal(
"arn:aws:ssm:us-east-1:1234567890:parameter/test-2"
"arn:aws:ssm:us-east-1:{}:parameter/test-2".format(ACCOUNT_ID)
)
response = client.get_parameter(Name="test-2:test-label", WithDecryption=False)
@ -452,7 +453,7 @@ def test_get_parameter_with_version_and_labels():
response["Parameter"]["Type"].should.equal("String")
response["Parameter"]["LastModifiedDate"].should.be.a(datetime.datetime)
response["Parameter"]["ARN"].should.equal(
"arn:aws:ssm:us-east-1:1234567890:parameter/test-2"
"arn:aws:ssm:us-east-1:{}:parameter/test-2".format(ACCOUNT_ID)
)
with pytest.raises(ClientError) as ex: