mirror of
https://github.com/TencentCloud/Octop.git
synced 2026-10-20 06:19:17 +08:00
2209cdc90f
Replace orcakit-harness-agent / harness-* with octop-harness, octop-gateway, octop-memory, and octop-browser, and align docs, UI copy, and generated paths. Co-authored-by: jubaoliang <jubaoliang@tencent.com> Co-authored-by: Cursor <cursoragent@cursor.com>
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
"""tests/unit/test_host_paths.py"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from octop.infra.gateway.media.backend_files import (
|
|
is_allowed_host_temp_path,
|
|
is_blocked_host_download_path,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("path", "blocked"),
|
|
[
|
|
("/Users/me/.harness-browser/screenshots/x.png", True),
|
|
("/Users/me/.octop-browser/screenshots/x.png", True),
|
|
("/tmp/secret.png", True),
|
|
("/home/user/x.png", True),
|
|
(r"C:\Users\me\screenshot.png", True),
|
|
(r"c:\users\me\file.png", True),
|
|
(r"C:\Windows\Temp\shot.png", True),
|
|
(r"C:\Program Files\app\data.bin", True),
|
|
(r"%LOCALAPPDATA%\Temp\shot.png", False), # literal percent — not expanded
|
|
("/outbound/chart.png", False),
|
|
("outbound/chart.png", False),
|
|
("/logo.png", False),
|
|
("tmp/cache.bin", False),
|
|
("var/data.json", False),
|
|
("home/readme.md", False),
|
|
],
|
|
)
|
|
def test_is_blocked_host_download_path(path: str, blocked: bool) -> None:
|
|
assert is_blocked_host_download_path(path) is blocked
|
|
|
|
|
|
def test_is_blocked_appdata_temp_segment() -> None:
|
|
assert is_blocked_host_download_path(r"C:\Users\me\AppData\Local\Temp\shot.png") is True
|
|
|
|
|
|
def test_is_allowed_host_temp_path_under_system_temp(
|
|
tmp_path: Path,
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
sample = tmp_path / "preview.png"
|
|
sample.write_bytes(b"png")
|
|
if os.name == "nt":
|
|
monkeypatch.setenv("TEMP", str(tmp_path))
|
|
assert is_allowed_host_temp_path(sample.resolve()) is True
|
|
else:
|
|
assert isinstance(is_allowed_host_temp_path(sample.resolve()), bool)
|