38 lines
1.3 KiB
Python
38 lines
1.3 KiB
Python
|
import os
|
||
|
|
import yaml
|
||
|
|
import glob
|
||
|
|
import testinfra.utils.ansible_runner
|
||
|
|
from ansible.playbook import Playbook
|
||
|
|
from ansible.cli.playbook import PlaybookCLI
|
||
|
|
|
||
|
|
testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
|
||
|
|
os.environ['MOLECULE_INVENTORY_FILE']).get_hosts('all')
|
||
|
|
|
||
|
|
def read_playbook(playbook):
|
||
|
|
cli_args = [os.path.realpath(playbook), testinfra_hosts]
|
||
|
|
cli = PlaybookCLI(cli_args)
|
||
|
|
cli.parse()
|
||
|
|
loader, inventory, variable_manager = cli._play_prereqs()
|
||
|
|
|
||
|
|
pb = Playbook.load(cli.args[0], variable_manager, loader)
|
||
|
|
|
||
|
|
for play in pb.get_plays():
|
||
|
|
yield variable_manager.get_vars(play)
|
||
|
|
|
||
|
|
def get_playbook():
|
||
|
|
with open(os.path.realpath(' '.join(map(str,glob.glob('molecule.*')))), 'r') as yamlfile:
|
||
|
|
data = yaml.load(yamlfile, Loader=yaml.FullLoader)
|
||
|
|
if 'playbooks' in data['provisioner'].keys():
|
||
|
|
if 'converge' in data['provisioner']['playbooks'].keys():
|
||
|
|
return data['provisioner']['playbooks']['converge']
|
||
|
|
else:
|
||
|
|
return ' '.join(map(str,glob.glob('converge.*')))
|
||
|
|
|
||
|
|
def test_user(host):
|
||
|
|
for vars in read_playbook(get_playbook()):
|
||
|
|
assert host.user(vars['user']['name']).exists
|
||
|
|
if 'group' in vars['user'].keys():
|
||
|
|
assert host.group(vars['user']['group']).exists
|
||
|
|
else:
|
||
|
|
assert host.group(vars['user']['name']).exists
|