File size: 3,517 Bytes
5ef3875 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 | from pathlib import Path
import pytest
from ycode.config import DEFAULT_IGNORE_DIRS, DEFAULT_MODEL, load_config, load_env_files
from ycode.errors import ConfigError
def write(path: Path, text: str) -> Path:
path.parent.mkdir(parents=True, exist_ok=True)
path.write_text(text)
return path
def test_defaults(workspace, isolated_home):
cfg = load_config(workspace, env={})
assert cfg.model == DEFAULT_MODEL
assert cfg.max_steps == 50
assert cfg.approval_mode == "normal"
assert cfg.ignore_dirs == DEFAULT_IGNORE_DIRS
def test_global_then_project_then_env_then_cli(workspace, isolated_home):
write(isolated_home / "config.toml", 'model = "global-model"\nmax_steps = 10\ntheme = "mono"\n')
write(workspace / ".ycode" / "config.toml", 'model = "project-model"\nmax_steps = 20\n')
cfg = load_config(workspace, env={})
assert (cfg.model, cfg.max_steps, cfg.theme) == ("project-model", 20, "mono")
cfg = load_config(workspace, env={"YCODE_MODEL": "env-model", "YCODE_MAX_STEPS": "30"})
assert (cfg.model, cfg.max_steps) == ("env-model", 30)
cfg = load_config(workspace, env={"YCODE_MODEL": "env-model"}, cli_overrides={"model": "cli-model",
"max_steps": None})
assert cfg.model == "cli-model"
assert cfg.max_steps == 20
def test_project_config_cannot_loosen_security(workspace, isolated_home):
write(workspace / ".ycode" / "config.toml",
'approval_mode = "auto"\nallow_outside_workspace = true\n')
cfg = load_config(workspace, env={})
assert cfg.approval_mode == "normal"
assert cfg.allow_outside_workspace is False
def test_project_config_can_tighten_security(workspace, isolated_home):
write(workspace / ".ycode" / "config.toml", 'approval_mode = "strict"\n')
assert load_config(workspace, env={}).approval_mode == "strict"
def test_global_config_may_enable_auto(workspace, isolated_home):
write(isolated_home / "config.toml", 'approval_mode = "auto"\n')
assert load_config(workspace, env={}).approval_mode == "auto"
def test_extra_ignore_dirs_extend_defaults(workspace, isolated_home):
write(workspace / ".ycode" / "config.toml", 'extra_ignore_dirs = ["generated"]\n')
cfg = load_config(workspace, env={})
assert "generated" in cfg.ignore_dirs and "node_modules" in cfg.ignore_dirs
@pytest.mark.parametrize("text, message", [
('max_steps = "lots"', "integer"),
('max_steps = 0', "positive"),
('approval_mode = "yolo"', "approval_mode"),
('unknown_key = 1', "unknown setting"),
('model = [', "Invalid TOML"),
('effort = "ultra"', "effort"),
])
def test_invalid_config_raises_clear_error(workspace, isolated_home, text, message):
write(isolated_home / "config.toml", text + "\n")
with pytest.raises(ConfigError, match=message):
load_config(workspace, env={})
def test_invalid_env_value(workspace):
with pytest.raises(ConfigError, match="YCODE_MAX_STEPS|max_steps"):
load_config(workspace, env={"YCODE_MAX_STEPS": "abc"})
def test_env_file_loading_does_not_override(workspace, monkeypatch):
write(workspace / ".env", "ANTHROPIC_API_KEY=from-file\nYCODE_MODEL=file-model\n")
monkeypatch.setenv("YCODE_MODEL", "already-set")
loaded = load_env_files(workspace)
import os
assert workspace.resolve() / ".env" in loaded
assert os.environ["ANTHROPIC_API_KEY"] == "from-file"
assert os.environ["YCODE_MODEL"] == "already-set"
|