test: improve coverage and raise threshold to 80% (#154)

This commit is contained in:
banteg
2026-01-16 13:19:41 +04:00
committed by GitHub
parent 92b33c5181
commit da881fcee5
26 changed files with 4282 additions and 412 deletions
+68
View File
@@ -6,6 +6,7 @@ from pathlib import Path
import pytest
from takopi.telegram import files as tg_files
from takopi.telegram.files import ZipTooLargeError, zip_directory
@@ -41,3 +42,70 @@ def test_zip_directory_limits_size(tmp_path: Path) -> None:
with pytest.raises(ZipTooLargeError):
zip_directory(root, Path("dir"), deny_globs=(), max_bytes=10)
def test_split_command_args_falls_back_on_bad_quotes() -> None:
assert tg_files.split_command_args('bad "quote') == ("bad", '"quote')
def test_parse_file_command_unknown_command() -> None:
command, rest, error = tg_files.parse_file_command("nope arg")
assert command is None
assert rest == "arg"
assert error == tg_files.file_usage()
def test_parse_file_prompt_errors() -> None:
path, force, error = tg_files.parse_file_prompt("--wat", allow_empty=False)
assert path is None
assert force is False
assert error == "unknown flag: --wat"
path, force, error = tg_files.parse_file_prompt("", allow_empty=False)
assert path is None
assert force is False
assert error == "missing path"
def test_parse_file_prompt_force_flag() -> None:
path, force, error = tg_files.parse_file_prompt(
"--force note.txt", allow_empty=False
)
assert path == "note.txt"
assert force is True
assert error is None
def test_normalize_relative_path_rejects_invalid() -> None:
for value in ("", " ", "~/.ssh", "/etc/passwd", "../secret", ".git/config"):
assert tg_files.normalize_relative_path(value) is None
def test_normalize_relative_path_rejects_dot_only() -> None:
assert tg_files.normalize_relative_path("./") is None
def test_normalize_relative_path_strips_dots() -> None:
assert tg_files.normalize_relative_path("docs/./guide.txt") == Path(
"docs/guide.txt"
)
def test_resolve_path_within_root_rejects_escape(tmp_path: Path) -> None:
assert tg_files.resolve_path_within_root(tmp_path, Path("../escape")) is None
def test_deny_reason_matches_patterns() -> None:
assert tg_files.deny_reason(Path(".git/config"), ["**/*.pem"]) == ".git/**"
assert tg_files.deny_reason(Path("secrets/key.pem"), ["**/*.pem"]) == "**/*.pem"
def test_format_bytes_various_units() -> None:
assert tg_files.format_bytes(0) == "0 b"
assert tg_files.format_bytes(1536) == "1.5 kb"
assert tg_files.format_bytes(20480) == "20 kb"
def test_default_upload_name_fallbacks() -> None:
assert tg_files.default_upload_name("", "files/report.txt") == "report.txt"
assert tg_files.default_upload_name(None, None) == "upload.bin"