prometheus-pve-sd/prometheuspvesd/test/fixtures/fixtures.py

"""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",
            },
        },
    ]