validators, tests

This commit is contained in:
Devaev Maxim
2019-04-06 05:32:02 +03:00
parent 73e04b71ed
commit 1d75b738a0
44 changed files with 1616 additions and 311 deletions

59
tests/__init__.py Normal file
View File

@@ -0,0 +1,59 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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/>. #
# #
# ========================================================================== #
import sys
from typing import Dict
from typing import Optional
import fake_rpi.RPi
# =====
class _GPIO(fake_rpi.RPi._GPIO): # pylint: disable=protected-access
def __init__(self) -> None:
super().__init__()
self.__states: Dict[int, int] = {}
@fake_rpi.RPi.printf
def setup(self, channel: int, state: int, initial: int=0, pull_up_down: Optional[int]=None) -> None:
_ = state # Makes linter happy
_ = pull_up_down # Makes linter happy
self.__states[int(channel)] = int(initial)
@fake_rpi.RPi.printf
def output(self, channel: int, state: int) -> None:
self.__states[int(channel)] = int(state)
@fake_rpi.RPi.printf
def input(self, channel: int) -> int: # pylint: disable=arguments-differ
return self.__states[int(channel)]
@fake_rpi.RPi.printf
def cleanup(self, channel: Optional[int]=None) -> None: # pylint: disable=arguments-differ
_ = channel # Makes linter happy
self.__states = {}
# =====
fake_rpi.RPi.GPIO = _GPIO()
sys.modules["RPi"] = fake_rpi.RPi

115
tests/test_aioregion.py Normal file
View File

@@ -0,0 +1,115 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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/>. #
# #
# ========================================================================== #
import asyncio
import pytest
from kvmd.aioregion import RegionIsBusyError
from kvmd.aioregion import AioExclusiveRegion
# =====
@pytest.mark.asyncio
async def test_aioregion__one__ok(event_loop: asyncio.AbstractEventLoop) -> None:
_ = event_loop
region = AioExclusiveRegion(RegionIsBusyError)
async def func() -> None:
assert not region.is_busy()
with region:
assert region.is_busy()
assert not region.is_busy()
await func()
assert not region.is_busy()
region.exit()
assert not region.is_busy()
@pytest.mark.asyncio
async def test_aioregion__one__fail(event_loop: asyncio.AbstractEventLoop) -> None:
_ = event_loop
region = AioExclusiveRegion(RegionIsBusyError)
async def func() -> None:
assert not region.is_busy()
with region:
assert region.is_busy()
region.enter()
assert not region.is_busy()
with pytest.raises(RegionIsBusyError):
await func()
assert not region.is_busy()
region.exit()
assert not region.is_busy()
# =====
@pytest.mark.asyncio
async def test_aioregion__two__ok(event_loop: asyncio.AbstractEventLoop) -> None:
region = AioExclusiveRegion(RegionIsBusyError)
async def func1() -> None:
with region:
await asyncio.sleep(1, loop=event_loop)
print("done func1()")
async def func2() -> None:
await asyncio.sleep(2)
print("waiking up func2()")
with region:
await asyncio.sleep(1, loop=event_loop)
print("done func2()")
await asyncio.gather(func1(), func2())
assert not region.is_busy()
region.exit()
assert not region.is_busy()
@pytest.mark.asyncio
async def test_aioregion__two__fail(event_loop: asyncio.AbstractEventLoop) -> None:
region = AioExclusiveRegion(RegionIsBusyError)
async def func1() -> None:
with region:
await asyncio.sleep(2, loop=event_loop)
print("done func1()")
async def func2() -> None:
await asyncio.sleep(1)
with region:
await asyncio.sleep(1, loop=event_loop)
print("done func2()")
results = await asyncio.gather(func1(), func2(), loop=event_loop, return_exceptions=True)
assert results[0] is None
assert type(results[1]) == RegionIsBusyError # pylint: disable=unidiomatic-typecheck
assert not region.is_busy()
region.exit()
assert not region.is_busy()

34
tests/test_app_cleanup.py Normal file
View File

@@ -0,0 +1,34 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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 kvmd.apps.cleanup import main
# =====
def test_main() -> None:
open("/tmp/foobar.sock", "w").close()
main([
"kvmd-cleanup",
"--set-options",
"kvmd/hid/device=/dev/null",
"kvmd/streamer/unix=/tmp/foobar.sock",
])

49
tests/test_gpio.py Normal file
View File

@@ -0,0 +1,49 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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 kvmd import gpio
# =====
def test_gpio__loopback_initial_false() -> None:
# pylint: disable=singleton-comparison
with gpio.bcm():
assert gpio.set_output(0) == 0
assert gpio.read(0) == False # noqa: E712
gpio.write(0, True)
assert gpio.read(0) == True # noqa: E712
def test_gpio__loopback_initial_true() -> None:
# pylint: disable=singleton-comparison
with gpio.bcm():
assert gpio.set_output(0, True) == 0
assert gpio.read(0) == True # noqa: E712
gpio.write(0, False)
assert gpio.read(0) == False # noqa: E712
def test_gpio__input() -> None:
# pylint: disable=singleton-comparison
with gpio.bcm():
assert gpio.set_input(0) == 0
assert gpio.read(0) == False # noqa: E712

35
tests/test_logging.py Normal file
View File

@@ -0,0 +1,35 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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/>. #
# #
# ========================================================================== #
import pytest
from kvmd.logging import get_logger
# =====
@pytest.mark.parametrize("depth, name", [
(0, "tests.test_logging"),
(1, "_pytest.python"),
(2, "pluggy.callers"),
])
def test_get_logger(depth: int, name: str) -> None:
assert get_logger(depth).name == name

View File

@@ -0,0 +1,119 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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.auth import valid_user
from kvmd.validators.auth import valid_passwd
from kvmd.validators.auth import valid_auth_token
from kvmd.validators.auth import valid_auth_type
# =====
@pytest.mark.parametrize("arg", [
"test-",
"glados",
"test",
"_",
"_foo_bar_",
" aix",
])
def test_ok__valid_user(arg: Any) -> None:
assert valid_user(arg) == arg.strip()
@pytest.mark.parametrize("arg", [
"тест",
"-molestia",
"te~st",
"-",
"-foo_bar",
" ",
"",
None,
])
def test_fail__valid_user(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_user(arg))
# =====
@pytest.mark.parametrize("arg", [
"glados",
"test",
"_",
"_foo_bar_",
" aix",
" ",
"",
" O(*#&@)FD*S)D(F ",
])
def test_ok__valid_passwd(arg: Any) -> None:
assert valid_passwd(arg) == arg
@pytest.mark.parametrize("arg", [
"тест",
"\n",
" \n",
"\n\n",
"\r",
None,
])
def test_fail__valid_passwd(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_passwd(arg))
# =====
@pytest.mark.parametrize("arg", [
("0" * 64) + " ",
("f" * 64) + " ",
])
def test_ok__valid_auth_token(arg: Any) -> None:
assert valid_auth_token(arg) == arg.strip()
@pytest.mark.parametrize("arg", [
("F" * 64),
"0" * 63,
"0" * 65,
"",
None,
])
def test_fail__valid_auth_token(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_auth_token(arg))
@pytest.mark.parametrize("arg", ["BASIC ", "basic"])
def test_ok__valid_auth_type(arg: Any) -> None:
assert valid_auth_type(arg) == arg.strip().lower()
@pytest.mark.parametrize("arg", ["test", "", None])
def test_fail__valid_auth_type(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_auth_type(arg))

View File

@@ -0,0 +1,107 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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.basic import valid_bool
from kvmd.validators.basic import valid_number
from kvmd.validators.basic import valid_int_f1
from kvmd.validators.basic import valid_float_f01
# =====
@pytest.mark.parametrize("arg, retval", [
("1", True),
("true", True),
("TRUE", True),
("yes ", True),
(1, True),
(True, True),
("0", False),
("false", False),
("FALSE", False),
("no ", False),
(0, False),
(False, False),
])
def test_ok__valid_bool(arg: Any, retval: bool) -> None:
assert valid_bool(arg) == retval
@pytest.mark.parametrize("arg", ["test", "", None, -1, "x"])
def test_fail__valid_bool(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_bool(arg))
# =====
@pytest.mark.parametrize("arg", ["1 ", "-1", 1, -1, 0, 100500])
def test_ok__valid_number(arg: Any) -> None:
assert valid_number(arg) == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, "1x", 100500.0])
def test_fail__valid_number(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_number(arg))
@pytest.mark.parametrize("arg", [-5, 0, 5, "-5 ", "0 ", "5 "])
def test_ok__valid_number__min_max(arg: Any) -> None:
assert valid_number(arg, -5, 5) == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, -6, 6, "-6 ", "6 "])
def test_fail__valid_number__min_max(arg: Any) -> None: # pylint: disable=invalid-name
with pytest.raises(ValidatorError):
print(valid_number(arg, -5, 5))
# =====
@pytest.mark.parametrize("arg", [1, 5, "5 "])
def test_ok__valid_int_f1(arg: Any) -> None:
value = valid_int_f1(arg)
assert type(value) == int # pylint: disable=unidiomatic-typecheck
assert value == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, -6, "-6 ", 0, "0 ", "5.0"])
def test_fail__valid_int_f1(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_int_f1(arg))
# =====
@pytest.mark.parametrize("arg", [0.1, 1, 5, "5 ", "5.0 "])
def test_ok__valid_float_f01(arg: Any) -> None:
value = valid_float_f01(arg)
assert type(value) == float # pylint: disable=unidiomatic-typecheck
assert value == float(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, 0.0, "0.0", -6, "-6", 0, "0"])
def test_fail__valid_float_f01(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_float_f01(arg))

View File

@@ -0,0 +1,91 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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/>. #
# #
# ========================================================================== #
import os
from typing import Any
import pytest
from kvmd.validators import ValidatorError
from kvmd.validators.fs import valid_abs_path
from kvmd.validators.fs import valid_abs_path_exists
from kvmd.validators.fs import valid_unix_mode
# =====
@pytest.mark.parametrize("arg, retval", [
("/..", "/"),
("/root/..", "/"),
("/root", "/root"),
("/f/o/o/b/a/r", "/f/o/o/b/a/r"),
("~", os.path.abspath(".") + "/~"),
("/foo~", "/foo~"),
("/foo/~", "/foo/~"),
(".", os.path.abspath(".")),
])
def test_ok__valid_abs_path(arg: Any, retval: str) -> None:
assert valid_abs_path(arg) == retval
@pytest.mark.parametrize("arg", ["", " ", None])
def test_fail__valid_abs_path(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_abs_path(arg))
# =====
@pytest.mark.parametrize("arg, retval", [
("/..", "/"),
("/root/..", "/"),
("/root", "/root"),
(".", os.path.abspath(".")),
])
def test_ok__valid_abs_path_exists(arg: Any, retval: str) -> None:
assert valid_abs_path_exists(arg) == retval
@pytest.mark.parametrize("arg", [
"/f/o/o/b/a/r",
"~",
"/foo~",
"/foo/~",
"",
None,
])
def test_fail__valid_abs_path_exists(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_abs_path_exists(arg))
# =====
@pytest.mark.parametrize("arg", [0, 5, "1000"])
def test_ok__valid_unix_mode(arg: Any) -> None:
value = valid_unix_mode(arg)
assert type(value) == int # pylint: disable=unidiomatic-typecheck
assert value == int(str(value).strip())
@pytest.mark.parametrize("arg", ["test", "", None, -6, "-6", "5.0"])
def test_fail__valid_unix_mode(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_unix_mode(arg))

View File

@@ -0,0 +1,72 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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.hw import valid_tty_speed
from kvmd.validators.hw import valid_gpio_pin
from kvmd.validators.hw import valid_gpio_pin_optional
# =====
@pytest.mark.parametrize("arg", ["1200 ", 1200, 2400, 4800, 9600, 19200, 38400, 57600, 115200])
def test_ok__valid_tty_speed(arg: Any) -> None:
value = valid_tty_speed(arg)
assert type(value) == int # pylint: disable=unidiomatic-typecheck
assert value == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, 0, 1200.1])
def test_fail__valid_tty_speed(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_tty_speed(arg))
# =====
@pytest.mark.parametrize("arg", ["0 ", 0, 1, 13])
def test_ok__valid_gpio_pin(arg: Any) -> None:
value = valid_gpio_pin(arg)
assert type(value) == int # pylint: disable=unidiomatic-typecheck
assert value == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, -1, -13, 1.1])
def test_fail__valid_gpio_pin(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_gpio_pin(arg))
# =====
@pytest.mark.parametrize("arg", ["0 ", -1, 0, 1, 13])
def test_ok__valid_gpio_pin_optional(arg: Any) -> None:
value = valid_gpio_pin_optional(arg)
assert type(value) == int # pylint: disable=unidiomatic-typecheck
assert value == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, -2, -13, 1.1])
def test_fail__valid_gpio_pin_optional(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_gpio_pin_optional(arg))

View File

@@ -0,0 +1,98 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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.kvm import valid_atx_button
from kvmd.validators.kvm import valid_kvm_target
from kvmd.validators.kvm import valid_log_seek
from kvmd.validators.kvm import valid_stream_quality
from kvmd.validators.kvm import valid_stream_fps
# =====
@pytest.mark.parametrize("arg", ["POWER ", "POWER_LONG ", "RESET "])
def test_ok__valid_atx_button(arg: Any) -> None:
assert valid_atx_button(arg) == arg.strip().lower()
@pytest.mark.parametrize("arg", ["test", "", None])
def test_fail__valid_atx_button(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_atx_button(arg))
# =====
@pytest.mark.parametrize("arg", ["KVM ", "SERVER "])
def test_ok__valid_kvm_target(arg: Any) -> None:
assert valid_kvm_target(arg) == arg.strip().lower()
@pytest.mark.parametrize("arg", ["test", "", None])
def test_fail__valid_kvm_target(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_kvm_target(arg))
# =====
@pytest.mark.parametrize("arg", ["0 ", 0, 1, 13])
def test_ok__valid_log_seek(arg: Any) -> None:
value = valid_log_seek(arg)
assert type(value) == int # pylint: disable=unidiomatic-typecheck
assert value == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, -1, -13, 1.1])
def test_fail__valid_log_seek(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_log_seek(arg))
# =====
@pytest.mark.parametrize("arg", ["1 ", 20, 100])
def test_ok__valid_stream_quality(arg: Any) -> None:
value = valid_stream_quality(arg)
assert type(value) == int # pylint: disable=unidiomatic-typecheck
assert value == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, 0, 101, 1.1])
def test_fail__valid_stream_quality(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_stream_quality(arg))
# =====
@pytest.mark.parametrize("arg", ["1 ", 30])
def test_ok__valid_stream_fps(arg: Any) -> None:
value = valid_stream_fps(arg)
assert type(value) == int # pylint: disable=unidiomatic-typecheck
assert value == int(str(arg).strip())
@pytest.mark.parametrize("arg", ["test", "", None, 31, 1.1])
def test_fail__valid_stream_fps(arg: Any) -> None:
with pytest.raises(ValidatorError):
print(valid_stream_fps(arg))

View File

@@ -0,0 +1,122 @@
# ========================================================================== #
# #
# KVMD - The main Pi-KVM daemon. #
# #
# Copyright (C) 2018 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_rfc_host
from kvmd.validators.net import valid_port
# =====
@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", [
"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) == 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))