mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-02-04 07:51:50 +00:00
58bb1e7c04
* fix ruff case B007 * rollback inventory/iocage * re-do the fix in inventory/iocage * add cases in tests/unit/plugins * rollback plugins/module_utils/memset.py * rollback extraneous changes in plugins/modules/xcc_redfish_command.py * add changelog frag
132 lines
4.6 KiB
Python
132 lines
4.6 KiB
Python
# Copyright (c) 2024 Vladimir Botka <vbotka@gmail.com>
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
import yaml
|
|
|
|
from ansible.inventory.data import InventoryData
|
|
from ansible.template import Templar
|
|
from ansible_collections.community.general.plugins.inventory.iocage import InventoryModule
|
|
from ansible_collections.community.internal_test_tools.tests.unit.utils.trust import make_trusted
|
|
|
|
|
|
@pytest.fixture
|
|
def inventory():
|
|
inv = InventoryModule()
|
|
inv.inventory = InventoryData()
|
|
inv.templar = Templar(None)
|
|
inv.jails = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails.txt")
|
|
inv.js_ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails.yml")
|
|
inv.jails_dhcp = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp.txt")
|
|
inv.js_dhcp_ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp.yml")
|
|
inv.jails_dhcp_nr = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp_not_running.txt")
|
|
inv.js_dhcp_nr_ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_jails_dhcp_not_running.yml")
|
|
prpts_101 = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_101.txt")
|
|
prpts_102 = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_102.txt")
|
|
prpts_103 = load_txt_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_properties_test_103.txt")
|
|
inv.prpts = {"test_101": prpts_101, "test_102": prpts_102, "test_103": prpts_103}
|
|
inv.ps_ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_properties.yml")
|
|
inv.ok = load_yml_data("tests/unit/plugins/inventory/fixtures/iocage/iocage_inventory.yml")
|
|
return inv
|
|
|
|
|
|
def load_txt_data(path):
|
|
with open(path, "r") as f:
|
|
s = f.read()
|
|
return s
|
|
|
|
|
|
def load_yml_data(path):
|
|
with open(path, "r") as f:
|
|
d = yaml.safe_load(f)
|
|
return d
|
|
|
|
|
|
def get_option(option):
|
|
groups = {}
|
|
groups["test"] = make_trusted("inventory_hostname.startswith('test')")
|
|
|
|
if option == "groups":
|
|
return groups
|
|
elif option == "keyed_groups":
|
|
return []
|
|
elif option == "compose":
|
|
return {}
|
|
elif option == "strict":
|
|
return False
|
|
else:
|
|
return None
|
|
|
|
|
|
def test_verify_file_bad_config(inventory):
|
|
assert inventory.verify_file("foobar.iocage.yml") is False
|
|
|
|
|
|
def test_verify_file(tmp_path, inventory):
|
|
file = tmp_path / "foobar.iocage.yml"
|
|
file.touch()
|
|
assert inventory.verify_file(str(file))
|
|
|
|
|
|
def test_get_jails(inventory):
|
|
# jails
|
|
results = {"_meta": {"hostvars": {}}}
|
|
inventory.get_jails(inventory.jails, results)
|
|
assert results == inventory.js_ok
|
|
|
|
# jails_dhcp
|
|
results = {"_meta": {"hostvars": {}}}
|
|
inventory.get_jails(inventory.jails_dhcp, results)
|
|
assert results == inventory.js_dhcp_ok
|
|
|
|
# jails_dhcp_not_running
|
|
results = {"_meta": {"hostvars": {}}}
|
|
inventory.get_jails(inventory.jails_dhcp_nr, results)
|
|
assert results == inventory.js_dhcp_nr_ok
|
|
|
|
|
|
def test_get_properties(inventory):
|
|
results = {"_meta": {"hostvars": {}}}
|
|
inventory.get_jails(inventory.jails, results)
|
|
for hostname in results["_meta"]["hostvars"]:
|
|
inventory.get_properties(inventory.prpts[hostname], results, hostname)
|
|
assert results == inventory.ps_ok
|
|
|
|
|
|
def test_populate(inventory, mocker):
|
|
results = {"_meta": {"hostvars": {}}}
|
|
inventory.get_jails(inventory.jails, results)
|
|
for hostname in results["_meta"]["hostvars"]:
|
|
inventory.get_properties(inventory.prpts[hostname], results, hostname)
|
|
inventory.get_option = mocker.MagicMock(side_effect=get_option)
|
|
inventory.populate(results)
|
|
|
|
# test
|
|
hosts = ("test_101", "test_102", "test_103")
|
|
vars = (
|
|
"iocage_basejail",
|
|
"iocage_boot",
|
|
"iocage_ip4",
|
|
"iocage_ip6",
|
|
"iocage_properties",
|
|
"iocage_release",
|
|
"iocage_state",
|
|
"iocage_template",
|
|
"iocage_type",
|
|
)
|
|
|
|
# test host_vars
|
|
for host in hosts:
|
|
h = inventory.inventory.get_host(host)
|
|
for var in vars:
|
|
assert inventory.ok["all"]["children"]["test"]["hosts"][host][var] == h.get_vars()[var]
|
|
|
|
# test groups
|
|
test_101_info = inventory.inventory.get_host("test_101")
|
|
test_102_info = inventory.inventory.get_host("test_102")
|
|
test_103_info = inventory.inventory.get_host("test_103")
|
|
g = inventory.inventory.groups["test"]
|
|
assert g.hosts == [test_101_info, test_102_info, test_103_info]
|