def setUp(self):
super(TestConfig, self).setUp()
self.data = {
'config_file': 'test.cfg',
'requirements_file': 'requirements.yml'
}
self.galaxy_install = AnsibleGalaxyInstall(
self.data['requirements_file'])
class TestConfig(testtools.TestCase):
def setUp(self):
super(TestConfig, self).setUp()
self.data = {
'config_file': 'test.cfg',
'requirements_file': 'requirements.yml'
}
self.galaxy_install = AnsibleGalaxyInstall(self.data['requirements_file'])
def test_requirements_file_loading(self):
self.assertEqual(self.galaxy_install.requirements_file, self.data['requirements_file'])
def test_add_env_arg(self):
self.galaxy_install.add_env_arg('MOLECULE_1', 'test')
self.assertEqual(self.galaxy_install.env['MOLECULE_1'], 'test')
def setUp(self):
super(TestConfig, self).setUp()
self.data = {
'config_file': 'test.cfg',
'requirements_file': 'requirements.yml'
}
self.galaxy_install = AnsibleGalaxyInstall(self.data['requirements_file'])
class TestConfig(testtools.TestCase):
def setUp(self):
super(TestConfig, self).setUp()
self.data = {
'config_file': 'test.cfg',
'requirements_file': 'requirements.yml'
}
self.galaxy_install = AnsibleGalaxyInstall(
self.data['requirements_file'])
def test_requirements_file_loading(self):
self.assertEqual(self.galaxy_install.requirements_file,
self.data['requirements_file'])
def test_add_env_arg(self):
self.galaxy_install.add_env_arg('MOLECULE_1', 'test')
self.assertEqual(self.galaxy_install.env['MOLECULE_1'], 'test')
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
