def test_services_are_enabled_and_stared():
configuration = SimpleMachineConfiguration(services=[SimpleServiceConfig('sshd')])
result = as_userdata_string(configuration)
assert 'systemctl enable sshd' in result
assert 'systemctl start sshd' in result
def first_boot(self, machine_configuration, size_slug):
try:
existing_id = self.find_image_id(machine_configuration)
raise ValueError(
'Snapshot already exists for \'{name}\' - id is \'{id}\''.
format(name=machine_configuration.name, id=existing_id))
except KeyError:
pass
configuration = str(userdata.as_userdata_string(machine_configuration))
droplet = digitalocean.Droplet(
token=self.authentication_token,
name=machine_configuration.name,
region=self.region_name,
image=self.find_image_id(
machine_configuration.base_image_configuration),
size_slug=size_slug,
backups=False,
ssh_keys=[self.root_ssh_key_fingerprint],
user_data=configuration)
droplet.create()
return Droplet(self, droplet, machine_configuration)
def test_services_not_requiring_start_are_not_started():
configuration = SimpleMachineConfiguration(
services=[SimpleServiceConfig('sshd', requires_start=False)])
result = as_userdata_string(configuration)
assert 'systemctl start sshd' not in result
def test_service_installation_come_after_raw_commands():
configuration = SimpleMachineConfiguration(
services=[SimpleServiceConfig('sshd')], raw_commands=['cmd1'])
result = as_userdata_string(configuration)
assert result.index('sshd') > result.index('cmd1')
def test_outputs_raw_commands_as_yaml_items():
assert '''runcmd:
- cmd1 with args
- cmd2
- cmd3
''' in as_userdata_string(
SimpleMachineConfiguration(raw_commands=['cmd1 with args', 'cmd2', 'cmd3'], required_packages=None))
def test_services_requiring_restart_come_after_raw_commands():
configuration = SimpleMachineConfiguration(services=[SimpleServiceConfig('sshd', requires_restart=True)],
raw_commands=['cmd1'])
result = as_userdata_string(configuration)
assert 'systemctl restart sshd' in result
assert result.index('restart') > result.index('cmd1')
def test_services_are_enabled_and_stared():
configuration = SimpleMachineConfiguration(
services=[SimpleServiceConfig('sshd')])
result = as_userdata_string(configuration)
assert 'systemctl enable sshd' in result
assert 'systemctl start sshd' in result
def test_services_requiring_restart_come_after_raw_commands():
configuration = SimpleMachineConfiguration(
services=[SimpleServiceConfig('sshd', requires_restart=True)],
raw_commands=['cmd1'])
result = as_userdata_string(configuration)
assert 'systemctl restart sshd' in result
assert result.index('restart') > result.index('cmd1')
def test_outputs_raw_commands_as_yaml_items():
assert '''runcmd:
- cmd1 with args
- cmd2
- cmd3
''' in as_userdata_string(
SimpleMachineConfiguration(
raw_commands=['cmd1 with args', 'cmd2', 'cmd3'],
required_packages=None))
def test_files_marked_with_executable_flags_if_they_were_configured_for_them():
configuration = SimpleMachineConfiguration(raw_commands=['cmd1'],
files=[
SimpleFileForCopyingRemote(b'content', '/usr/bin/runnable',
flags=['executable']),
SimpleFileForCopyingRemote(b'content', '/usr/bin/not_runnable')
])
result = as_userdata_string(configuration)
assert 'chmod +x /usr/bin/runnable' in result
assert 'chmod +x /usr/bin/not_runnable' not in result
def test_file_copy_steps_are_gzip_and_b64_encoded():
configuration = SimpleMachineConfiguration(raw_commands=['cmd1'],
files=[SimpleFileForCopyingRemote(b'content', '/etc/something.conf')])
result = as_userdata_string(configuration)
line_with_file_copy = [l for l in result.split('\n') if '/etc/something.conf' in l][0]
content = line_with_file_copy[len(' - echo \''):line_with_file_copy.find('\' |')]
result = gzip.decompress(base64.standard_b64decode(content))
assert result == b'content'
assert 'gzip -d ' in line_with_file_copy
assert 'base64 -d ' in line_with_file_copy
def test_files_marked_with_executable_flags_if_they_were_configured_for_them():
configuration = SimpleMachineConfiguration(
raw_commands=['cmd1'],
files=[
SimpleFileForCopyingRemote(b'content',
'/usr/bin/runnable',
flags=['executable']),
SimpleFileForCopyingRemote(b'content', '/usr/bin/not_runnable')
])
result = as_userdata_string(configuration)
assert 'chmod +x /usr/bin/runnable' in result
assert 'chmod +x /usr/bin/not_runnable' not in result
def test_file_copy_steps_are_gzip_and_b64_encoded():
configuration = SimpleMachineConfiguration(
raw_commands=['cmd1'],
files=[SimpleFileForCopyingRemote(b'content', '/etc/something.conf')])
result = as_userdata_string(configuration)
line_with_file_copy = [
l for l in result.split('\n') if '/etc/something.conf' in l
][0]
content = line_with_file_copy[len(' - echo \''):line_with_file_copy.
find('\' |')]
result = gzip.decompress(base64.standard_b64decode(content))
assert result == b'content'
assert 'gzip -d ' in line_with_file_copy
assert 'base64 -d ' in line_with_file_copy
def first_boot(self, machine_configuration, size_slug):
try:
existing_id = self.find_image_id(machine_configuration)
raise ValueError(
'Snapshot already exists for \'{name}\' - id is \'{id}\''.format(name=machine_configuration.name,
id=existing_id))
except KeyError:
pass
configuration = str(userdata.as_userdata_string(machine_configuration))
droplet = digitalocean.Droplet(token=self.authentication_token,
name=machine_configuration.name,
region=self.region_name,
image=self.find_image_id(machine_configuration.base_image_configuration),
size_slug=size_slug,
backups=False,
ssh_keys=[self.root_ssh_key_fingerprint],
user_data=configuration)
droplet.create()
return Droplet(self, droplet, machine_configuration)
def test_service_installation_come_after_raw_commands():
configuration = SimpleMachineConfiguration(services=[SimpleServiceConfig('sshd')], raw_commands=['cmd1'])
result = as_userdata_string(configuration)
assert result.index('sshd') > result.index('cmd1')
def test_adds_file_copy_steps_before_other_runcmds():
configuration = SimpleMachineConfiguration(
raw_commands=['cmd1'],
files=[SimpleFileForCopyingRemote(b'content', '/etc/something.conf')])
result = as_userdata_string(configuration)
assert result.index('cmd1') > result.index('/etc/something.conf')
def test_adds_required_packages_before_runcmd_section():
result = as_userdata_string(
SimpleMachineConfiguration(raw_commands=['cmd1'],
required_packages=['package1']))
assert result.index('packages:') < result.index('runcmd:')
def test_adds_required_packages():
assert '''packages:
- super_service
''' in as_userdata_string(
SimpleMachineConfiguration(raw_commands=[],
required_packages=['super_service']))
def test_starts_with_cloud_config_header():
result = as_userdata_string(SimpleMachineConfiguration(None, None))
assert result.startswith("#cloud-config")
def test_shuts_down_machine_when_done():
empty_config = SimpleMachineConfiguration()
result = as_userdata_string(empty_config)
assert result.endswith('shutdown +1')
def test_services_not_requiring_start_are_not_started():
configuration = SimpleMachineConfiguration(services=[SimpleServiceConfig('sshd', requires_start=False)])
result = as_userdata_string(configuration)
assert 'systemctl start sshd' not in result
def test_shuts_down_machine_when_done():
empty_config = SimpleMachineConfiguration()
result = as_userdata_string(empty_config)
assert result.endswith('shutdown +1')
def test_starts_with_cloud_config_header():
result = as_userdata_string(SimpleMachineConfiguration(None, None))
assert result.startswith("#cloud-config")
def test_adds_required_packages():
assert '''packages:
- super_service
''' in as_userdata_string(SimpleMachineConfiguration(raw_commands=[], required_packages=['super_service']))
def test_adds_required_packages_before_runcmd_section():
result = as_userdata_string(SimpleMachineConfiguration(raw_commands=['cmd1'], required_packages=['package1']))
assert result.index('packages:') < result.index('runcmd:')
def test_adds_file_copy_steps_before_other_runcmds():
configuration = SimpleMachineConfiguration(raw_commands=['cmd1'],
files=[SimpleFileForCopyingRemote(b'content', '/etc/something.conf')])
result = as_userdata_string(configuration)
assert result.index('cmd1') > result.index('/etc/something.conf')