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)
