def test_load_org():
mock_org = MockOrganization()
mock_org._load_org(SIMPLE_ORG_SPEC)
assert isinstance(mock_org.spec, dict)
assert mock_org.spec == yaml.safe_load(SIMPLE_ORG_SPEC)
assert isinstance(mock_org.org_id, str)
assert isinstance(mock_org.root_id, str)
def test_account_gen():
mock_org = MockOrganization()
mock_org._load_org(SIMPLE_ORG_SPEC)
account = dict(name='mock_account', policies=['p1', 'p2'])
mock_org._account_gen(account, mock_org.root_id)
assert len(mock_org.policy_list) == 2
response = mock_org.client.list_accounts_for_parent(ParentId=mock_org.root_id)
assert [a for a in response['Accounts'] if a['Name'] == account['name']]
def test_ou_gen():
mock_org = MockOrganization()
mock_org._load_org(SIMPLE_ORG_SPEC)
ou = dict(name='mock_ou')
ou_id = mock_org._ou_gen(ou, mock_org.root_id)
assert isinstance(ou_id, str)
response = mock_org.client.describe_organizational_unit(OrganizationalUnitId=ou_id)
assert response['OrganizationalUnit']['Name'] == ou['name']
def test_policy_gen():
mock_org = MockOrganization()
mock_org._load_org(SIMPLE_ORG_SPEC)
policy_name = 'mock_policy'
mock_org._policy_gen(policy_name, mock_org.root_id)
assert len(mock_org.policy_list) == 1
assert mock_org.policy_list[0]['Name'] == policy_name
response = mock_org.client.list_targets_for_policy(
PolicyId=mock_org.policy_list[0]['Id'])
assert response['Targets'][0]['TargetId'] == mock_org.root_id
