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