One-KVM/testenv/tests/validators/test_net.py

# ========================================================================== #
#                                                                            #
#    KVMD - The main PiKVM daemon.                                           #
#                                                                            #
#    Copyright (C) 2018-2024  Maxim Devaev <mdevaev@gmail.com>               #
#                                                                            #
#    This program is free software: you can redistribute it and/or modify    #
#    it under the terms of the GNU General Public License as published by    #
#    the Free Software Foundation, either version 3 of the License, or       #
#    (at your option) any later version.                                     #
#                                                                            #
#    This program is distributed in the hope that it will be useful,         #
#    but WITHOUT ANY WARRANTY; without even the implied warranty of          #
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           #
#    GNU General Public License for more details.                            #
#                                                                            #
#    You should have received a copy of the GNU General Public License       #
#    along with this program.  If not, see <https://www.gnu.org/licenses/>.  #
#                                                                            #
# ========================================================================== #


from typing import Any

import pytest

from kvmd.validators import ValidatorError
from kvmd.validators.net import valid_ip_or_host
from kvmd.validators.net import valid_ip
from kvmd.validators.net import valid_net
from kvmd.validators.net import valid_rfc_host
from kvmd.validators.net import valid_port
from kvmd.validators.net import valid_ports_list
from kvmd.validators.net import valid_mac
from kvmd.validators.net import valid_ssl_ciphers


# =====
@pytest.mark.parametrize("arg", [
    "yandex.ru ",
    "foobar",
    "foo-bar.ru",
    "127.0.0.1",
    "8.8.8.8",
    "::",
    "::1",
    "2001:500:2f::f",
])
def test_ok__valid_ip_or_host(arg: Any) -> None:
    assert valid_ip_or_host(arg) == arg.strip()


@pytest.mark.parametrize("arg", [
    "foo_bar.ru",
    "1.1.1.",
    ":",
    "",
    None,
])
def test_fail__valid_ip_or_host(arg: Any) -> None:
    with pytest.raises(ValidatorError):
        print(valid_ip_or_host(arg))


# =====
@pytest.mark.parametrize("arg", [
    "127.0.0.1 ",
    "8.8.8.8",
    "::",
    "::1",
    "2001:500:2f::f",
])
def test_ok__valid_ip(arg: Any) -> None:
    assert valid_ip(arg) == arg.strip()


@pytest.mark.parametrize("arg", [
    "ya.ru",
    "1",
    "1.1.1",
    "1.1.1.",
    ":",
    "",
    None,
])
def test_fail__valid_ip(arg: Any) -> None:
    with pytest.raises(ValidatorError):
        print(valid_ip(arg))


# =====
@pytest.mark.parametrize("arg", [
    "127.0.0.0/24 ",
    "8.8.8.8/31",
    "::/16",
    "::ffff:0:0:0/96",
    "64:ff9b::/96",
])
def test_ok__valid_net(arg: Any) -> None:
    assert valid_net(arg) == arg.strip()


@pytest.mark.parametrize("arg", [
    "127.0.0.1/33",
    "127.0.0.1/0",
    "127.0.0.1/",
    "127.0.0.1",
    "8.8.8.8//31",
    "ya.ru",
    "1",
    "1.1.1",
    "1.1.1.",
    ":",
    "",
    None,
])
def test_fail__valid_net(arg: Any) -> None:
    with pytest.raises(ValidatorError):
        print(valid_net(arg))


# =====
@pytest.mark.parametrize("arg", [
    "yandex.ru ",
    "foobar",
    "foo-bar.ru",
    "z0r.de",
    "11.ru",
    "127.0.0.1",
])
def test_ok__valid_rfc_host(arg: Any) -> None:
    assert valid_rfc_host(arg) == arg.strip()


@pytest.mark.parametrize("arg", [
    "foobar.ru.",
    "foo_bar.ru",
    "",
    None,
])
def test_fail__valid_rfc_host(arg: Any) -> None:
    with pytest.raises(ValidatorError):
        print(valid_rfc_host(arg))


# =====
@pytest.mark.parametrize("arg", ["0 ", 0, "22", 443, 65535])
def test_ok__valid_port(arg: Any) -> None:
    value = valid_port(arg)
    assert type(value) is int  # pylint: disable=unidiomatic-typecheck
    assert value == int(str(arg).strip())


@pytest.mark.parametrize("arg", ["test", "", None, 1.1])
def test_fail__valid_port(arg: Any) -> None:
    with pytest.raises(ValidatorError):
        print(valid_port(arg))


# =====
@pytest.mark.parametrize("arg, retval", [
    ("",            []),
    (",, , ",       []),
    ("0 ",          [0]),
    ("1,",          [1]),
    ("22,23",       [22, 23]),
    ("80,443,443,", [80, 443, 443]),
    (65535,         [65535]),
])
def test_ok__valid_ports_list(arg: Any, retval: list[int]) -> None:
    assert valid_ports_list(arg) == retval


@pytest.mark.parametrize("arg", ["test", "13,test", None, 1.1])
def test_fail__valid_ports_list(arg: Any) -> None:
    with pytest.raises(ValidatorError):
        print(valid_ports_list(arg))


# =====
@pytest.mark.parametrize("arg", [
    " 00:00:00:00:00:00 ",
    " 9f:00:00:00:00:00 ",
    " FF:FF:FF:FF:FF:FF ",
])
def test_ok__valid_mac(arg: Any) -> None:
    assert valid_mac(arg) == arg.strip().lower()


@pytest.mark.parametrize("arg", [
    "00:00:00:00:00:0",
    "9x:00:00:00:00:00",
    "",
    None,
])
def test_fail__valid_mac(arg: Any) -> None:
    with pytest.raises(ValidatorError):
        print(valid_mac(arg))


# =====
@pytest.mark.parametrize("arg", ["ALL", " ALL:@SECLEVEL=0 "])
def test_ok__valid_ssl_ciphers(arg: Any) -> None:
    assert valid_ssl_ciphers(arg) == str(arg).strip()


@pytest.mark.parametrize("arg", ["test", "all", "", None])
def test_fail__valid_ssl_ciphers(arg: Any) -> None:
    with pytest.raises(ValidatorError):
        print(valid_ssl_ciphers(arg))