"""Global pytest fixtures.""" import environs import pytest from prometheuspvesd.model import Host, 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", }, }, ]