mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-03-22 21:29:19 +00:00
71f8c15d2e
Allowing the domain suffix to be appended independent of the `host_fqdn` setting enables the inventory plugin to construct proper FQDNs if a network has the `dns.domain` property set. Otherwise you would always end up with something like `host01.project.local.example.net` despite `host01.example.net` being the expected result.
183 lines
5.6 KiB
Python
183 lines
5.6 KiB
Python
# Copyright (c) 2025 Stéphane Graber <stgraber@stgraber.org>
|
|
# 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
|
|
from ansible.inventory.data import InventoryData
|
|
from ansible.parsing.dataloader import DataLoader
|
|
from ansible.template import Templar
|
|
|
|
from ansible_collections.community.general.plugins.inventory.incus import (
|
|
InventoryModule,
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def inventory():
|
|
plugin = InventoryModule()
|
|
plugin.inventory = InventoryData()
|
|
plugin.templar = Templar(loader=DataLoader())
|
|
return plugin
|
|
|
|
|
|
def test_verify_file_yml(tmp_path, inventory):
|
|
file = tmp_path / "foobar.incus.yml"
|
|
file.touch()
|
|
assert inventory.verify_file(str(file)) is True
|
|
|
|
|
|
def test_verify_file_yaml(tmp_path, inventory):
|
|
file = tmp_path / "foobar.incus.yaml"
|
|
file.touch()
|
|
assert inventory.verify_file(str(file)) is True
|
|
|
|
|
|
def test_verify_file_bad_config_yml(inventory):
|
|
assert inventory.verify_file("foobar.incus.yml") is False
|
|
|
|
|
|
def test_verify_file_bad_config_yaml(inventory):
|
|
assert inventory.verify_file("foobar.incus.yaml") is False
|
|
|
|
|
|
def test_verify_file_bad_config(inventory):
|
|
assert inventory.verify_file("foobar.wrongcloud.yml") is False
|
|
|
|
|
|
def _build_get_option(options):
|
|
def _get_option(option):
|
|
return options.get(option, False)
|
|
|
|
return _get_option
|
|
|
|
|
|
def _make_host(name):
|
|
entry = {}
|
|
entry["name"] = name
|
|
|
|
for prop in (
|
|
"architecture",
|
|
"config",
|
|
"description",
|
|
"devices",
|
|
"ephemeral",
|
|
"expanded_config",
|
|
"expanded_devices",
|
|
"location",
|
|
"profiles",
|
|
"status",
|
|
"type",
|
|
):
|
|
entry[prop] = ""
|
|
|
|
return entry
|
|
|
|
|
|
def run_incus(*args):
|
|
if args == ("project", "list", "r1:"):
|
|
return [{"name": "default"}]
|
|
|
|
if args == ("project", "list", "r2:"):
|
|
return [{"name": "foo"}]
|
|
|
|
if args == ("list", "r1:", "--project", "default", "status=running"):
|
|
return [_make_host("c1")]
|
|
|
|
if args == ("list", "r2:", "--project", "foo", "status=running"):
|
|
return [_make_host("c2")]
|
|
|
|
if args == ("list", "r3:", "--project", "proj1", "status=running"):
|
|
return [_make_host("c3")]
|
|
|
|
if args == ("list", "r3:", "--project", "proj2", "status=running"):
|
|
return [_make_host("c4"), _make_host("c5")]
|
|
|
|
return []
|
|
|
|
|
|
def test_build_inventory(inventory, mocker):
|
|
get_option = _build_get_option(
|
|
{
|
|
"default_groups": True,
|
|
"remotes": ["r1", "r2", "r3:proj1", "r3:proj2"],
|
|
"filters": ["status=running"],
|
|
"host_fqdn": True,
|
|
"host_domain": "example.net",
|
|
},
|
|
)
|
|
inventory.get_option = mocker.MagicMock(side_effect=get_option)
|
|
inventory._run_incus = mocker.MagicMock(side_effect=run_incus)
|
|
inventory.populate()
|
|
|
|
c1 = inventory.inventory.get_host("c1.default.r1.example.net")
|
|
assert c1
|
|
assert "ansible_incus_status" in c1.get_vars()
|
|
|
|
c2 = inventory.inventory.get_host("c2.foo.r2.example.net")
|
|
assert c2
|
|
assert "ansible_incus_status" in c2.get_vars()
|
|
|
|
c3 = inventory.inventory.get_host("c3.proj1.r3.example.net")
|
|
assert c3
|
|
assert "ansible_incus_status" in c3.get_vars()
|
|
|
|
c4 = inventory.inventory.get_host("c4.proj2.r3.example.net")
|
|
assert c4
|
|
assert "ansible_incus_status" in c4.get_vars()
|
|
|
|
c5 = inventory.inventory.get_host("c5.proj2.r3.example.net")
|
|
assert c5
|
|
assert "ansible_incus_status" in c5.get_vars()
|
|
|
|
assert len(inventory.inventory.groups["all"].hosts) == 5
|
|
assert len(inventory.inventory.groups["incus"].child_groups) == 3
|
|
assert len(inventory.inventory.groups["incus_r1"].child_groups) == 1
|
|
assert len(inventory.inventory.groups["incus_r2"].child_groups) == 1
|
|
assert len(inventory.inventory.groups["incus_r3"].child_groups) == 2
|
|
assert len(inventory.inventory.groups["incus_r3_proj1"].hosts) == 1
|
|
assert len(inventory.inventory.groups["incus_r3_proj2"].hosts) == 2
|
|
|
|
|
|
def test_build_inventory_fqdn_from_host_domain(inventory, mocker):
|
|
get_option = _build_get_option(
|
|
{
|
|
"default_groups": True,
|
|
"remotes": ["r1", "r2", "r3:proj1", "r3:proj2"],
|
|
"filters": ["status=running"],
|
|
"host_fqdn": False,
|
|
"host_domain": "example.net",
|
|
},
|
|
)
|
|
inventory.get_option = mocker.MagicMock(side_effect=get_option)
|
|
inventory._run_incus = mocker.MagicMock(side_effect=run_incus)
|
|
inventory.populate()
|
|
|
|
c1 = inventory.inventory.get_host("c1.example.net")
|
|
assert c1
|
|
assert "ansible_incus_status" in c1.get_vars()
|
|
|
|
c2 = inventory.inventory.get_host("c2.example.net")
|
|
assert c2
|
|
assert "ansible_incus_status" in c2.get_vars()
|
|
|
|
c3 = inventory.inventory.get_host("c3.example.net")
|
|
assert c3
|
|
assert "ansible_incus_status" in c3.get_vars()
|
|
|
|
c4 = inventory.inventory.get_host("c4.example.net")
|
|
assert c4
|
|
assert "ansible_incus_status" in c4.get_vars()
|
|
|
|
c5 = inventory.inventory.get_host("c5.example.net")
|
|
assert c5
|
|
assert "ansible_incus_status" in c5.get_vars()
|
|
|
|
assert len(inventory.inventory.groups["all"].hosts) == 5
|
|
assert len(inventory.inventory.groups["incus"].child_groups) == 3
|
|
assert len(inventory.inventory.groups["incus_r1"].child_groups) == 1
|
|
assert len(inventory.inventory.groups["incus_r2"].child_groups) == 1
|
|
assert len(inventory.inventory.groups["incus_r3"].child_groups) == 2
|
|
assert len(inventory.inventory.groups["incus_r3_proj1"].hosts) == 1
|
|
assert len(inventory.inventory.groups["incus_r3_proj2"].hosts) == 2
|