def launch_instance(env_type, size=None,
user_data_uri=None, user_data_params=None,
security_groups=None, extra_tags=None,
image_name=None, owner=None):
""" Launch instances
Arguments:
env_type (string): environment-server_type
size (string): size of the instance to launch.
user_data_uri (string): File path or URL for user data script.
If None, Config value will be used.
user_data_params (dict): Dictionary or parameters to supplement
the defaults when generating user-data.
security_groups (list): List of security groups to create (if
necessary) and supplement the defaults.
owner (string): username to set as owner
"""
cloud = get_current_cloud()
environment, server_type = env_type.split("-", 1)
name = get_next_hostname(env_type)
image_name = config.get_cloud_config_value('IMAGE_NAME',
override=image_name)
if size is None:
sizes = config.SIZES
default_size = sizes['default']
size = sizes.get(server_type, default_size)
zone = cloud.next_az(server_type)
key_name = config.CLOUD['PUBLIC_KEY_NAME']
tags = extra_tags or {}
tags.update({
'environment': environment,
'server_type': server_type,
})
if owner:
tags['owner'] = owner
security_groups = add_default_security_groups(server_type, security_groups)
for security_group in security_groups:
create_if_not_exist_security_group(security_group)
user_data = None
user_data_uri = config.get_cloud_config_value('DEFAULT_USER_DATA',
override=user_data_uri)
if user_data_uri is not None:
user_data = get_parsed_document(name, user_data_uri,
'USER_DATA_PARAMS', user_data_params)
return cloud.launch_instance(
name, image_name, size, zone, security_groups, key_name,
user_data=user_data, tags=tags)
def test_config_specified_file_source(minimum_config_fixture):
""" Test that a user data file can be specified from cloud configuration
and then rendered with config based parameters """
desired_subs = {
'key_1': 'config_value_1',
'key_2': 'config_value_2',
}
hostname = 'staging-test-host-001'
with NamedTemporaryFile() as tmp_file:
for key in desired_subs.keys():
tmp_file.write("{{%s}} " % key)
tmp_file.flush()
config.get_cloud().update({
'DEFAULT_USER_DATA': tmp_file.name,
'USER_DATA_PARAMS': desired_subs
})
uri = config.get_cloud_config_value('DEFAULT_USER_DATA', override=None)
user_data = get_parsed_document(entity_name=hostname, uri=uri,
config_params_key='USER_DATA_PARAMS',
additional_params=None)
for value in desired_subs.values():
assert value in user_data
def test_config_specified_file_source(minimum_config_fixture):
""" Test that a user data file can be specified from cloud configuration
and then rendered with config based parameters """
desired_subs = {
'key_1': 'config_value_1',
'key_2': 'config_value_2',
}
hostname = 'staging-test-host-001'
with NamedTemporaryFile() as tmp_file:
for key in desired_subs.keys():
tmp_file.write("{{%s}} " % key)
tmp_file.flush()
config.get_cloud().update({
'DEFAULT_USER_DATA': tmp_file.name,
'USER_DATA_PARAMS': desired_subs
})
uri = config.get_cloud_config_value('DEFAULT_USER_DATA', override=None)
user_data = get_parsed_document(entity_name=hostname,
uri=uri,
config_params_key='USER_DATA_PARAMS',
additional_params=None)
for value in desired_subs.values():
assert value in user_data
def test_arg_specified_url_source(req, minimum_config_fixture):
""" Test that a user data file can be specified from cloud configuration
and then rendered with config and argument based parameters"""
hostname = 'staging-test-host-001'
desired_subs = {
'key_1': 'config_value_1',
'key_2': 'argument_value_2',
'hostname': hostname,
}
config_params = {
'key_1': 'config_value_1',
'key_2': 'config_value_2',
}
config_ud_url = 'http://this.should.not.be.requested.com/user-data.txt'
config.get_cloud().update({
'DEFAULT_USER_DATA': config_ud_url,
'USER_DATA_PARAMS': config_params,
})
cli_params = csv_dict('key_2=argument_value_2')
cli_uri = 'http://this.should.be.requested.com/user-data.txt'
uri = config.get_cloud_config_value('DEFAULT_USER_DATA', override=cli_uri)
ud_contents = ""
for key in desired_subs.keys():
ud_contents += "{{%s}} " % key
req.return_value = Mock(text=ud_contents, status_code=200)
user_data = get_parsed_document(entity_name=hostname,
uri=uri,
config_params_key='USER_DATA_PARAMS',
additional_params=cli_params)
assert_called_with(req, uri)
for value in desired_subs.values():
assert value in user_data
def test_arg_specified_url_source(req, minimum_config_fixture):
""" Test that a user data file can be specified from cloud configuration
and then rendered with config and argument based parameters"""
hostname = 'staging-test-host-001'
desired_subs = {
'key_1': 'config_value_1',
'key_2': 'argument_value_2',
'hostname': hostname,
}
config_params = {
'key_1': 'config_value_1',
'key_2': 'config_value_2',
}
config_ud_url = 'http://this.should.not.be.requested.com/user-data.txt'
config.get_cloud().update({
'DEFAULT_USER_DATA': config_ud_url,
'USER_DATA_PARAMS': config_params,
})
cli_params = csv_dict('key_2=argument_value_2')
cli_uri = 'http://this.should.be.requested.com/user-data.txt'
uri = config.get_cloud_config_value('DEFAULT_USER_DATA', override=cli_uri)
ud_contents = ""
for key in desired_subs.keys():
ud_contents += "{{%s}} " % key
req.return_value = Mock(text=ud_contents, status_code=200)
user_data = get_parsed_document(entity_name=hostname, uri=uri,
config_params_key='USER_DATA_PARAMS',
additional_params=cli_params)
assert_called_with(req, uri)
for value in desired_subs.values():
assert value in user_data
