wehub-resource-sync
77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
"""Tests for the session environment context block (system-prompt injection)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import subprocess
|
|
import sys
|
|
|
|
from coworker.environment import environment_context
|
|
|
|
|
|
def _git_repo(tmp_path):
|
|
ws = tmp_path / "repo"
|
|
ws.mkdir()
|
|
run = lambda *a: subprocess.run(
|
|
["git", "-C", str(ws), *a], capture_output=True, check=True
|
|
)
|
|
run("init", "-q", "-b", "main")
|
|
run("config", "user.email", "t@t.io")
|
|
run("config", "user.name", "T")
|
|
(ws / "f.txt").write_text("1", encoding="utf-8")
|
|
run("add", "-A")
|
|
run("commit", "-qm", "first commit")
|
|
return ws
|
|
|
|
|
|
def test_context_includes_workspace_platform_and_date(tmp_path):
|
|
block = environment_context(tmp_path)
|
|
assert str(tmp_path.resolve()) in block
|
|
assert sys.platform in block
|
|
assert "Today's date:" in block
|
|
assert "<environment>" in block and "</environment>" in block
|
|
|
|
|
|
def test_context_outside_git_repo(tmp_path):
|
|
assert "not a git repository" in environment_context(tmp_path)
|
|
|
|
|
|
def test_context_with_git_repo(tmp_path):
|
|
ws = _git_repo(tmp_path)
|
|
block = environment_context(ws)
|
|
assert "Git branch: main" in block
|
|
assert "Git status: clean" in block
|
|
assert "first commit" in block
|
|
|
|
|
|
def test_context_shows_dirty_status(tmp_path):
|
|
ws = _git_repo(tmp_path)
|
|
(ws / "f.txt").write_text("2", encoding="utf-8")
|
|
(ws / "new.txt").write_text("x", encoding="utf-8")
|
|
block = environment_context(ws)
|
|
assert "Git status (2 changed):" in block
|
|
assert "f.txt" in block and "new.txt" in block
|
|
|
|
|
|
class _Stub:
|
|
def complete(self, **kwargs): # pragma: no cover
|
|
raise NotImplementedError
|
|
|
|
def capabilities(self, model):
|
|
from coworker.providers import ModelCapabilities
|
|
|
|
return ModelCapabilities()
|
|
|
|
|
|
def test_build_engine_injects_environment(tmp_path):
|
|
from coworker.agent import build_engine
|
|
from coworker.agents import code_agent
|
|
|
|
engine = build_engine(agent=code_agent(), workspace=tmp_path, provider=_Stub())
|
|
try:
|
|
system = engine.messages[0]
|
|
assert system["role"] == "system"
|
|
assert "<environment>" in system["content"]
|
|
assert str(tmp_path.resolve()) in system["content"]
|
|
finally:
|
|
engine.executor.close()
|