def test_debug(capsys):
util.debug('test_title', 'test_data')
result_title, _ = capsys.readouterr()
print(''.join([colorama.Back.WHITE, colorama.Style.BRIGHT,
colorama.Fore.BLACK, 'DEBUG: ' + 'test_title',
colorama.Fore.RESET, colorama.Back.RESET,
colorama.Style.RESET_ALL]))
print(''.join([colorama.Fore.BLACK, colorama.Style.BRIGHT, 'test_data',
colorama.Style.RESET_ALL, colorama.Fore.RESET]))
expected_title, _ = capsys.readouterr()
assert expected_title == result_title
def main(self):
if not os.path.exists(self.config.config['molecule']['molecule_dir']):
os.makedirs(self.config.config['molecule']['molecule_dir'])
self.state = state.State(
state_file=self.config.config.get('molecule').get('state_file'))
try:
self.driver = self._get_driver()
except basedriver.InvalidDriverSpecified:
LOG.error("Invalid driver '{}'".format(self._get_driver_name()))
# TODO(retr0h): Print valid drivers.
util.sysexit()
except basedriver.InvalidProviderSpecified:
LOG.error("Invalid provider '{}'".format(self.args['--provider']))
self.args['--provider'] = None
self.args['--platform'] = None
self.driver = self._get_driver()
self.print_valid_providers()
util.sysexit()
except basedriver.InvalidPlatformSpecified:
LOG.error("Invalid platform '{}'".format(self.args['--platform']))
self.args['--provider'] = None
self.args['--platform'] = None
self.driver = self._get_driver()
self.print_valid_platforms()
util.sysexit()
# updates instances config with full machine names
self.config.populate_instance_names(self.driver.platform)
if self.args.get('--debug'):
util.debug(
'RUNNING CONFIG',
yaml.dump(self.config.config,
default_flow_style=False,
indent=2))
self._add_or_update_vars('group_vars')
self._add_or_update_vars('host_vars')
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.created:
create_instances = False
if self.molecule._state.converged:
create_inventory = False
if self.molecule._state.multiple_platforms:
self.args['--platform'] = 'all'
else:
if self.args[
'--platform'] == 'all' and self.molecule._state.created:
create_instances = True
create_inventory = True
if create_instances and not idempotent:
command_args, args = util.remove_args(self.command_args, self.args,
['--tags'])
c = create.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'] and not self.molecule._state.installed_deps:
galaxy = ansible_galaxy.AnsibleGalaxy(self.molecule.config.config)
galaxy.install()
self.molecule._state.change_state('installed_deps', True)
ansible = ansible_playbook.AnsiblePlaybook(self.molecule.config.config[
'ansible'])
# params to work with driver
for k, v in self.molecule._driver.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}
util.debug('OTHER ENVIRONMENT',
yaml.dump(other_env,
default_flow_style=False,
indent=2))
util.debug('ANSIBLE ENVIRONMENT',
yaml.dump(ansible_env,
default_flow_style=False,
indent=2))
util.debug('ANSIBLE PLAYBOOK', str(ansible.ansible))
util.print_info("Starting Ansible Run ...")
status, output = ansible.execute(hide_errors=hide_errors)
if status is not None:
if exit:
util.sysexit(status)
return status, None
if not self.molecule._state.converged:
self.molecule._state.change_state('converged', True)
return None, output
def execute(self,
idempotent=False,
create_instances=True,
create_inventory=True,
exit=True,
hide_errors=True):
"""
Execute the actions necessary to perform a `molecule converge` and
return a tuple.
:param idempotent: An optional flag to perform the converge again, and
parse the output for idempotence.
:param create_inventory: An optional flag to toggle inventory creation.
:param create_instances: An optional flag to toggle instance creation.
:return: Return a tuple of (`exit status`, `command output`), otherwise
sys.exit on command failure.
"""
if self.molecule.state.created:
create_instances = False
if self.molecule.state.converged:
create_inventory = False
if self.molecule.state.multiple_platforms:
self.args['--platform'] = 'all'
else:
if self.args['--platform'] == 'all' and self.molecule.state.created:
create_instances = True
create_inventory = True
if create_instances and not idempotent:
command_args, args = util.remove_args(self.command_args, self.args,
['--tags'])
c = create.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'] and not self.molecule.state.installed_deps:
galaxy = ansible_galaxy.AnsibleGalaxy(self.molecule.config.config)
galaxy.install()
self.molecule.state.change_state('installed_deps', True)
ansible = ansible_playbook.AnsiblePlaybook(
self.molecule.config.config['ansible'])
# Params to work with driver
for k, v in self.molecule.driver.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:
# Don't log stdout/err
ansible.remove_cli_arg('_out')
ansible.remove_cli_arg('_err')
# Disable color for regexp
ansible.add_env_arg('ANSIBLE_NOCOLOR', 'true')
ansible.add_env_arg('ANSIBLE_FORCE_COLOR', 'false')
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
}
util.debug(
'OTHER ENVIRONMENT',
yaml.dump(other_env, default_flow_style=False, indent=2))
util.debug(
'ANSIBLE ENVIRONMENT',
yaml.dump(ansible_env, default_flow_style=False, indent=2))
util.debug('ANSIBLE PLAYBOOK', str(ansible._ansible))
util.print_info('Starting Ansible Run ...')
status, output = ansible.execute(hide_errors=hide_errors)
if status is not None:
if exit:
util.sysexit(status)
return status, None
if not self.molecule.state.converged:
self.molecule.state.change_state('converged', True)
return None, output