def test_dependencies_resolved_through_conditions(self):
# Arrange
t = Template.parse_file(os.path.join(self.test_data_dir, 'WordPress_Multi_AZ.template'))
expected_parameters = {
'DBName',
'MultiAZDatabase',
'DBUser',
'DBPassword',
'DBClass',
'DBAllocatedStorage',
'SSHLocation'
}
# These resources resolve through an "If"
expected_resources = {
'DBEC2SecurityGroup',
'DBSecurityGroup',
'ElasticLoadBalancer',
'WebServerSecurityGroup'
}
# Act
db_instance = t.get_resource('DBInstance')
# Assert
dependencies = db_instance.get_all_dependencies()
resources = set(r.logical_id for r in dependencies if r.element_type == ElementType.resource)
parameters = set(p.logical_id for p in dependencies if p.element_type == ElementType.parameter)
self.assertSetEqual(expected_parameters, parameters)
self.assertSetEqual(expected_resources, resources)
def test_resource_dependency_tree(self):
# Arrange
t = Template.parse_file(os.path.join(self.test_data_dir, 'single_server.template'))
expected_parameters = {
'KeyName',
'InstanceType',
'SourceCidrForRDP'
}
expected_resources = {
'SharePointFoundationSecurityGroup',
'SharePointFoundationWaitHandle'
}
# Act
instance = t.get_resource('SharePointFoundation')
# Assert
parameters = set()
resources = set()
def collect(item, level, visited):
if item.element_type == ElementType.parameter:
parameters.add(item.logical_id)
elif item.element_type == ElementType.resource:
resources.add(item.logical_id)
instance.visit_dependencies(collect)
self.assertSetEqual(expected_parameters, parameters)
self.assertSetEqual(expected_resources, resources)
