def test_c4_orchestration_manager_find_stack():
mocked_boto3 = MockBoto3()
mock_stack_names = [
'foo-bar-baz', 'awseb-e-foo-stack', 'c4-network-cgap-test-123',
'c4-iam-cgap-test-456'
]
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mock_stack_names=mock_stack_names)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
manager = cloudformation_utils.C4OrchestrationManager()
network_stack = manager.find_stack('network')
assert network_stack.name == 'c4-network-cgap-test-123'
mocked_boto3 = MockBoto3()
mock_stack_names = [
'foo-bar-baz', 'awseb-e-foo-stack', 'c4-network-cgap-test-123',
'c4-network-cgap-prod-456'
]
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mock_stack_names=mock_stack_names)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
manager = cloudformation_utils.C4OrchestrationManager()
with pytest.raises(ValueError, match="too many"):
manager.find_stack('network')
def test_get_ecs_real_url():
mocked_boto3 = MockBoto3()
mastertest_url = 'http://C4EcsTrialAlphacgapmastertest-1234512345.us-east-1.elb.amazonaws.com'
mastertest_outputs = [
{
'OutputKey': 'ECSApplicationURLcgapmastertest',
'OutputValue': mastertest_url,
'Description': 'URL of CGAP-Portal.',
},
]
supertest_url = 'http://cgap-supertest-11223344.us-east-1.elb.amazonaws.com'
supertest_outputs = [{
'OutputKey': 'ECSApplicationURLcgapsupertest',
'OutputValue': supertest_url,
'Description': 'URL of CGAP-Portal.',
}]
mocked_stacks = [
MockBotoCloudFormationStack('c4-iam-main-stack'),
MockBotoCloudFormationStack('c4-network-main-stack'),
MockBotoCloudFormationStack('c4-ecs-cgap-supertest-stack',
mock_outputs=supertest_outputs),
MockBotoCloudFormationStack('c4-ecs-cgap-mastertest-stack',
mock_outputs=mastertest_outputs),
]
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mocked_stacks=mocked_stacks)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
assert cloudformation_utils.get_ecs_real_url(
'cgap-supertest') == supertest_url
assert cloudformation_utils.get_ecs_real_url(
'cgap-mastertest') == mastertest_url
def test_c4_orchestration_manager_find_stack_resource():
mocked_boto3 = MockBoto3()
private_subnet_a = MockBotoCloudFormationResourceSummary(
logical_id='MyPrivateSubnetA', physical_resource_id='subnet-111111')
private_subnet_b = MockBotoCloudFormationResourceSummary(
logical_id='MyPrivateSubnetB', physical_resource_id='subnet-222222')
public_subnet_a = MockBotoCloudFormationResourceSummary(
logical_id='MyPublicSubnetA', physical_resource_id='subnet-333333')
public_subnet_b = MockBotoCloudFormationResourceSummary(
logical_id='MyPublicSubnetA', physical_resource_id='subnet-444444')
network_resource_summaries = [
private_subnet_a, private_subnet_b, public_subnet_a, public_subnet_b
]
mocked_stacks = [
MockBotoCloudFormationStack(
'c4-network-main-stack',
mock_resource_summaries=network_resource_summaries),
MockBotoCloudFormationStack('c4-iam-main-stack')
]
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mocked_stacks=mocked_stacks)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
manager = cloudformation_utils.C4OrchestrationManager()
assert manager.find_stack_resource(
'network', 'MyPrivateSubnetA') == private_subnet_a
assert manager.find_stack_resource(
'network', 'MyPrivateSubnetB') == private_subnet_b
assert manager.find_stack_resource('network',
'MyPrivateSubnetC') is None
assert manager.find_stack_resource('network', 'MyPrivateSubnetA',
'logical_id') == 'MyPrivateSubnetA'
assert manager.find_stack_resource('network', 'MyPrivateSubnetC',
'logical_id') is None
assert manager.find_stack_resource(
'network', 'MyPrivateSubnetA',
'physical_resource_id') == 'subnet-111111'
assert manager.find_stack_resource('network', 'MyPrivateSubnetC',
'physical_resource_id') is None
assert manager.find_stack_resource('network', 'MyPrivateSubnetC',
'physical_resource_id',
'foo') == 'foo'
def test_c4_orchestration_manager_find_stack_outputs():
mocked_boto3 = MockBoto3()
network_outputs = [
{
'OutputKey': 'rds',
'OutputValue': 'some-db-thing'
},
{
'OutputKey': 'es',
'OutputValue': 'some-es-thing'
},
]
iam_outputs = [
{
'OutputKey': 'user1',
'OutputValue': 'Joe'
},
{
'OutputKey': 'user2',
'OutputValue': 'Sally'
},
]
mocked_stacks = [
MockBotoCloudFormationStack('c4-network-main-stack',
mock_outputs=network_outputs),
MockBotoCloudFormationStack('c4-iam-main-stack',
mock_outputs=iam_outputs),
]
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mocked_stacks=mocked_stacks)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
manager = cloudformation_utils.C4OrchestrationManager()
# Looking up a simple key by name
assert manager.find_stack_outputs('user1') == {'user1': 'Joe'}
assert manager.find_stack_outputs('user1', value_only=True) == ['Joe']
assert manager.find_stack_outputs('rds') == {'rds': 'some-db-thing'}
assert manager.find_stack_outputs(
'rds', value_only=True) == ['some-db-thing']
assert manager.find_stack_outputs('not-there') == {}
assert manager.find_stack_outputs('not-there', value_only=True) == []
# Use of predicate to find several related keys
assert manager.find_stack_outputs(lambda x: x.startswith("user")) == {
'user1': 'Joe',
'user2': 'Sally'
}
assert sorted(
manager.find_stack_outputs(lambda x: x.startswith("user"),
value_only=True)) == ['Joe', 'Sally']
def test_get_ecr_repo_url_hybrid(use_ecosystem_repo):
main_repo = 'xxx6742.dkr.ecr.us-east-1.amazonaws.com/main'
main_stack = 'c4-ecr-main-stack'
main_outputs = [
{
'OutputKey': 'C4ECRMainRepoURL',
'OutputValue': main_repo,
'Description': 'CGAPDocker Image Repository URL',
'ExportName': f'{main_stack}-RepoURL'
},
]
mocked_boto3 = MockBoto3()
mastertest_repo = 'xxx6742.dkr.ecr.us-east-1.amazonaws.com/cgap-mastertest'
mastertest_stack = 'c4-ecr-cgap-mastertest-stack'
mastertest_outputs = [
{
'OutputKey': 'C4EcrCgapMastertestRepoURL',
'OutputValue': mastertest_repo,
'Description': 'CGAPDocker Image Repository URL',
'ExportName': f'{mastertest_stack}-RepoURL'
},
]
hotseat_repo = 'xxx6742.dkr.ecr.us-east-1.amazonaws.com/cgap-hotseat'
hotseat_stack = 'c4-ecr-cgap-hotseat-stack'
hotseat_outputs = [
{
'OutputKey': 'C4EcrCgapHotseatRepoURL',
'OutputValue': hotseat_repo,
'Description': 'CGAPDocker Image Repository URL',
'ExportName': f'{hotseat_stack}-RepoURL'
},
]
mocked_stacks = [
MockBotoCloudFormationStack('c4-iam-trial-alpha-stack'),
MockBotoCloudFormationStack('c4-network-trial-alpha-stack'),
MockBotoCloudFormationStack(mastertest_stack,
mock_outputs=mastertest_outputs),
MockBotoCloudFormationStack(hotseat_stack,
mock_outputs=hotseat_outputs),
]
if use_ecosystem_repo:
mocked_stacks.append(
MockBotoCloudFormationStack(main_stack, mock_outputs=main_outputs))
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mocked_stacks=mocked_stacks)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
assert cloudformation_utils.get_ecr_repo_url(
'cgap-mastertest') == mastertest_repo
assert cloudformation_utils.get_ecr_repo_url(
env_name='cgap-hotseat') == hotseat_repo
assert cloudformation_utils.get_ecr_repo_url(
env_name='cgap-wolf') == (main_repo if use_ecosystem_repo else '')
def test_c4_orchestration_manager_all_stacks():
mocked_boto3 = MockBoto3()
mock_stack_names = [
'foo-bar-baz', 'awseb-e-123-stack', 'c4-foo-123', 'c4-foo-456'
]
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mock_stack_names=mock_stack_names)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
manager = cloudformation_utils.C4OrchestrationManager()
all_stacks = manager.all_stacks()
assert sorted(s.name
for s in all_stacks) == ['c4-foo-123', 'c4-foo-456']
def test_get_ecr_repo_url_kmp(
): # Test for my test environment. -kmp 17-Aug-2021
mocked_boto3 = MockBoto3()
supertest_repo = 'xxx0312.dkr.ecr.us-east-1.amazonaws.com/main'
supertest_outputs = [{
'OutputKey': 'C4ECRMainRepoURL',
'OutputValue': supertest_repo,
'Description': 'CGAPDocker Image Repository URL',
'ExportName': 'c4-ecr-main-stack-RepoURL'
}]
mocked_stacks = [
MockBotoCloudFormationStack('c4-iam-trial-alpha-stack'),
MockBotoCloudFormationStack('c4-network-trial-alpha-stack'),
MockBotoCloudFormationStack('c4-ecr-trial-alpha-stack',
mock_outputs=supertest_outputs),
]
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mocked_stacks=mocked_stacks)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
assert cloudformation_utils.get_ecr_repo_url(
'cgap-supertest') == supertest_repo
def test_get_ecr_repo_url_alpha():
mocked_boto3 = MockBoto3()
mastertest_repo = 'xxx6742.dkr.ecr.us-east-1.amazonaws.com/cgap-mastertest'
mastertest_outputs = [
{
'OutputKey': 'C4EcrTrialAlphaECRRepoURL',
'OutputValue': mastertest_repo,
'Description': 'CGAPDocker Image Repository URL',
'ExportName': 'c4-ecr-trial-alpha-stack-ECRRepoURL'
},
]
mocked_stacks = [
MockBotoCloudFormationStack('c4-iam-trial-alpha-stack'),
MockBotoCloudFormationStack('c4-network-trial-alpha-stack'),
MockBotoCloudFormationStack('c4-ecr-trial-alpha-stack',
mock_outputs=mastertest_outputs),
]
MockBotoCloudFormationClient.setup_boto3_mocked_stacks(
boto3=mocked_boto3, mocked_stacks=mocked_stacks)
with mock.patch.object(cloudformation_utils, "boto3", mocked_boto3):
assert cloudformation_utils.get_ecr_repo_url(
'cgap-mastertest') == mastertest_repo