prometheus-pve-sd/prometheuspvesd/test/unit/test_model.py

"""Test Host class."""

import pytest

from prometheuspvesd.model import Host

pytest_plugins = [
    "prometheuspvesd.test.fixtures.fixtures",
]


@pytest.mark.parametrize(
    "testinput,expected",
    [
        (
            {
                "vmid": 101,
                "hostname": "host1",
                "ipv4_address": False,
                "ipv6_address": False,
                "pve_type": "qemu",
            },
            {
                "__meta_pve_vmid": "101",
                "__meta_pve_name": "host1",
                "__meta_pve_ipv4": "False",
                "__meta_pve_ipv6": "False",
                "__meta_pve_type": "qemu",
            },
        ),
        (
            {
                "vmid": "202",
                "hostname": "host2",
                "ipv4_address": "129.168.0.1",
                "ipv6_address": "2001:db8:3333:4444:5555:6666:7777:8888",
                "pve_type": "qemu",
            },
            {
                "__meta_pve_vmid": "202",
                "__meta_pve_name": "host2",
                "__meta_pve_ipv4": "129.168.0.1",
                "__meta_pve_ipv6": "2001:db8:3333:4444:5555:6666:7777:8888",
                "__meta_pve_type": "qemu",
            },
        ),
    ],
)
def test_host(testinput, expected):
    host = Host(
        testinput["vmid"],
        testinput["hostname"],
        testinput["ipv4_address"],
        testinput["ipv6_address"],
        testinput["pve_type"],
    )

    assert host.labels == expected