def test_node(self): environments = { NeckbeardLoader.VERSION_OPTION: '0.1', 'test1': { 'name': 'test1', 'seed_environment_name': 'test2', 'aws_nodes': { 'ec2': { 'web': { "name": "web", "unique_id": "web-{{ node.scaling_index }}", "seed": { "name": "web", "scaling_index": 0, }, }, }, }, }, 'test2': { 'name': 'test2', 'aws_nodes': { 'ec2': { 'web': { "name": "web", "unique_id": "web-{{ node.scaling_index }}", }, }, }, }, } configuration = ConfigurationManager( constants={}, secrets={}, secrets_tpl={}, environments=environments, node_templates={}, scaling_backend=MaxScalingBackend(), ) node_context = configuration._get_resource_context( 'test1', 'ec2', 'web', 1, ) expected = { 'environment_name': 'test1', 'seed_environment_name': 'test2', 'resource_type': 'ec2', 'name': 'web', 'scaling_index': 1, } self.assertEqual(node_context, expected)
def test_node_not_zero(self): environments = { NeckbeardLoader.VERSION_OPTION: '0.1', 'test1': { 'name': 'test1', 'aws_nodes': { 'ec2': { 'web': { "name": "web", "unique_id": "web-{{ node.scaling_index }}", }, }, }, }, } configuration = ConfigurationManager( constants={}, secrets={}, secrets_tpl={}, environments=environments, node_templates={}, scaling_backend=MaxScalingBackend(), ) node_context = configuration._get_resource_context( 'test1', 'ec2', 'web', 7, ) expected = { 'environment_name': 'test1', 'seed_environment_name': None, 'resource_type': 'ec2', 'name': 'web', 'scaling_index': 7, } self.assertEqual(len(node_context), len(expected)) for key, value in expected.items(): self.assertEqual(node_context.get(key), value)