d29da5fe20
Conflicts resolved toward the branch: PM owns dependency preparation, the Windows shim re-exec/hand-off path stays retired (main's shim-parent wait, gateway-resume env token and update_cmd_deps tests dropped), docs describe the PM update flow. The docker workflow parks install-stamp.json around the toolchain step instead of deleting it so tests/docker can compare provenance.
121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
"""Tests for workspace + project-root resolution."""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from agent.lsp.workspace import (
|
|
clear_cache,
|
|
find_git_worktree,
|
|
is_inside_workspace,
|
|
nearest_root,
|
|
normalize_path,
|
|
resolve_workspace_for_file,
|
|
)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _clear():
|
|
clear_cache()
|
|
yield
|
|
clear_cache()
|
|
|
|
|
|
def test_find_git_worktree_finds_dotgit(tmp_path: Path):
|
|
repo = tmp_path / "repo"
|
|
repo.mkdir()
|
|
(repo / ".git").mkdir()
|
|
sub = repo / "src" / "deep"
|
|
sub.mkdir(parents=True)
|
|
assert find_git_worktree(str(sub)) == str(repo)
|
|
|
|
|
|
def test_nearest_root_finds_first_marker(tmp_path: Path):
|
|
root = tmp_path / "p"
|
|
deep = root / "src" / "pkg"
|
|
deep.mkdir(parents=True)
|
|
(root / "pyproject.toml").write_text("", encoding="utf-8")
|
|
found = nearest_root(str(deep / "mod.py"), ["pyproject.toml"])
|
|
assert found == str(root)
|
|
|
|
|
|
def test_nearest_root_skips_package_dirs(tmp_path: Path):
|
|
# hermes_cli/setup.py is a module inside a package, not a project
|
|
# marker; treating it as one spawned a second pyright per worktree.
|
|
root = tmp_path / "p"
|
|
pkg = root / "hermes_cli"
|
|
pkg.mkdir(parents=True)
|
|
(root / "pyproject.toml").write_text("", encoding="utf-8")
|
|
(pkg / "__init__.py").write_text("", encoding="utf-8")
|
|
(pkg / "setup.py").write_text("", encoding="utf-8")
|
|
found = nearest_root(str(pkg / "main.py"), ["pyproject.toml", "setup.py"])
|
|
assert found == str(root)
|
|
|
|
|
|
def test_resolve_workspace_for_file_uses_cwd_first(tmp_path: Path, monkeypatch):
|
|
repo = tmp_path / "repo"
|
|
(repo / ".git").mkdir(parents=True)
|
|
file_path = repo / "x.py"
|
|
file_path.write_text("", encoding="utf-8")
|
|
# cwd is inside the repo
|
|
monkeypatch.chdir(str(repo))
|
|
root, gated = resolve_workspace_for_file(str(file_path))
|
|
assert root == str(repo)
|
|
assert gated is True
|
|
|
|
|
|
def test_resolve_workspace_for_file_survives_deleted_cwd(tmp_path: Path, monkeypatch):
|
|
"""A removed process cwd must read as "no anchor", not raise — the LSP
|
|
workspace resolver runs inside a write tool and must never break a write
|
|
that already landed on disk."""
|
|
repo = tmp_path / "repo"
|
|
(repo / ".git").mkdir(parents=True)
|
|
file_path = repo / "x.py"
|
|
file_path.write_text("")
|
|
scratch = tmp_path / "scratch"
|
|
scratch.mkdir()
|
|
monkeypatch.chdir(scratch)
|
|
scratch.rmdir()
|
|
with pytest.raises(OSError):
|
|
os.getcwd()
|
|
|
|
root, gated = resolve_workspace_for_file(str(file_path))
|
|
|
|
assert root == str(repo)
|
|
assert gated is True
|
|
# The diagnostics path logs through eventlog; its cwd-relative shortener must
|
|
# not raise either, or the write succeeds with diagnostics silently dropped.
|
|
from agent.lsp.eventlog import _short_path
|
|
|
|
assert _short_path(str(file_path)) == str(file_path)
|
|
|
|
|
|
@pytest.mark.platforms("linux")
|
|
def test_normalize_path_expands_tilde(monkeypatch):
|
|
# expanduser keys off USERPROFILE on native Windows, HOME elsewhere —
|
|
# set the var the running host actually consults (host-native rule:
|
|
# never fake the platform).
|
|
if sys.platform == "win32":
|
|
monkeypatch.setenv("USERPROFILE", r"C:\Users\fakeuser")
|
|
expected_base = r"C:\Users\fakeuser"
|
|
else:
|
|
monkeypatch.setenv("HOME", "/home/user")
|
|
expected_base = "/home/user"
|
|
p = normalize_path("~/x.py")
|
|
assert p == os.path.abspath(os.path.join(expected_base, "x.py"))
|
|
|
|
|
|
def test_find_git_worktree_cache_is_capped(tmp_path: Path, monkeypatch):
|
|
"""The start-dir cache resets past _WORKSPACE_CACHE_CAP instead of growing per distinct dir touched."""
|
|
import agent.lsp.workspace as ws
|
|
|
|
monkeypatch.setattr(ws, "_WORKSPACE_CACHE_CAP", 4)
|
|
for i in range(6):
|
|
d = tmp_path / f"d{i}"
|
|
d.mkdir()
|
|
assert find_git_worktree(str(d)) is None
|
|
assert len(ws._workspace_cache) <= 4
|