package webui

import (
	"path/filepath"
	"testing"
	"time"

	"bee/audit/internal/app"
)

func TestParseKmsgLine_xid64EscalatesToCritical(t *testing.T) {
	line := "6,1234,555555,-;NVRM: Xid (PCI:0000:65:00.0): 64, pid=1234, name=python, Row Remapper: New row remapping failed"
	evt, ok := parseKmsgLine(line)
	if !ok {
		t.Fatalf("expected line to match a pattern")
	}
	if evt.severity != "critical" {
		t.Fatalf("severity: got %q, want %q", evt.severity, "critical")
	}
	if len(evt.ids) != 1 || evt.ids[0] != "0000:65:00.0" {
		t.Fatalf("ids: got %v", evt.ids)
	}
}

func TestParseKmsgLine_xid63StaysWarning(t *testing.T) {
	line := "6,1234,555555,-;NVRM: Xid (PCI:0000:65:00.0): 63, pid=1234, Class 0x90a0"
	evt, ok := parseKmsgLine(line)
	if !ok {
		t.Fatalf("expected line to match a pattern")
	}
	if evt.severity != "warning" {
		t.Fatalf("severity: got %q, want %q", evt.severity, "warning")
	}
}

func TestFlushWindow_escalatesToCriticalOnXid64(t *testing.T) {
	dbPath := filepath.Join(t.TempDir(), "status.json")
	db, err := app.OpenComponentStatusDB(dbPath)
	if err != nil {
		t.Fatalf("open status db: %v", err)
	}
	w := newKmsgWatcher(db)

	window := &kmsgWindow{
		targets:   []string{"gpu-burn"},
		startedAt: time.Now(),
		events: []kmsgEvent{
			{raw: "NVRM: GPU 0000:65:00.0: RmInitAdapter failed", ids: []string{"0000:65:00.0"}, category: "gpu", severity: "warning"},
			{raw: "NVRM: Xid (PCI:0000:65:00): 64, pid=1234, name=python", ids: []string{"0000:65:00.0"}, category: "gpu", severity: "critical"},
		},
	}

	w.flushWindow(window)

	rec, ok := db.Get("pcie:gpu:0000:65:00.0")
	if !ok {
		t.Fatalf("expected a status record for the gpu")
	}
	if rec.Status != "Critical" {
		t.Fatalf("status: got %q, want %q", rec.Status, "Critical")
	}
}