/tests/tpath/idem_aws/tool/org_test_util.py

https://gitlab.com/pgeorgiev_vmw/idem-aws · Python · 37 lines · 32 code · 5 blank · 0 comment · 0 complexity · 762b06cdcf7e0539d57d2745f6cb4ccc MD5 · raw file 

    

  1. import re
    2. 

  2. 

  3. ORGANIZATION_ARN_FORMAT = "arn:aws:organizations::{0}:organization/{1}"
    4. 

  4. MASTER_ACCOUNT_ARN_FORMAT = "arn:aws:organizations::{0}:account/{1}/{0}"
    5. 

  5. ACCOUNT_ARN_FORMAT = "arn:aws:organizations::{0}:account/{1}/{2}"
    6. 

  6. 

  7. ORG_ID_SIZE = 10
    8. 

  8. ORG_ID_REGEX = r"o-[a-z0-9]{%s}" % ORG_ID_SIZE
    9. 

  9. 

  10. 

  11. def validate_organization(hub, response):
    12. 

  12.     org = response
    13. 

  13.     assert sorted(org.keys()) == (
    14. 

  14.         [
    15. 

  15.             "arn",
    16. 

  16.             "available_policy_types",
    17. 

  17.             "feature_set",
    18. 

  18.             "master_account_arn",
    19. 

  19.             "master_account_email",
    20. 

  20.             "master_account_id",
    21. 

  21.             "name",
    22. 

  22.             "resource_id",
    23. 

  23.             "roots",
    24. 

  24.         ]
    25. 

  25.     )
    26. 

  26. 

  27.     assert re.compile(ORG_ID_REGEX).match(org["resource_id"])
    28. 

  28.     assert org["master_account_arn"] == MASTER_ACCOUNT_ARN_FORMAT.format(
    29. 

  29.         org["master_account_id"], org["resource_id"]
    30. 

  30.     )
    31. 

  31.     assert org["arn"] == ORGANIZATION_ARN_FORMAT.format(
    32. 

  32.         org["master_account_id"], org["resource_id"]
    33. 

  33.     )
    34. 

  34.     assert org["feature_set"] in ["ALL", "CONSOLIDATED_BILLING"]
    35. 

  35.     assert org["available_policy_types"] == [
    36. 

  36.         {"Type": "SERVICE_CONTROL_POLICY", "Status": "ENABLED"}
    37. 

  37.     ]
    38.