wehub-resource-sync
76 lines
3.0 KiB
Python
76 lines
3.0 KiB
Python
"""Unit tests for opensquilla.contrib.codetask.config."""
|
|
|
|
import sys
|
|
|
|
from opensquilla.contrib.codetask import config
|
|
|
|
|
|
class TestSlugify:
|
|
def test_basic(self):
|
|
assert config.slugify("Fix CSV export bug") == "fix-csv-export-bug"
|
|
|
|
def test_strips_punctuation_and_unicode(self):
|
|
assert config.slugify("修复 CSV 导出 bug!!!") == "csv-bug"
|
|
|
|
def test_truncates(self):
|
|
s = config.slugify("a" * 100, max_len=10)
|
|
assert len(s) <= 10
|
|
|
|
def test_empty_fallback(self):
|
|
assert config.slugify("!!!") == "task"
|
|
|
|
|
|
class TestPaths:
|
|
def test_runs_root_override(self, monkeypatch, tmp_path):
|
|
monkeypatch.setenv("OPENSQUILLA_CODETASK_RUNS_DIR", str(tmp_path))
|
|
assert config.runs_root() == tmp_path
|
|
assert config.run_dir("r1") == tmp_path / "r1"
|
|
assert config.repo_dir("r1") == tmp_path / "r1" / "repo"
|
|
assert config.artifact_path("r1", "result.json") == tmp_path / "r1" / "result.json"
|
|
|
|
def test_scratch_is_sandbox_writable_not_under_runs_root(self, monkeypatch, tmp_path):
|
|
# The scratch dir MUST NOT live under the runs root (which defaults to
|
|
# ~/.opensquilla, i.e. /root/... — hard-blocked by the agent sandbox).
|
|
# It must default under the system temp dir.
|
|
import tempfile
|
|
|
|
monkeypatch.setenv("OPENSQUILLA_CODETASK_RUNS_DIR", str(tmp_path))
|
|
monkeypatch.delenv("OPENSQUILLA_CODETASK_SCRATCH_DIR", raising=False)
|
|
scratch = config.scratch_dir("r1")
|
|
assert tempfile.gettempdir() in str(scratch)
|
|
assert str(tmp_path) not in str(scratch)
|
|
assert scratch.name == "scratch"
|
|
|
|
def test_scratch_override(self, monkeypatch, tmp_path):
|
|
monkeypatch.setenv("OPENSQUILLA_CODETASK_SCRATCH_DIR", str(tmp_path / "s"))
|
|
assert config.scratch_dir("r1") == tmp_path / "s" / "r1" / "scratch"
|
|
|
|
def test_agent_python_default(self, monkeypatch):
|
|
monkeypatch.delenv("OPENSQUILLA_CODETASK_AGENT_PYTHON", raising=False)
|
|
assert config.agent_python() == sys.executable
|
|
|
|
def test_agent_python_override(self, monkeypatch):
|
|
monkeypatch.setenv("OPENSQUILLA_CODETASK_AGENT_PYTHON", "/opt/py/bin/python")
|
|
assert config.agent_python() == "/opt/py/bin/python"
|
|
|
|
def test_prompt_template_shipped(self, monkeypatch):
|
|
monkeypatch.delenv("OPENSQUILLA_CODETASK_PROMPT_TEMPLATE", raising=False)
|
|
path = config.prompt_template_path()
|
|
assert path.exists()
|
|
body = path.read_text(encoding="utf-8")
|
|
for slot in ("{task}", "{env_hints}", "{scratch_dir}", "{manifest_name}"):
|
|
assert slot in body
|
|
|
|
|
|
class TestAgentConfigPath:
|
|
def test_default_exists(self):
|
|
p = config.agent_config_path()
|
|
assert p.name == "config.toml"
|
|
assert p.exists()
|
|
|
|
def test_env_override(self, monkeypatch, tmp_path):
|
|
custom = tmp_path / "custom.toml"
|
|
custom.write_text("")
|
|
monkeypatch.setenv("OPENSQUILLA_CODETASK_AGENT_CONFIG", str(custom))
|
|
assert config.agent_config_path() == custom
|