moto/tests/test_wafv2/test_utils.py

20 lines
614 B
Python

import uuid
from moto.wafv2.utils import make_arn_for_wacl
from moto.core import DEFAULT_ACCOUNT_ID as ACCOUNT_ID
def test_make_arn_for_wacl():
uniqueID = str(uuid.uuid4())
region = "us-east-1"
name = "testName"
scope = "REGIONAL"
arn = make_arn_for_wacl(name, ACCOUNT_ID, region, uniqueID, scope)
assert (
arn == f"arn:aws:wafv2:{region}:{ACCOUNT_ID}:regional/webacl/{name}/{uniqueID}"
)
scope = "CLOUDFRONT"
arn = make_arn_for_wacl(name, ACCOUNT_ID, region, uniqueID, scope)
assert arn == f"arn:aws:wafv2:{region}:{ACCOUNT_ID}:global/webacl/{name}/{uniqueID}"