mirror of
https://github.com/thegeeklab/prometheus-pve-sd.git
synced 2024-11-25 02:40:39 +00:00
Robert Kaussow
35bfa8bc6a
Co-authored-by: Bruno MATEU <pro+github@brunomat.eu> Co-authored-by: Bruno MATEU <mateubruno@gmail.com>
414 lines
11 KiB
Python
414 lines
11 KiB
Python
"""Global pytest fixtures."""
|
|
|
|
import environs
|
|
import pytest
|
|
|
|
from prometheuspvesd.model import Host
|
|
from prometheuspvesd.model import HostList
|
|
|
|
|
|
@pytest.fixture
|
|
def builtins():
|
|
return {
|
|
"metrics.enabled": {
|
|
"default": True,
|
|
"env": "METRICS_ENABLED",
|
|
"type": environs.Env().bool
|
|
},
|
|
"metrics.address": {
|
|
"default": "127.0.0.1",
|
|
"env": "METRICS_ADDRESS",
|
|
"type": environs.Env().str
|
|
},
|
|
"metrics.port": {
|
|
"default": 8000,
|
|
"env": "METRICS_PORT",
|
|
"type": environs.Env().int
|
|
},
|
|
"config_file": {
|
|
"default": "",
|
|
"env": "CONFIG_FILE",
|
|
"type": environs.Env().str
|
|
},
|
|
"logging.level": {
|
|
"default": "WARNING",
|
|
"env": "LOG_LEVEL",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"logging.format": {
|
|
"default": "console",
|
|
"env": "LOG_FORMAT",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"output_file": {
|
|
"default": "dummy",
|
|
"env": "OUTPUT_FILE",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"output_file_mode": {
|
|
"default": "0640",
|
|
"env": "OUTPUT_FILE_MODE",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"loop_delay": {
|
|
"default": 300,
|
|
"env": "LOOP_DELAY",
|
|
"file": True,
|
|
"type": environs.Env().int
|
|
},
|
|
"service": {
|
|
"default": False,
|
|
"env": "SERVICE",
|
|
"file": True,
|
|
"type": environs.Env().bool
|
|
},
|
|
"exclude_state": {
|
|
"default": [],
|
|
"env": "EXCLUDE_STATE",
|
|
"file": True,
|
|
"type": environs.Env().list
|
|
},
|
|
"exclude_vmid": {
|
|
"default": [],
|
|
"env": "EXCLUDE_VMID",
|
|
"file": True,
|
|
"type": environs.Env().list
|
|
},
|
|
"exclude_tags": {
|
|
"default": [],
|
|
"env": "EXCLUDE_TAGS",
|
|
"file": True,
|
|
"type": environs.Env().list
|
|
},
|
|
"include_vmid": {
|
|
"default": [],
|
|
"env": "INCLUDE_VMID",
|
|
"file": True,
|
|
"type": environs.Env().list
|
|
},
|
|
"include_tags": {
|
|
"default": [],
|
|
"env": "INCLUDE_TAGS",
|
|
"file": True,
|
|
"type": environs.Env().list
|
|
},
|
|
"pve.server": {
|
|
"default": "dummyserver",
|
|
"env": "PVE_SERVER",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"pve.user": {
|
|
"default": "dummyuser",
|
|
"env": "PVE_USER",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"pve.password": {
|
|
"default": "dummypass",
|
|
"env": "PVE_PASSWORD",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"pve.token_name": {
|
|
"default": "dummyname",
|
|
"env": "PVE_TOKEN_NAME",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"pve.token_value": {
|
|
"default": "dummyvalue",
|
|
"env": "PVE_TOKEN_VALUE",
|
|
"file": True,
|
|
"type": environs.Env().str
|
|
},
|
|
"pve.auth_timeout": {
|
|
"default": 5,
|
|
"env": "PVE_AUTH_TIMEOUT",
|
|
"file": True,
|
|
"type": environs.Env().int
|
|
},
|
|
"pve.verify_ssl": {
|
|
"default": True,
|
|
"env": "PVE_VERIFY_SSL",
|
|
"file": True,
|
|
"type": environs.Env().bool
|
|
}
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def defaults():
|
|
return {
|
|
"exclude_state": [],
|
|
"exclude_tags": [],
|
|
"exclude_vmid": [],
|
|
"include_tags": [],
|
|
"include_vmid": [],
|
|
"logging": {
|
|
"format": "console",
|
|
"level": "WARNING"
|
|
},
|
|
"loop_delay": 300,
|
|
"metrics": {
|
|
"address": "127.0.0.1",
|
|
"enabled": True,
|
|
"port": 8000
|
|
},
|
|
"output_file": "dummy",
|
|
"output_file_mode": "0640",
|
|
"pve": {
|
|
"auth_timeout": 5,
|
|
"password": "",
|
|
"server": "",
|
|
"user": "",
|
|
"token_name": "",
|
|
"token_value": "",
|
|
"verify_ssl": True
|
|
},
|
|
"service": True,
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def nodes():
|
|
return [{
|
|
"level": "",
|
|
"id": "node/example-node",
|
|
"disk": 4783488,
|
|
"cpu": 0.0935113631167406,
|
|
"maxcpu": 24,
|
|
"maxmem": 142073272990,
|
|
"mem": 135884478304,
|
|
"node": "example-node",
|
|
"type": "node",
|
|
"status": "online",
|
|
"maxdisk": 504209920,
|
|
"uptime": 200
|
|
}]
|
|
|
|
|
|
@pytest.fixture
|
|
def qemus():
|
|
return [
|
|
{
|
|
"diskwrite": 0,
|
|
"vmid": "100",
|
|
"name": "100.example.com",
|
|
"cpu": 0.0202130478509556,
|
|
"diskread": 0,
|
|
"template": "",
|
|
"uptime": 3101505,
|
|
"maxdisk": 26843545600,
|
|
"maxmem": 1073741824,
|
|
"pid": "1765",
|
|
"cpus": 1,
|
|
"netin": 2856071643,
|
|
"disk": 0,
|
|
"status": "running",
|
|
"netout": 12159205236,
|
|
"mem": 496179157,
|
|
"tags": "unmonitored,excluded,postgres"
|
|
},
|
|
{
|
|
"diskwrite": 0,
|
|
"vmid": "101",
|
|
"name": "101.example.com",
|
|
"cpu": 0.0202130478509556,
|
|
"diskread": 0,
|
|
"template": "",
|
|
"uptime": 3101505,
|
|
"maxdisk": 26843545600,
|
|
"maxmem": 1073741824,
|
|
"pid": "1765",
|
|
"cpus": 1,
|
|
"netin": 2856071643,
|
|
"disk": 0,
|
|
"status": "running",
|
|
"netout": 12159205236,
|
|
"mem": 496179157
|
|
},
|
|
{
|
|
"diskwrite": 0,
|
|
"vmid": "102",
|
|
"name": "102.example.com",
|
|
"cpu": 0.0202130478509556,
|
|
"diskread": 0,
|
|
"template": "",
|
|
"uptime": 3101505,
|
|
"maxdisk": 26843545600,
|
|
"maxmem": 1073741824,
|
|
"pid": "1765",
|
|
"cpus": 1,
|
|
"netin": 2856071643,
|
|
"disk": 0,
|
|
"status": "prelaunch",
|
|
"netout": 12159205236,
|
|
"mem": 496179157,
|
|
"tags": "monitored"
|
|
},
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def instance_config():
|
|
return {
|
|
"name": "102.example.com",
|
|
"description": '{"groups": "test-group"}',
|
|
"net0": "virtio=D8-85-75-47-2E-8D,bridge=vmbr122,ip=192.0.2.25,ip=2001:db8::666:77:8888",
|
|
"cpu": 2,
|
|
"cores": 2
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def agent_info():
|
|
return {
|
|
"supported_commands": [{
|
|
"name": "guest-network-get-interfaces",
|
|
"enabled": True,
|
|
"success-response": True
|
|
}],
|
|
"version": "5.2.0"
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def addresses():
|
|
return {
|
|
"ipv4_valid": [
|
|
"192.0.2.1",
|
|
"198.51.100.1",
|
|
],
|
|
"ipv4_invalid": [
|
|
"127.0.0.1",
|
|
"169.254.1.1",
|
|
],
|
|
"ipv6_valid": [
|
|
"2001:db8:3333:4444:5555:6666:7777:8888",
|
|
"2001:db8:3333:4444:CCCC:DDDD:EEEE:FFFF",
|
|
"::",
|
|
"2001:db8::",
|
|
"::1234:5678",
|
|
"2001:db8::1234:5678",
|
|
"2001:0db8:0001:0000:0000:0ab9:C0A8:0102",
|
|
],
|
|
"ipv6_invalid": [
|
|
"::1",
|
|
"fe80::903a:1c1a:e802:11e4",
|
|
],
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def networks():
|
|
return [
|
|
{
|
|
"hardware-address": "00:00:00:00:00:00",
|
|
"ip-addresses": [
|
|
{
|
|
"ip-address": "127.0.0.1",
|
|
"ip-address-type": "ipv4",
|
|
"prefix": 8
|
|
},
|
|
{
|
|
"ip-address": "::1",
|
|
"ip-address-type": "ipv6",
|
|
"prefix": 128
|
|
},
|
|
],
|
|
"name": "lo",
|
|
"statistics": {
|
|
"rx-bytes": 9280,
|
|
"rx-dropped": 0,
|
|
"rx-errs": 0,
|
|
"rx-packets": 92,
|
|
"tx-bytes": 9280,
|
|
"tx-dropped": 0,
|
|
"tx-errs": 0,
|
|
"tx-packets": 92
|
|
}
|
|
},
|
|
{
|
|
"hardware-address": "92:0b:bd:c1:f8:39",
|
|
"ip-addresses": [
|
|
{
|
|
"ip-address": "192.0.2.1",
|
|
"ip-address-type": "ipv4",
|
|
"prefix": 32
|
|
},
|
|
{
|
|
"ip-address": "192.0.2.4",
|
|
"ip-address-type": "ipv4",
|
|
"prefix": 32
|
|
},
|
|
{
|
|
"ip-address": "2001:db8:3333:4444:5555:6666:7777:8888",
|
|
"ip-address-type": "ipv6",
|
|
"prefix": 64
|
|
},
|
|
],
|
|
"name": "eth0",
|
|
"statistics": {
|
|
"rx-bytes": 2861070337,
|
|
"rx-dropped": 0,
|
|
"rx-errs": 0,
|
|
"rx-packets": 18065580,
|
|
"tx-bytes": 12185866619,
|
|
"tx-dropped": 0,
|
|
"tx-errs": 0,
|
|
"tx-packets": 14423878
|
|
}
|
|
},
|
|
{
|
|
"hardware-address": "ba:97:85:bd:9a:a5",
|
|
"name": "eth1"
|
|
},
|
|
]
|
|
|
|
|
|
@pytest.fixture
|
|
def inventory():
|
|
hostlist = HostList()
|
|
hostlist.add_host(Host("100", "100.example.com", "192.0.2.1", False, "qemu"))
|
|
hostlist.add_host(Host("101", "101.example.com", "192.0.2.2", False, "qemu"))
|
|
hostlist.add_host(Host("102", "102.example.com", "192.0.2.3", False, "qemu"))
|
|
|
|
return hostlist
|
|
|
|
|
|
@pytest.fixture
|
|
def labels():
|
|
return [{
|
|
"targets": ["100.example.com"],
|
|
"labels": {
|
|
"__meta_pve_ipv4": "192.0.2.1",
|
|
"__meta_pve_ipv6": "False",
|
|
"__meta_pve_name": "100.example.com",
|
|
"__meta_pve_type": "qemu",
|
|
"__meta_pve_vmid": "100"
|
|
}
|
|
}, {
|
|
"targets": ["101.example.com"],
|
|
"labels": {
|
|
"__meta_pve_ipv4": "192.0.2.2",
|
|
"__meta_pve_ipv6": "False",
|
|
"__meta_pve_name": "101.example.com",
|
|
"__meta_pve_type": "qemu",
|
|
"__meta_pve_vmid": "101"
|
|
}
|
|
}, {
|
|
"targets": ["102.example.com"],
|
|
"labels": {
|
|
"__meta_pve_ipv4": "192.0.2.3",
|
|
"__meta_pve_ipv6": "False",
|
|
"__meta_pve_name": "102.example.com",
|
|
"__meta_pve_type": "qemu",
|
|
"__meta_pve_vmid": "102"
|
|
}
|
|
}]
|