def execute(self, exit=True):
self.molecule._create_templates()
ansible = AnsiblePlaybook(self.molecule._config.config['ansible'])
ansible.add_cli_arg('syntax-check', True)
return ansible.execute(hide_errors=True)
def execute(self,
idempotent=False,
create_instances=True,
create_inventory=True,
exit=True,
hide_errors=True):
"""
:param idempotent: Optionally provision servers quietly so output can be parsed for idempotence
:param create_inventory: Toggle inventory creation
:param create_instances: Toggle instance creation
:return: Provisioning output
"""
if self.molecule._state.get('created'):
create_instances = False
if self.molecule._state.get('converged'):
create_inventory = False
if self.static:
create_instances = False
create_inventory = False
if create_instances and not idempotent:
command_args, args = utilities.remove_args(self.command_args,
self.args, ['--tags'])
c = Create(command_args, args, self.molecule)
c.execute()
if create_inventory:
self.molecule._create_inventory_file()
# install role dependencies only during `molecule converge`
if not idempotent and 'requirements_file' in self.molecule._config.config[
'ansible']:
print('{}Installing role dependencies ...{}'.format(
colorama.Fore.CYAN, colorama.Fore.RESET))
galaxy_install = AnsibleGalaxyInstall(self.molecule._config.config[
'ansible']['requirements_file'])
galaxy_install.add_env_arg(
'ANSIBLE_CONFIG',
self.molecule._config.config['ansible']['config_file'])
galaxy_install.bake()
output = galaxy_install.execute()
ansible = AnsiblePlaybook(self.molecule._config.config['ansible'])
# params to work with provisioner
for k, v in self.molecule._provisioner.ansible_connection_params.items(
):
ansible.add_cli_arg(k, v)
# target tags passed in via CLI
if self.molecule._args.get('--tags'):
ansible.add_cli_arg('tags', self.molecule._args['--tags'].pop(0))
if idempotent:
ansible.remove_cli_arg('_out')
ansible.remove_cli_arg('_err')
ansible.add_env_arg('ANSIBLE_NOCOLOR', 'true')
ansible.add_env_arg('ANSIBLE_FORCE_COLOR', 'false')
# Save the previous callback plugin if any.
callback_plugin = ansible.env.get('ANSIBLE_CALLBACK_PLUGINS', '')
# Set the idempotence plugin.
if callback_plugin:
ansible.add_env_arg(
'ANSIBLE_CALLBACK_PLUGINS',
callback_plugin + ':' + os.path.join(
sys.prefix,
'share/molecule/ansible/plugins/callback/idempotence'))
else:
ansible.add_env_arg('ANSIBLE_CALLBACK_PLUGINS', os.path.join(
sys.prefix,
'share/molecule/ansible/plugins/callback/idempotence'))
ansible.bake()
if self.molecule._args.get('--debug'):
ansible_env = {k: v
for (k, v) in ansible.env.items() if 'ANSIBLE' in k}
other_env = {k: v
for (k, v) in ansible.env.items()
if 'ANSIBLE' not in k}
utilities.debug('OTHER ENVIRONMENT',
yaml.dump(other_env,
default_flow_style=False,
indent=2))
utilities.debug('ANSIBLE ENVIRONMENT',
yaml.dump(ansible_env,
default_flow_style=False,
indent=2))
utilities.debug('ANSIBLE PLAYBOOK', str(ansible.ansible))
status, output = ansible.execute(hide_errors=hide_errors)
if status is not None:
if exit:
sys.exit(status)
return status, None
if not self.molecule._state.get('converged'):
self.molecule._state['converged'] = True
self.molecule._write_state_file()
return None, output
def execute(self,
idempotent=False,
create_instances=True,
create_inventory=True,
exit=True,
hide_errors=True):
"""
:param idempotent: Optionally provision servers quietly so output can be parsed for idempotence
:param create_inventory: Toggle inventory creation
:param create_instances: Toggle instance creation
:return: Provisioning output
"""
if self.molecule._state.get('created'):
create_instances = False
if self.molecule._state.get('converged'):
create_inventory = False
if self.static:
create_instances = False
create_inventory = False
if create_instances and not idempotent:
command_args, args = utilities.remove_args(self.command_args,
self.args, ['--tags'])
c = Create(command_args, args, self.molecule)
c.execute()
if create_inventory:
self.molecule._create_inventory_file()
# install role dependencies only during `molecule converge`
if not idempotent and 'requirements_file' in self.molecule._config.config[
'ansible']:
print('{}Installing role dependencies ...{}'.format(
Fore.CYAN, Fore.RESET))
galaxy_install = AnsibleGalaxyInstall(
self.molecule._config.config['ansible']['requirements_file'])
galaxy_install.add_env_arg(
'ANSIBLE_CONFIG',
self.molecule._config.config['ansible']['config_file'])
galaxy_install.bake()
output = galaxy_install.execute()
ansible = AnsiblePlaybook(self.molecule._config.config['ansible'])
# target tags passed in via CLI
if self.molecule._args.get('--tags'):
ansible.add_cli_arg('tags', self.molecule._args['--tags'].pop(0))
if idempotent:
ansible.remove_cli_arg('_out')
ansible.remove_cli_arg('_err')
ansible.add_env_arg('ANSIBLE_NOCOLOR', 'true')
ansible.add_env_arg('ANSIBLE_FORCE_COLOR', 'false')
# Save the previous callback plugin if any.
callback_plugin = ansible.env.get('ANSIBLE_CALLBACK_PLUGINS', '')
# Set the idempotence plugin.
if callback_plugin:
ansible.add_env_arg(
'ANSIBLE_CALLBACK_PLUGINS',
callback_plugin + ':' + os.path.join(
sys.prefix,
'share/molecule/ansible/plugins/callback/idempotence'))
else:
ansible.add_env_arg(
'ANSIBLE_CALLBACK_PLUGINS',
os.path.join(
sys.prefix,
'share/molecule/ansible/plugins/callback/idempotence'))
ansible.bake()
if self.molecule._args.get('--debug'):
ansible_env = {
k: v
for (k, v) in ansible.env.items() if 'ANSIBLE' in k
}
other_env = {
k: v
for (k, v) in ansible.env.items() if 'ANSIBLE' not in k
}
utilities.debug(
'OTHER ENVIRONMENT',
yaml.dump(other_env, default_flow_style=False, indent=2))
utilities.debug(
'ANSIBLE ENVIRONMENT',
yaml.dump(ansible_env, default_flow_style=False, indent=2))
utilities.debug('ANSIBLE PLAYBOOK', str(ansible.ansible))
status, output = ansible.execute(hide_errors=hide_errors)
if status is not None:
if exit:
sys.exit(status)
return status, None
if not self.molecule._state.get('converged'):
self.molecule._state['converged'] = True
self.molecule._write_state_file()
return None, output
class TestDockerProvisioner(testtools.TestCase):
def setUp(self):
super(TestDockerProvisioner, self).setUp()
# Setup mock molecule
self._mock_molecule = Molecule(dict())
self.temp = '/tmp/test_config_load_defaults_external_file.yml'
data = {
'molecule': {
'molecule_dir': '.test_molecule',
'inventory_file': 'tests/ansible_inventory'
},
'docker': {
'containers': [
{'name': 'test1',
'image': 'ubuntu',
'image_version': 'latest',
'ansible_groups': ['group1']},
{'name': 'test2',
'image': 'ubuntu',
'image_version': 'latest',
'ansible_groups': ['group2']}
]
},
'ansible': {
'config_file': 'test_config',
'inventory_file': 'test_inventory'
}
}
with open(self.temp, 'w') as f:
f.write(yaml.dump(data, default_flow_style=True))
self._mock_molecule._config.load_defaults_file(defaults_file=self.temp)
self._mock_molecule._state = dict()
def test_name(self):
docker_provisioner = provisioners.DockerProvisioner(self._mock_molecule)
# false values don't exist in arg dict at all
self.assertEqual(docker_provisioner.name, 'docker')
def test_get_provisioner(self):
self.assertEqual(provisioners.get_provisioner(self._mock_molecule).name, 'docker')
def test_up(self):
docker_provisioner = provisioners.DockerProvisioner(self._mock_molecule)
docker_provisioner.up()
docker_provisioner.destroy()
def test_instances(self):
docker_provisioner = provisioners.DockerProvisioner(self._mock_molecule)
self.assertEqual(docker_provisioner.instances[0]['name'], 'test1')
self.assertEqual(docker_provisioner.instances[1]['name'], 'test2')
def test_status(self):
docker_provisioner = provisioners.DockerProvisioner(self._mock_molecule)
docker_provisioner.up()
self.assertEquals('test1', docker_provisioner.status()[1].name)
self.assertEquals('test2', docker_provisioner.status()[0].name)
self.assertIn('Up', docker_provisioner.status()[1].state)
self.assertIn('Up', docker_provisioner.status()[0].state)
self.assertEqual('docker', docker_provisioner.status()[0].provider)
self.assertEqual('docker', docker_provisioner.status()[1].provider)
def test_destroy(self):
docker_provisioner = provisioners.DockerProvisioner(self._mock_molecule)
docker_provisioner.up()
self.assertEquals('test1', docker_provisioner.status()[1].name)
self.assertEquals('test2', docker_provisioner.status()[0].name)
self.assertIn('Up', docker_provisioner.status()[1].state)
self.assertIn('Up', docker_provisioner.status()[0].state)
docker_provisioner.destroy()
self.assertIn('Not Created', docker_provisioner.status()[1].state)
self.assertIn('Not Created', docker_provisioner.status()[0].state)
def test_provision(self):
docker_provisioner = provisioners.DockerProvisioner(self._mock_molecule)
docker_provisioner.destroy()
docker_provisioner.up()
self.book = docker_provisioner.ansible_connection_params
self.book['playbook'] = 'tests/playbook.yml'
self.book['inventory'] = 'test1,test2,'
self.ansible = AnsiblePlaybook(self.book)
self.assertEqual((None, ''), self.ansible.execute())
docker_provisioner.destroy()
def test_inventory_generation(self):
self._mock_molecule._provisioner = provisioners.get_provisioner(self._mock_molecule)
self._mock_molecule._provisioner.destroy()
self._mock_molecule._provisioner.up()
self._mock_molecule._create_inventory_file()
self.book = self._mock_molecule._provisioner.ansible_connection_params
self.book['playbook'] = 'tests/playbook.yml'
self.book['inventory'] = 'tests/ansible_inventory'
self.ansible = AnsiblePlaybook(self.book)
self.assertEqual((None, ''), self.ansible.execute())
def execute(self, idempotent=False, create_instances=True, create_inventory=True):
"""
Provisions all instances using ansible-playbook.
:param idempotent: Optionally provision servers quietly so output can be parsed for idempotence
:param create_inventory: Toggle inventory creation
:param create_instances: Toggle instance creation
:return: Provisioning output
"""
if self.molecule._state.get('created'):
create_instances = False
if self.molecule._state.get('converged'):
create_inventory = False
if self.static:
create_instances = False
create_inventory = False
if create_instances and not idempotent:
c = Create(self.args, self.molecule)
c.execute()
if create_inventory:
self.molecule._create_inventory_file()
ansible = AnsiblePlaybook(self.molecule._config.config['ansible'])
# target tags passed in via CLI
ansible.add_cli_arg('tags', self.molecule._args.get('--tags'))
if idempotent:
ansible.remove_cli_arg('_out')
ansible.remove_cli_arg('_err')
ansible.add_env_arg('ANSIBLE_NOCOLOR', 'true')
ansible.add_env_arg('ANSIBLE_FORCE_COLOR', 'false')
# Save the previous callback plugin if any.
callback_plugin = ansible.env.get('ANSIBLE_CALLBACK_PLUGINS', '')
# Set the idempotence plugin.
if callback_plugin:
ansible.add_env_arg('ANSIBLE_CALLBACK_PLUGINS', callback_plugin + ':' + os.path.join(
sys.prefix, 'share/molecule/ansible/plugins/callback/idempotence'))
else:
ansible.add_env_arg('ANSIBLE_CALLBACK_PLUGINS',
os.path.join(sys.prefix, 'share/molecule/ansible/plugins/callback/idempotence'))
ansible.bake()
if self.molecule._args['--debug']:
ansible_env = {k: v for (k, v) in ansible.env.items() if 'ANSIBLE' in k}
other_env = {k: v for (k, v) in ansible.env.items() if 'ANSIBLE' not in k}
utilities.debug('OTHER ENVIRONMENT', yaml.dump(other_env, default_flow_style=False, indent=2))
utilities.debug('ANSIBLE ENVIRONMENT', yaml.dump(ansible_env, default_flow_style=False, indent=2))
utilities.debug('ANSIBLE PLAYBOOK', str(ansible.ansible))
output = ansible.execute()
if not self.molecule._state.get('converged'):
self.molecule._state['converged'] = True
self.molecule._write_state_file()
return output
class TestDockerProvisioner(testtools.TestCase):
def setUp(self):
super(TestDockerProvisioner, self).setUp()
# Setup mock molecule
self._mock_molecule = Molecule(dict())
self.temp = '/tmp/test_config_load_defaults_external_file.yml'
data = {
'molecule': {
'molecule_dir': '.test_molecule',
'inventory_file': 'tests/ansible_inventory'
},
'docker': {
'containers': [{
'name': 'test1',
'image': 'ubuntu',
'image_version': 'latest',
'ansible_groups': ['group1']
}, {
'name': 'test2',
'image': 'ubuntu',
'image_version': 'latest',
'ansible_groups': ['group2']
}]
},
'ansible': {
'config_file': 'test_config',
'inventory_file': 'test_inventory'
}
}
with open(self.temp, 'w') as f:
f.write(yaml.dump(data, default_flow_style=True))
self._mock_molecule._config.load_defaults_file(defaults_file=self.temp)
self._mock_molecule._state = dict()
def test_name(self):
docker_provisioner = provisioners.DockerProvisioner(
self._mock_molecule)
# false values don't exist in arg dict at all
self.assertEqual(docker_provisioner.name, 'docker')
def test_get_provisioner(self):
self.assertEqual(
provisioners.get_provisioner(self._mock_molecule).name, 'docker')
def test_up(self):
docker_provisioner = provisioners.DockerProvisioner(
self._mock_molecule)
docker_provisioner.up()
docker_provisioner.destroy()
def test_instances(self):
docker_provisioner = provisioners.DockerProvisioner(
self._mock_molecule)
self.assertEqual(docker_provisioner.instances[0]['name'], 'test1')
self.assertEqual(docker_provisioner.instances[1]['name'], 'test2')
def test_status(self):
docker_provisioner = provisioners.DockerProvisioner(
self._mock_molecule)
docker_provisioner.up()
self.assertEquals('test1', docker_provisioner.status()[1].name)
self.assertEquals('test2', docker_provisioner.status()[0].name)
self.assertIn('Up', docker_provisioner.status()[1].state)
self.assertIn('Up', docker_provisioner.status()[0].state)
self.assertEqual('docker', docker_provisioner.status()[0].provider)
self.assertEqual('docker', docker_provisioner.status()[1].provider)
def test_destroy(self):
docker_provisioner = provisioners.DockerProvisioner(
self._mock_molecule)
docker_provisioner.up()
self.assertEquals('test1', docker_provisioner.status()[1].name)
self.assertEquals('test2', docker_provisioner.status()[0].name)
self.assertIn('Up', docker_provisioner.status()[1].state)
self.assertIn('Up', docker_provisioner.status()[0].state)
docker_provisioner.destroy()
self.assertIn('Not Created', docker_provisioner.status()[1].state)
self.assertIn('Not Created', docker_provisioner.status()[0].state)
def test_provision(self):
docker_provisioner = provisioners.DockerProvisioner(
self._mock_molecule)
docker_provisioner.destroy()
docker_provisioner.up()
self.book = docker_provisioner.ansible_connection_params
self.book['playbook'] = 'tests/playbook.yml'
self.book['inventory'] = 'test1,test2,'
self.ansible = AnsiblePlaybook(self.book)
self.assertEqual((None, ''), self.ansible.execute())
docker_provisioner.destroy()
def test_inventory_generation(self):
self._mock_molecule._provisioner = provisioners.get_provisioner(
self._mock_molecule)
self._mock_molecule._provisioner.destroy()
self._mock_molecule._provisioner.up()
self._mock_molecule._create_inventory_file()
self.book = self._mock_molecule._provisioner.ansible_connection_params
self.book['playbook'] = 'tests/playbook.yml'
self.book['inventory'] = 'tests/ansible_inventory'
self.ansible = AnsiblePlaybook(self.book)
self.assertEqual((None, ''), self.ansible.execute())
