mirror of
https://github.com/thegeeklab/prometheus-pve-sd.git
synced 2024-11-23 01:40:40 +00:00
51 lines
1.3 KiB
Python
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
|