god-eye/internal/config/ai_profile_test.go

package config

import "testing"

func TestAIProfileByName(t *testing.T) {
	cases := []struct {
		in      string
		wantOK  bool
		wantTag string
	}{
		{"lean", true, "qwen3:1.7b"},
		{"LEAN", true, "qwen3:1.7b"},
		{"balanced", true, "qwen3:4b"},
		{"balance", true, "qwen3:4b"},
		{"mid", true, "qwen3:4b"},
		{"heavy", true, "qwen3:8b"},
		{"max", true, "qwen3:8b"},
		{"power", true, "qwen3:8b"},
		{"Heavy", true, "qwen3:8b"},
		{"nope", false, ""},
		{"", false, ""},
	}
	for _, c := range cases {
		p, ok := AIProfileByName(c.in)
		if ok != c.wantOK {
			t.Errorf("AIProfileByName(%q) ok = %v, want %v", c.in, ok, c.wantOK)
			continue
		}
		if ok && p.FastModel != c.wantTag {
			t.Errorf("AIProfileByName(%q).FastModel = %q, want %q", c.in, p.FastModel, c.wantTag)
		}
	}
}

func TestBuiltinAIProfiles_Unique(t *testing.T) {
	names := map[string]bool{}
	for _, p := range BuiltinAIProfiles {
		if p.Name == "" {
			t.Error("profile with empty name")
		}
		if p.FastModel == "" || p.DeepModel == "" {
			t.Errorf("profile %q missing models", p.Name)
		}
		if p.Description == "" {
			t.Errorf("profile %q missing description", p.Name)
		}
		if names[p.Name] {
			t.Errorf("duplicate profile name: %q", p.Name)
		}
		names[p.Name] = true
	}
}

func TestApplyAIProfile_RespectsOverrides(t *testing.T) {
	cfg := &Config{
		AIFastModel: "user-chose-this:1b",
		AIDeepModel: "user-chose-that:7b",
	}
	ApplyAIProfile(cfg, AIProfileHeavy, true, true)
	if cfg.AIFastModel != "user-chose-this:1b" {
		t.Errorf("overrideFast was ignored: %q", cfg.AIFastModel)
	}
	if cfg.AIDeepModel != "user-chose-that:7b" {
		t.Errorf("overrideDeep was ignored: %q", cfg.AIDeepModel)
	}
	if cfg.AIProfile != "heavy" {
		t.Errorf("AIProfile not set to heavy, got %q", cfg.AIProfile)
	}
}

func TestApplyAIProfile_FillsUnsetFields(t *testing.T) {
	cfg := &Config{}
	ApplyAIProfile(cfg, AIProfileBalanced, false, false)
	if cfg.AIFastModel != "qwen3:4b" {
		t.Errorf("FastModel not applied: %q", cfg.AIFastModel)
	}
	if cfg.AIDeepModel != "qwen3-coder:30b" {
		t.Errorf("DeepModel not applied: %q", cfg.AIDeepModel)
	}
	if cfg.AIProfile != "balanced" {
		t.Errorf("AIProfile not set: %q", cfg.AIProfile)
	}
}

func TestApplyAIProfile_NilConfigNoop(t *testing.T) {
	ApplyAIProfile(nil, AIProfileLean, false, false) // must not panic
}

func TestApplyAIProfile_PartialOverride(t *testing.T) {
	cfg := &Config{AIFastModel: "custom:1b"}
	ApplyAIProfile(cfg, AIProfileHeavy, true, false)
	if cfg.AIFastModel != "custom:1b" {
		t.Errorf("FastModel overridden: %q", cfg.AIFastModel)
	}
	if cfg.AIDeepModel != "qwen3-coder:30b" {
		t.Errorf("DeepModel not applied: %q", cfg.AIDeepModel)
	}
}