def test_with_urls(self, log):
collection = loader.load(
self._get_full_path_for_test_data('with_urls.yaml'),
substitutions={'domain': 'example.com'})
self.assertIn(
cloud_models.URLConfStatic(hostname='example.com',
path='/',
local_path='www'), collection.urls)
self.assertIn(
cloud_models.URLConfBackend(hostname='example.com',
path='/api',
destination='webapp/api'),
collection.urls)
self.assertIn(
cloud_models.Node(name='webapp',
flavor='webapp',
image='trusty',
disk=10,
networks=[]), collection.nodes)
self.assertEqual(len(collection.nodes), 1)
self.assertEqual(len(collection.urls), 2)
log.check(('aasemble.deployment.loader', 'DEBUG',
'Loaded static URL ${domain}/ from stack'),
('aasemble.deployment.loader', 'DEBUG',
'Loaded backend URL ${domain}/api from stack'),
('aasemble.deployment.loader', 'DEBUG',
'Loaded node webapp from stack'))
def test_simple(self, log):
collection = loader.load(
self._get_full_path_for_test_data('simple.yaml'))
self.assertIn(
cloud_models.Node(name='webapp',
flavor='webapp',
image='trusty',
disk=10,
networks=[]), collection.nodes)
self.assertEqual(len(collection.nodes), 1)
self.assertEqual(len(collection.security_groups), 0)
self.assertEqual(len(collection.security_group_rules), 0)
log.check(('aasemble.deployment.loader', 'DEBUG',
'Loaded node webapp from stack'))
def test_with_script(self, interpolate, log):
collection = loader.load(
self._get_full_path_for_test_data('with_script.yaml'))
sg = cloud_models.SecurityGroup(name='webapp')
script = '#!/bin/sh\nadduser --system web\napt-get install python-virtualenv\netc. etc. etc.\n'
self.assertIn(
cloud_models.Node(name='webapp1',
flavor='webapp',
image='trusty',
disk=10,
networks=[],
security_groups=set([sg]),
script=interpolate.return_value),
collection.nodes)
self.assertIn(
cloud_models.Node(name='webapp2',
flavor='webapp',
image='trusty',
disk=10,
networks=[],
security_groups=set([sg]),
script=interpolate.return_value),
collection.nodes)
interpolate.assert_called_with(script, None)
self.assertIn(sg, collection.security_groups)
self.assertIn(
cloud_models.SecurityGroupRule(security_group=sg,
source_ip='0.0.0.0/0',
from_port=443,
to_port=443,
protocol='tcp'),
collection.security_group_rules)
self.assertEqual(len(collection.nodes), 2)
self.assertEqual(len(collection.security_groups), 1)
self.assertEqual(len(collection.security_group_rules), 1)
log.check(('aasemble.deployment.loader', 'DEBUG',
'Loaded node webapp1 from stack'),
('aasemble.deployment.loader', 'DEBUG',
'Loaded node webapp2 from stack'),
('aasemble.deployment.loader', 'DEBUG',
'Loaded security group webapp from stack'),
('aasemble.deployment.loader', 'DEBUG',
'Loaded security group rule from stack: tcp: 443-443'))
def test_with_security_groups(self, log):
collection = loader.load(
self._get_full_path_for_test_data('with_security_groups.yaml'))
sg = cloud_models.SecurityGroup(name='webapp')
self.assertIn(
cloud_models.Node(name='webapp1',
flavor='webapp',
image='trusty',
disk=10,
networks=[],
security_groups=set([sg])), collection.nodes)
self.assertIn(
cloud_models.Node(name='webapp2',
flavor='webapp',
image='trusty',
disk=10,
networks=[],
security_groups=set([sg])), collection.nodes)
self.assertIn(sg, collection.security_groups)
self.assertIn(
cloud_models.SecurityGroupRule(security_group=sg,
source_ip='0.0.0.0/0',
from_port=443,
to_port=443,
protocol='tcp'),
collection.security_group_rules)
self.assertEqual(len(collection.nodes), 2)
self.assertEqual(len(collection.security_groups), 1)
self.assertEqual(len(collection.security_group_rules), 1)
log.check(('aasemble.deployment.loader', 'DEBUG',
'Loaded node webapp1 from stack'),
('aasemble.deployment.loader', 'DEBUG',
'Loaded node webapp2 from stack'),
('aasemble.deployment.loader', 'DEBUG',
'Loaded security group webapp from stack'),
('aasemble.deployment.loader', 'DEBUG',
'Loaded security group rule from stack: tcp: 443-443'))
def apply(options):
substitutions = extract_substitutions(options.substitutions)
cluster = handle_cluster_opts(options, substitutions)
LOG.info('Cluster ID: %s', cluster)
resources = loader.load(options.stack, substitutions)
cloud_driver_class, cloud_driver_kwargs, mappings = load_cloud_config(cloud_config_path(options.cloud))
pool = ThreadPool(options.threads)
cloud_driver = cloud_driver_class(mappings=mappings,
pool=pool,
namespace=options.namespace,
cluster=cluster,
**cloud_driver_kwargs)
if not options.assume_empty:
current_resources = cloud_driver.detect_resources()
resources = resources - current_resources
cloud_driver.apply_resources(resources)
print(format_collection(resources))
print('Cluster ID: {}'.format(cluster))