def add_nodes(self, count, ec2_user_data='', dc_idx=0, rack=0, enable_auto_bootstrap=False):
post_boot_script = self.test_config.get_startup_script()
if self.extra_network_interface:
post_boot_script += configure_eth1_script()
if self.params.get('ip_ssh_connections') == 'ipv6':
post_boot_script += network_config_ipv6_workaround_script()
if isinstance(ec2_user_data, dict):
ec2_user_data['post_configuration_script'] = base64.b64encode(
post_boot_script.encode('utf-8')).decode('ascii')
ec2_user_data = json.dumps(ec2_user_data)
else:
if 'clustername' in ec2_user_data:
ec2_user_data += " --base64postscript={0}".format(
base64.b64encode(post_boot_script.encode('utf-8')).decode('ascii'))
else:
ec2_user_data = post_boot_script
instances = self._create_or_find_instances(count=count, ec2_user_data=ec2_user_data, dc_idx=dc_idx)
added_nodes = [self._create_node(instance, self._ec2_ami_username,
self.node_prefix, node_index,
self.logdir, dc_idx=dc_idx)
for node_index, instance in
enumerate(instances, start=self._node_index + 1)]
for node in added_nodes:
node.enable_auto_bootstrap = enable_auto_bootstrap
if self.params.get('ip_ssh_connections') == 'ipv6' and not node.distro.is_amazon2 and \
not node.distro.is_ubuntu:
node.config_ipv6_as_persistent()
self._node_index += len(added_nodes)
self.nodes += added_nodes
self.write_node_public_ip_file()
self.write_node_private_ip_file()
return added_nodes
def to_string(self) -> str:
script = super().to_string()
if self.aws_additional_interface:
script += configure_eth1_script()
if self.aws_ipv6_workaround:
script += network_config_ipv6_workaround_script()
return script