- opencode ищет кастомных агентов в поддиректории agents/ (как .opencode), а не в корне OPENCODE_CONFIG_DIR → распаковка теперь в <dir>/agents/*.md - markdown-агентам добавлен mode: primary (иначе subagent по умолчанию) - idle-детекция сбрасывает таймер при live-строках (text/tool/agent/reasoning), а не только по росту БД → «LLM думает и стримит» не считается зависанием - дефолт idle_timeout поднят с 2м до 5м
248 lines
6.1 KiB
Go
248 lines
6.1 KiB
Go
package config
|
||
|
||
import (
|
||
"os"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func writeCfg(t *testing.T, content string) string {
|
||
t.Helper()
|
||
f, err := os.CreateTemp(t.TempDir(), "cfg*.yaml")
|
||
if err != nil {
|
||
t.Fatalf("temp: %v", err)
|
||
}
|
||
if _, err := f.WriteString(content); err != nil {
|
||
t.Fatalf("write: %v", err)
|
||
}
|
||
f.Close()
|
||
return f.Name()
|
||
}
|
||
|
||
func TestLoad_FullConfig(t *testing.T) {
|
||
t.Setenv("TG_TOKEN", "tok123")
|
||
|
||
yaml := `
|
||
telegram:
|
||
token: "${TG_TOKEN}"
|
||
chat_id: "123"
|
||
|
||
opencode:
|
||
bin: "my-opencode"
|
||
hard_timeout: "10m"
|
||
`
|
||
path := writeCfg(t, yaml)
|
||
cfg, err := Load(path)
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
if cfg.Telegram.Token != "tok123" {
|
||
t.Errorf("tg token = %q", cfg.Telegram.Token)
|
||
}
|
||
if cfg.Telegram.ChatID != "123" {
|
||
t.Errorf("tg chat_id = %q", cfg.Telegram.ChatID)
|
||
}
|
||
if cfg.OpenCode.Bin != "my-opencode" {
|
||
t.Errorf("opencode bin = %q", cfg.OpenCode.Bin)
|
||
}
|
||
if cfg.OpenCode.HardTimeout.Duration() != 10*time.Minute {
|
||
t.Errorf("hard timeout = %v", cfg.OpenCode.HardTimeout)
|
||
}
|
||
}
|
||
|
||
func TestLoad_Defaults(t *testing.T) {
|
||
t.Setenv("TG_TOKEN", "tok")
|
||
t.Setenv("TG_CHAT_ID", "42")
|
||
|
||
yaml := `
|
||
telegram:
|
||
token: "${TG_TOKEN}"
|
||
chat_id: "${TG_CHAT_ID}"
|
||
`
|
||
path := writeCfg(t, yaml)
|
||
cfg, err := Load(path)
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
// default opencode.bin
|
||
if cfg.OpenCode.Bin != "opencode" {
|
||
t.Errorf("bin = %q, want opencode", cfg.OpenCode.Bin)
|
||
}
|
||
if cfg.OpenCode.HardTimeout.Duration() != 20*time.Minute {
|
||
t.Errorf("hard timeout = %v", cfg.OpenCode.HardTimeout)
|
||
}
|
||
if cfg.OpenCode.IdleTimeout.Duration() != 5*time.Minute {
|
||
t.Errorf("idle timeout = %v", cfg.OpenCode.IdleTimeout)
|
||
}
|
||
if cfg.Paths.Worktree != "./worktrees" {
|
||
t.Errorf("worktree = %q, want ./worktrees", cfg.Paths.Worktree)
|
||
}
|
||
}
|
||
|
||
func TestLoad_OpenCodeConfigDir(t *testing.T) {
|
||
t.Setenv("TG_TOKEN", "tok")
|
||
t.Setenv("TG_CHAT_ID", "42")
|
||
|
||
// без config_dir — пусто
|
||
yaml := `telegram:
|
||
token: "${TG_TOKEN}"
|
||
chat_id: "${TG_CHAT_ID}"
|
||
`
|
||
cfg, err := Load(writeCfg(t, yaml))
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
if cfg.OpenCode.ConfigDir != "" {
|
||
t.Errorf("config_dir = %q, want пусто", cfg.OpenCode.ConfigDir)
|
||
}
|
||
|
||
// через env OPENCODE_CONFIG_DIR
|
||
t.Setenv("OPENCODE_CONFIG_DIR", "/opt/agents")
|
||
cfg, err = Load(writeCfg(t, yaml))
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
if cfg.OpenCode.ConfigDir != "/opt/agents" {
|
||
t.Errorf("config_dir = %q, want /opt/agents", cfg.OpenCode.ConfigDir)
|
||
}
|
||
}
|
||
|
||
func TestLoad_EnvOverride(t *testing.T) {
|
||
t.Setenv("TG_TOKEN", "tok")
|
||
t.Setenv("TG_CHAT_ID", "42")
|
||
// env override для opencode.timeout
|
||
t.Setenv("OPENCODE_CONFIG", "/custom/opencode.json")
|
||
|
||
yaml := `telegram:
|
||
token: "${TG_TOKEN}"
|
||
chat_id: "${TG_CHAT_ID}"
|
||
`
|
||
path := writeCfg(t, yaml)
|
||
cfg, err := Load(path)
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
if cfg.OpenCode.Config != "/custom/opencode.json" {
|
||
t.Errorf("opencode config = %q", cfg.OpenCode.Config)
|
||
}
|
||
}
|
||
|
||
func TestLoad_MissingEnvVarUsesDefault(t *testing.T) {
|
||
t.Setenv("TG_TOKEN", "tok")
|
||
t.Setenv("TG_CHAT_ID", "42")
|
||
|
||
yaml := `telegram:
|
||
token: "${TG_TOKEN}"
|
||
chat_id: "${TG_CHAT_ID}"
|
||
opencode:
|
||
db_path: "${OPENCODE_DB:-/default/db.sqlite}"
|
||
`
|
||
path := writeCfg(t, yaml)
|
||
cfg, err := Load(path)
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
if cfg.OpenCode.DBPath != "/default/db.sqlite" {
|
||
t.Errorf("db_path = %q, want /default/db.sqlite", cfg.OpenCode.DBPath)
|
||
}
|
||
}
|
||
|
||
func TestLoad_FileNotFound(t *testing.T) {
|
||
_, err := Load("/nonexistent/config.yaml")
|
||
if err == nil || !strings.Contains(err.Error(), "file not found") {
|
||
t.Fatalf("ожидалась ошибка file not found, got %v", err)
|
||
}
|
||
}
|
||
|
||
func TestResolveExePaths_RelativeToExe(t *testing.T) {
|
||
t.Setenv("TG_TOKEN", "tok")
|
||
t.Setenv("TG_CHAT_ID", "42")
|
||
|
||
yaml := `telegram:
|
||
token: "${TG_TOKEN}"
|
||
chat_id: "${TG_CHAT_ID}"
|
||
paths:
|
||
db: "./ratatoskr.db"
|
||
worktree: "./worktrees"
|
||
`
|
||
cfg, err := Load(writeCfg(t, yaml))
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
cfg.ResolveExePaths()
|
||
|
||
base := ExeDir()
|
||
if cfg.Paths.DB != filepath.Join(base, "ratatoskr.db") {
|
||
t.Errorf("db = %q, want %q", cfg.Paths.DB, filepath.Join(base, "ratatoskr.db"))
|
||
}
|
||
if cfg.Paths.Worktree != filepath.Join(base, "worktrees") {
|
||
t.Errorf("worktree = %q, want %q", cfg.Paths.Worktree, filepath.Join(base, "worktrees"))
|
||
}
|
||
}
|
||
|
||
func TestResolveExePaths_AbsoluteKept(t *testing.T) {
|
||
t.Setenv("TG_TOKEN", "tok")
|
||
t.Setenv("TG_CHAT_ID", "42")
|
||
|
||
absDB := filepath.Join(string(filepath.Separator), "data", "ratatoskr.db") // абсолютный для текущей ОС
|
||
absWt := filepath.Join(string(filepath.Separator), "worktrees")
|
||
yaml := `telegram:
|
||
username: "${TG_TOKEN}"
|
||
chat_id: "${TG_CHAT_ID}"
|
||
paths:
|
||
db: "` + absDB + `"
|
||
worktree: "` + absWt + `"
|
||
`
|
||
cfg, err := Load(writeCfg(t, yaml))
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
cfg.ResolveExePaths()
|
||
if cfg.Paths.DB != absDB {
|
||
t.Errorf("db = %q, want абсолютный путь сохранён %q", cfg.Paths.DB, absDB)
|
||
}
|
||
if cfg.Paths.Worktree != absWt {
|
||
t.Errorf("worktree = %q, want абсолютный путь сохранён %q", cfg.Paths.Worktree, absWt)
|
||
}
|
||
}
|
||
|
||
func TestLoad_MissingRequired(t *testing.T) {
|
||
// пустой конфиг — все обязательные поля отсутствуют
|
||
t.Setenv("TG_TOKEN", "")
|
||
t.Setenv("TG_CHAT_ID", "")
|
||
|
||
yaml := `telegram:
|
||
token: ""
|
||
chat_id: ""
|
||
`
|
||
path := writeCfg(t, yaml)
|
||
_, err := Load(path)
|
||
if err == nil {
|
||
t.Fatal("ожидалась ошибка валидации")
|
||
}
|
||
if !strings.Contains(err.Error(), "telegram.token") {
|
||
t.Errorf("ожидалось telegram.token в ошибке, got %v", err)
|
||
}
|
||
if !strings.Contains(err.Error(), "telegram.chat_id") {
|
||
t.Errorf("ожидалось telegram.chat_id в ошибке, got %v", err)
|
||
}
|
||
}
|
||
|
||
func TestLoad_InvalidDuration(t *testing.T) {
|
||
t.Setenv("TG_TOKEN", "x")
|
||
t.Setenv("TG_CHAT_ID", "x")
|
||
|
||
yaml := `telegram:
|
||
token: "x"
|
||
chat_id: "x"
|
||
opencode:
|
||
hard_timeout: "не-длительность"
|
||
`
|
||
path := writeCfg(t, yaml)
|
||
_, err := Load(path)
|
||
if err == nil || !strings.Contains(err.Error(), "parse duration") {
|
||
t.Fatalf("ожидалась ошибка парсинга duration, got %v", err)
|
||
}
|
||
} |