mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-05-08 14:22:46 +00:00
Reformat everything.
This commit is contained in:
@@ -1,4 +1,3 @@
|
||||
|
||||
# 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
|
||||
@@ -19,51 +18,51 @@ 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')
|
||||
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:
|
||||
with open(path, "r") as f:
|
||||
s = f.read()
|
||||
return s
|
||||
|
||||
|
||||
def load_yml_data(path):
|
||||
with open(path, 'r') as f:
|
||||
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')")
|
||||
groups["test"] = make_trusted("inventory_hostname.startswith('test')")
|
||||
|
||||
if option == 'groups':
|
||||
if option == "groups":
|
||||
return groups
|
||||
elif option == 'keyed_groups':
|
||||
elif option == "keyed_groups":
|
||||
return []
|
||||
elif option == 'compose':
|
||||
elif option == "compose":
|
||||
return {}
|
||||
elif option == 'strict':
|
||||
elif option == "strict":
|
||||
return False
|
||||
else:
|
||||
return None
|
||||
|
||||
|
||||
def test_verify_file_bad_config(inventory):
|
||||
assert inventory.verify_file('foobar.iocage.yml') is False
|
||||
assert inventory.verify_file("foobar.iocage.yml") is False
|
||||
|
||||
|
||||
def test_verify_file(tmp_path, inventory):
|
||||
@@ -73,53 +72,61 @@ def test_verify_file(tmp_path, inventory):
|
||||
|
||||
|
||||
def test_get_jails(inventory):
|
||||
|
||||
# jails
|
||||
results = {'_meta': {'hostvars': {}}}
|
||||
results = {"_meta": {"hostvars": {}}}
|
||||
inventory.get_jails(inventory.jails, results)
|
||||
assert results == inventory.js_ok
|
||||
|
||||
# jails_dhcp
|
||||
results = {'_meta': {'hostvars': {}}}
|
||||
results = {"_meta": {"hostvars": {}}}
|
||||
inventory.get_jails(inventory.jails_dhcp, results)
|
||||
assert results == inventory.js_dhcp_ok
|
||||
|
||||
# jails_dhcp_not_running
|
||||
results = {'_meta': {'hostvars': {}}}
|
||||
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': {}}}
|
||||
results = {"_meta": {"hostvars": {}}}
|
||||
inventory.get_jails(inventory.jails, results)
|
||||
for hostname, host_vars in results['_meta']['hostvars'].items():
|
||||
for hostname, host_vars in results["_meta"]["hostvars"].items():
|
||||
inventory.get_properties(inventory.prpts[hostname], results, hostname)
|
||||
assert results == inventory.ps_ok
|
||||
|
||||
|
||||
def test_populate(inventory, mocker):
|
||||
results = {'_meta': {'hostvars': {}}}
|
||||
results = {"_meta": {"hostvars": {}}}
|
||||
inventory.get_jails(inventory.jails, results)
|
||||
for hostname, host_vars in results['_meta']['hostvars'].items():
|
||||
for hostname, host_vars in results["_meta"]["hostvars"].items():
|
||||
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')
|
||||
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]
|
||||
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']
|
||||
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]
|
||||
|
||||
Reference in New Issue
Block a user