mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2025-12-12 01:00:29 +08:00
160 lines
5.2 KiB
Python
160 lines
5.2 KiB
Python
# ========================================================================== #
|
|
# #
|
|
# KVMD - The main PiKVM daemon. #
|
|
# #
|
|
# Copyright (C) 2018-2022 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.os import valid_abs_path
|
|
from kvmd.validators.os import valid_printable_filename
|
|
from kvmd.validators.os import valid_unix_mode
|
|
from kvmd.validators.os import valid_command
|
|
|
|
|
|
# =====
|
|
@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__dir(arg: Any, retval: str) -> None:
|
|
assert valid_abs_path(arg, type="dir") == retval
|
|
|
|
|
|
@pytest.mark.parametrize("arg", [
|
|
"/etc/passwd",
|
|
"/etc/passwd/",
|
|
"~",
|
|
"/foo~",
|
|
"/foo/~",
|
|
"",
|
|
None,
|
|
])
|
|
def test_fail__valid_abs_path__dir(arg: Any) -> None:
|
|
with pytest.raises(ValidatorError):
|
|
print(valid_abs_path(arg, type="dir"))
|
|
|
|
|
|
# =====
|
|
@pytest.mark.parametrize("arg, retval", [
|
|
("archlinux-2018.07.01-i686.iso", "archlinux-2018.07.01-i686.iso"),
|
|
("archlinux-2018.07.01-x86_64.iso", "archlinux-2018.07.01-x86_64.iso"),
|
|
("dsl-4.11.rc1.iso", "dsl-4.11.rc1.iso"),
|
|
("systemrescuecd-x86-5.3.1.iso", "systemrescuecd-x86-5.3.1.iso"),
|
|
("ubuntu-16.04.5-desktop-i386.iso", "ubuntu-16.04.5-desktop-i386.iso"),
|
|
(" тест(){}[ \t].iso\t", "тест(){}[ _].iso"),
|
|
("\n" + "x" * 1000, "x" * 255),
|
|
("test", "test"),
|
|
("test test [test] #test$", "test test [test] #test$"),
|
|
])
|
|
def test_ok__valid_printable_filename(arg: Any, retval: str) -> None:
|
|
assert valid_printable_filename(arg) == retval
|
|
|
|
|
|
@pytest.mark.parametrize("arg", [
|
|
".",
|
|
"..",
|
|
" ..",
|
|
"test/",
|
|
"/test",
|
|
"../test",
|
|
".test",
|
|
"./.",
|
|
"../.",
|
|
"./..",
|
|
"../..",
|
|
"/ ..",
|
|
".. /",
|
|
"/.. /",
|
|
"lost+found",
|
|
"",
|
|
" ",
|
|
None,
|
|
])
|
|
def test_fail__valid_printable_filename(arg: Any) -> None:
|
|
with pytest.raises(ValidatorError):
|
|
valid_printable_filename(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))
|
|
|
|
|
|
# =====
|
|
@pytest.mark.parametrize("arg, retval", [
|
|
(["/bin/true"], ["/bin/true"]),
|
|
(["/bin/true", 1, 2, 3], ["/bin/true", "1", "2", "3"]),
|
|
("/bin/true, 1, 2, 3,", ["/bin/true", "1", "2", "3"]),
|
|
("/bin/true", ["/bin/true"]),
|
|
])
|
|
def test_ok__valid_command(arg: Any, retval: list[str]) -> None:
|
|
assert valid_command(arg) == retval
|
|
|
|
|
|
@pytest.mark.parametrize("arg", [
|
|
["/bin/blahblahblah"],
|
|
["/bin/blahblahblah", 1, 2, 3],
|
|
[" "],
|
|
[],
|
|
"/bin/blahblahblah, 1, 2, 3,",
|
|
"/bin/blahblahblah",
|
|
" ",
|
|
])
|
|
def test_fail__valid_command(arg: Any) -> None:
|
|
with pytest.raises(ValidatorError):
|
|
print(valid_command(arg))
|