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

51 lines
1.3 KiB
Python

"""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(mocker, testinput, expected):
host = Host(
testinput["vmid"],
testinput["hostname"],
testinput["ipv4_address"],
testinput["ipv6_address"],
testinput["pve_type"],
)
assert host.labels == expected