#!/usr/bin/env python
import os
import functools
import tempfile
from subprocess import Popen
from unittest import TestCase
import pytest
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
def get_inventory_content():
ipa_server_host = os.getenv("IPA_SERVER_HOST")
return "[ipaserver]\n{}".format(ipa_server_host).encode("utf8")
def run_playbook(playbook):
with tempfile.NamedTemporaryFile() as inventory_file:
inventory_file.write(get_inventory_content())
inventory_file.flush()
cmd = [
"ansible-playbook",
"-i",
inventory_file.name,
playbook,
]
process = Popen(cmd, cwd=SCRIPT_DIR)
process.wait()
return process
def list_test_yaml(dir_path):
yamls = []
for yaml_name in os.listdir(dir_path):
if yaml_name.startswith("test_") and yaml_name.endswith(".yml"):
yamls.append(
{
"path": os.path.join(dir_path, yaml_name),
"name": yaml_name.split(".")[0],
}
)
return yamls
def get_test_groups():
test_dirs = os.listdir(SCRIPT_DIR)
groups = {}
for test_group_dir in test_dirs:
group_dir_path = os.path.join(SCRIPT_DIR, test_group_dir)
if not os.path.isdir(group_dir_path):
continue
yamls = list_test_yaml(group_dir_path)
if yamls:
groups[test_group_dir] = yamls
return groups
def prepare_test(test_name, test_path):
def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
kwargs["test_path"] = test_path
return func(*args, **kwargs)
return wrapper
decorator.__name__ = test_name
return decorator
# Dynamically create the TestCase classes with respective
# test_* methods.
for group_name, group_tests in get_test_groups().items():
_tests = {}
for test_config in group_tests:
test_name = test_config["name"].replace("-", "_")
test_path = test_config["path"]
@pytest.mark.skipif(
os.getenv("IPA_SERVER_HOST") is None,
reason="Environment variable IPA_SERVER_HOST must be set",
)
@prepare_test(test_name, test_path)
def method(self, test_path):
result = run_playbook(test_path)
assert result.returncode == 0
_tests[test_name] = method
globals()[group_name] = type(group_name, (TestCase,), _tests)