137 lines
4.5 KiB
Python
137 lines
4.5 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import six
|
|
import sure # noqa
|
|
import datetime
|
|
from moto.organizations import utils
|
|
|
|
EMAIL_REGEX = "^.+@[a-zA-Z0-9-.]+.[a-zA-Z]{2,3}|[0-9]{1,3}$"
|
|
ORG_ID_REGEX = r'o-[a-z0-9]{%s}' % utils.ORG_ID_SIZE
|
|
ROOT_ID_REGEX = r'r-[a-z0-9]{%s}' % utils.ROOT_ID_SIZE
|
|
OU_ID_REGEX = r'ou-[a-z0-9]{%s}-[a-z0-9]{%s}' % (utils.ROOT_ID_SIZE, utils.OU_ID_SUFFIX_SIZE)
|
|
ACCOUNT_ID_REGEX = r'[0-9]{%s}' % utils.ACCOUNT_ID_SIZE
|
|
CREATE_ACCOUNT_STATUS_ID_REGEX = r'car-[a-z0-9]{%s}' % utils.CREATE_ACCOUNT_STATUS_ID_SIZE
|
|
|
|
|
|
def test_make_random_org_id():
|
|
org_id = utils.make_random_org_id()
|
|
org_id.should.match(ORG_ID_REGEX)
|
|
|
|
|
|
def test_make_random_root_id():
|
|
root_id = utils.make_random_root_id()
|
|
root_id.should.match(ROOT_ID_REGEX)
|
|
|
|
|
|
def test_make_random_ou_id():
|
|
root_id = utils.make_random_root_id()
|
|
ou_id = utils.make_random_ou_id(root_id)
|
|
ou_id.should.match(OU_ID_REGEX)
|
|
|
|
|
|
def test_make_random_account_id():
|
|
account_id = utils.make_random_account_id()
|
|
account_id.should.match(ACCOUNT_ID_REGEX)
|
|
|
|
|
|
def test_make_random_create_account_status_id():
|
|
create_account_status_id = utils.make_random_create_account_status_id()
|
|
create_account_status_id.should.match(CREATE_ACCOUNT_STATUS_ID_REGEX)
|
|
|
|
|
|
def validate_organization(response):
|
|
org = response['Organization']
|
|
sorted(org.keys()).should.equal([
|
|
'Arn',
|
|
'AvailablePolicyTypes',
|
|
'FeatureSet',
|
|
'Id',
|
|
'MasterAccountArn',
|
|
'MasterAccountEmail',
|
|
'MasterAccountId',
|
|
])
|
|
org['Id'].should.match(ORG_ID_REGEX)
|
|
org['MasterAccountId'].should.equal(utils.MASTER_ACCOUNT_ID)
|
|
org['MasterAccountArn'].should.equal(utils.MASTER_ACCOUNT_ARN_FORMAT.format(
|
|
org['MasterAccountId'],
|
|
org['Id'],
|
|
))
|
|
org['Arn'].should.equal(utils.ORGANIZATION_ARN_FORMAT.format(
|
|
org['MasterAccountId'],
|
|
org['Id'],
|
|
))
|
|
org['MasterAccountEmail'].should.equal(utils.MASTER_ACCOUNT_EMAIL)
|
|
org['FeatureSet'].should.be.within(['ALL', 'CONSOLIDATED_BILLING'])
|
|
org['AvailablePolicyTypes'].should.equal([{
|
|
'Type': 'SERVICE_CONTROL_POLICY',
|
|
'Status': 'ENABLED'
|
|
}])
|
|
|
|
|
|
def validate_roots(org, response):
|
|
response.should.have.key('Roots').should.be.a(list)
|
|
response['Roots'].should_not.be.empty
|
|
root = response['Roots'][0]
|
|
root.should.have.key('Id').should.match(ROOT_ID_REGEX)
|
|
root.should.have.key('Arn').should.equal(utils.ROOT_ARN_FORMAT.format(
|
|
org['MasterAccountId'],
|
|
org['Id'],
|
|
root['Id'],
|
|
))
|
|
root.should.have.key('Name').should.be.a(six.string_types)
|
|
root.should.have.key('PolicyTypes').should.be.a(list)
|
|
root['PolicyTypes'][0].should.have.key('Type').should.equal('SERVICE_CONTROL_POLICY')
|
|
root['PolicyTypes'][0].should.have.key('Status').should.equal('ENABLED')
|
|
|
|
|
|
def validate_organizational_unit(org, response):
|
|
response.should.have.key('OrganizationalUnit').should.be.a(dict)
|
|
ou = response['OrganizationalUnit']
|
|
ou.should.have.key('Id').should.match(OU_ID_REGEX)
|
|
ou.should.have.key('Arn').should.equal(utils.OU_ARN_FORMAT.format(
|
|
org['MasterAccountId'],
|
|
org['Id'],
|
|
ou['Id'],
|
|
))
|
|
ou.should.have.key('Name').should.be.a(six.string_types)
|
|
|
|
|
|
def validate_account(org, account):
|
|
sorted(account.keys()).should.equal([
|
|
'Arn',
|
|
'Email',
|
|
'Id',
|
|
'JoinedMethod',
|
|
'JoinedTimestamp',
|
|
'Name',
|
|
'Status',
|
|
])
|
|
account['Id'].should.match(ACCOUNT_ID_REGEX)
|
|
account['Arn'].should.equal(utils.ACCOUNT_ARN_FORMAT.format(
|
|
org['MasterAccountId'],
|
|
org['Id'],
|
|
account['Id'],
|
|
))
|
|
account['Email'].should.match(EMAIL_REGEX)
|
|
account['JoinedMethod'].should.be.within(['INVITED', 'CREATED'])
|
|
account['Status'].should.be.within(['ACTIVE', 'SUSPENDED'])
|
|
account['Name'].should.be.a(six.string_types)
|
|
account['JoinedTimestamp'].should.be.a(datetime.datetime)
|
|
|
|
|
|
def validate_create_account_status(create_status):
|
|
sorted(create_status.keys()).should.equal([
|
|
'AccountId',
|
|
'AccountName',
|
|
'CompletedTimestamp',
|
|
'Id',
|
|
'RequestedTimestamp',
|
|
'State',
|
|
])
|
|
create_status['Id'].should.match(CREATE_ACCOUNT_STATUS_ID_REGEX)
|
|
create_status['AccountId'].should.match(ACCOUNT_ID_REGEX)
|
|
create_status['AccountName'].should.be.a(six.string_types)
|
|
create_status['State'].should.equal('SUCCEEDED')
|
|
create_status['RequestedTimestamp'].should.be.a(datetime.datetime)
|
|
create_status['CompletedTimestamp'].should.be.a(datetime.datetime)
|