Pulse/internal/ai/patrol_context_additional_test.go

package ai

import (
	"strings"
	"testing"

	"github.com/rcourtman/pulse-go-rewrite/internal/models"
)

func TestPatrolService_buildSeedContext_QuietSummary(t *testing.T) {
	ps := NewPatrolService(nil, nil)
	ps.SetConfig(PatrolConfig{
		Enabled:      true,
		AnalyzeNodes: true,
	})

	state := models.StateSnapshot{
		Nodes: []models.Node{
			{
				ID:     "node-1",
				Name:   "node-1",
				Status: "online",
				CPU:    0.10,
				Memory: models.Memory{Usage: 20.0},
			},
		},
	}

	seed, _ := ps.buildSeedContextState(patrolRuntimeStateForTest(ps, state), nil, nil)
	if !strings.Contains(seed, "# Node Metrics") {
		t.Fatalf("expected detailed node metrics section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "| node-1 | online | 10% | 20% |") {
		t.Fatalf("expected node row in metrics table, got:\n%s", seed)
	}
}

func TestPatrolService_buildSeedContext_ScopeSection(t *testing.T) {
	ps := NewPatrolService(nil, nil)
	scope := &PatrolScope{
		Reason:          TriggerReasonAlertFired,
		Context:         "CPU alert",
		ResourceIDs:     []string{"node-1"},
		ResourceTypes:   []string{"node"},
		AlertIdentifier: "alert-123",
		FindingID:       "finding-456",
		Depth:           PatrolDepthQuick,
	}

	seed, _ := ps.buildSeedContextState(patrolRuntimeStateForTest(ps, models.StateSnapshot{}), scope, nil)
	if !strings.Contains(seed, "# Patrol Scope") {
		t.Fatalf("expected patrol scope section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "Trigger: alert") {
		t.Fatalf("expected trigger in scope section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "Context: CPU alert") {
		t.Fatalf("expected context in scope section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "Requested resources: node-1") {
		t.Fatalf("expected resource IDs in scope section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "Requested resource types: node") {
		t.Fatalf("expected resource types in scope section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "Effective scope: 1 resource (node-1)") {
		t.Fatalf("expected effective scope in scope section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "Alert Identifier: alert-123") {
		t.Fatalf("expected alert identifier in scope section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "Finding ID: finding-456") {
		t.Fatalf("expected finding ID in scope section, got:\n%s", seed)
	}
	if !strings.Contains(seed, "Depth: quick") {
		t.Fatalf("expected depth in scope section, got:\n%s", seed)
	}
}

func TestPatrolService_buildSeedContext_TypeScopedEffectiveScopeSection(t *testing.T) {
	ps := NewPatrolService(nil, nil)
	scope := &PatrolScope{
		ResourceTypes: []string{"node"},
		Depth:         PatrolDepthQuick,
	}

	state := models.StateSnapshot{
		Nodes: []models.Node{
			{ID: "node-1", Name: "node-1", Status: "online"},
			{ID: "node-2", Name: "node-2", Status: "online"},
		},
	}

	seed, _ := ps.buildSeedContextState(patrolRuntimeStateForTest(ps, state), scope, nil)
	if !strings.Contains(seed, "Effective scope: 2 resources (node-1, node-2)") {
		t.Fatalf("expected type-scoped effective scope section, got:\n%s", seed)
	}
}