opnsense-prom-exporter/tests/test_opnsense_api.py

import responses

from opnsense_exporter.opnsense_api import OPNSenseAPI, OPNSenseRole

from .common import (
    BACKUP_HOST,
    LOGIN,
    MAIN_HOST,
    PASSWORD,
    generate_diagnostics_traffic_interface_paylaod,
    generate_get_vip_status_paylaod,
)


@responses.activate
def test_get_interface_vip_status_active():
    responses.add(
        responses.GET,
        f"https://{MAIN_HOST}/api/diagnostics/interface/get_vip_status/",
        body=generate_get_vip_status_paylaod("MASTER", "MASTER", False),
    )

    assert (
        OPNSenseAPI(
            OPNSenseRole.MAIN, MAIN_HOST, LOGIN, PASSWORD
        ).get_interface_vip_status()
        == "active"
    )


@responses.activate
def test_get_interface_vip_status_backup():
    responses.add(
        responses.GET,
        f"https://{MAIN_HOST}/api/diagnostics/interface/get_vip_status/",
        body=generate_get_vip_status_paylaod("BACKUP", "BACKUP", False),
    )

    assert (
        OPNSenseAPI(
            OPNSenseRole.MAIN, MAIN_HOST, LOGIN, PASSWORD
        ).get_interface_vip_status()
        == "hot_standby"
    )


@responses.activate
def test_get_interface_vip_status_mainteance_mode():
    responses.add(
        responses.GET,
        f"https://{MAIN_HOST}/api/diagnostics/interface/get_vip_status/",
        body=generate_get_vip_status_paylaod("MASTER", "MASTER", True),
    )

    assert (
        OPNSenseAPI(
            OPNSenseRole.MAIN, MAIN_HOST, LOGIN, PASSWORD
        ).get_interface_vip_status()
        == "maintenancemode"
    )


@responses.activate
def test_get_interface_vip_status_unavailable_weird_case():
    responses.add(
        responses.GET,
        f"https://{MAIN_HOST}/api/diagnostics/interface/get_vip_status/",
        body=generate_get_vip_status_paylaod("MASTER", "BACKUP", False),
    )
    assert (
        OPNSenseAPI(
            OPNSenseRole.MAIN, MAIN_HOST, LOGIN, PASSWORD
        ).get_interface_vip_status()
        == "unavailable"
    )


@responses.activate
def test_get_interface_vip_status_unavailable_rest_api_error():
    responses.add(
        responses.GET,
        f"https://{MAIN_HOST}/api/diagnostics/interface/get_vip_status/",
        json={"error": "not found"},
        status=404,
    )
    assert (
        OPNSenseAPI(
            OPNSenseRole.MAIN, MAIN_HOST, LOGIN, PASSWORD
        ).get_interface_vip_status()
        == "unavailable"
    )


@responses.activate
def test_get_wan_traffic():
    responses.add(
        responses.GET,
        f"https://{MAIN_HOST}/api/diagnostics/traffic/top/wan",
        body=generate_diagnostics_traffic_interface_paylaod(),
    )
    assert OPNSenseAPI(
        OPNSenseRole.MAIN, MAIN_HOST, LOGIN, PASSWORD
    ).get_wan_trafic() == (
        20538,
        10034,
    )


@responses.activate
def test_get_wan_traffic_none():
    responses.add(
        responses.GET,
        f"https://{MAIN_HOST}/api/diagnostics/traffic/top/wan",
        json={"error": "not found"},
        status=404,
    )
    assert OPNSenseAPI(
        OPNSenseRole.MAIN, MAIN_HOST, LOGIN, PASSWORD
    ).get_wan_trafic() == (
        None,
        None,
    )


def test_labels():
    assert OPNSenseAPI(OPNSenseRole.MAIN, MAIN_HOST, LOGIN, PASSWORD).labels == {
        "role": "main",
        "host": MAIN_HOST,
    }
    assert OPNSenseAPI(OPNSenseRole.BACKUP, BACKUP_HOST, LOGIN, PASSWORD).labels == {
        "role": "backup",
        "host": BACKUP_HOST,
    }