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