package inspur

import (
	"testing"
	"time"

	"git.mchus.pro/mchus/logpile/internal/models"
)

func TestEnrichGPUsFromHGXHWInfo_UsesHGXLogicalMapping(t *testing.T) {
	hw := &models.HardwareConfig{
		GPUs: []models.GPU{
			{Slot: "#GPU6"},
			{Slot: "#GPU7"},
			{Slot: "#GPU0"},
			{Slot: "#CPU0_PE1_E_BMC", Model: "AST2500 VGA"},
		},
	}

	content := []byte(`
# curl -X GET http://127.0.0.1/redfish/v1/Chassis/HGX_GPU_SXM_1/Assembly
{"Name":"GPU Board Assembly","Model":"B200 180GB HBM3e","PartNumber":"PN1","SerialNumber":"SXM1SN"}
# curl -X GET http://127.0.0.1/redfish/v1/Chassis/HGX_GPU_SXM_3/Assembly
{"Name":"GPU Board Assembly","Model":"B200 180GB HBM3e","PartNumber":"PN3","SerialNumber":"SXM3SN"}
# curl -X GET http://127.0.0.1/redfish/v1/Chassis/HGX_GPU_SXM_5/Assembly
{"Name":"GPU Board Assembly","Model":"B200 180GB HBM3e","PartNumber":"PN5","SerialNumber":"SXM5SN"}
`)

	enrichGPUsFromHGXHWInfo(content, hw)

	if hw.GPUs[0].SerialNumber != "SXM3SN" {
		t.Fatalf("expected #GPU6 to map to SXM3 serial, got %q", hw.GPUs[0].SerialNumber)
	}
	if hw.GPUs[1].SerialNumber != "SXM1SN" {
		t.Fatalf("expected #GPU7 to map to SXM1 serial, got %q", hw.GPUs[1].SerialNumber)
	}
	if hw.GPUs[2].SerialNumber != "SXM5SN" {
		t.Fatalf("expected #GPU0 to map to SXM5 serial, got %q", hw.GPUs[2].SerialNumber)
	}
	for _, g := range hw.GPUs {
		if g.Slot == "#CPU0_PE1_E_BMC" {
			t.Fatalf("expected non-HGX BMC VGA entry to be filtered out")
		}
	}
}

func TestEnrichGPUsFromHGXHWInfo_AddsMissingLogicalGPU(t *testing.T) {
	hw := &models.HardwareConfig{
		GPUs: []models.GPU{
			{Slot: "#GPU0"},
			{Slot: "#GPU1"},
			{Slot: "#GPU2"},
			{Slot: "#GPU3"},
			{Slot: "#GPU4"},
			{Slot: "#GPU5"},
			{Slot: "#GPU7"},
		},
	}

	content := []byte(`
# curl -X GET http://127.0.0.1/redfish/v1/Chassis/HGX_GPU_SXM_3/Assembly
{"Name":"GPU Board Assembly","Model":"B200 180GB HBM3e","PartNumber":"PN3","SerialNumber":"SXM3SN"}
`)

	enrichGPUsFromHGXHWInfo(content, hw)

	found := false
	for _, g := range hw.GPUs {
		if g.Slot == "#GPU6" {
			found = true
			if g.SerialNumber != "SXM3SN" {
				t.Fatalf("expected synthesized #GPU6 serial SXM3SN, got %q", g.SerialNumber)
			}
		}
	}
	if !found {
		t.Fatalf("expected synthesized #GPU6 entry")
	}
}

func TestApplyGPUStatusFromEvents_MarksFaultedGPU(t *testing.T) {
	hw := &models.HardwareConfig{
		GPUs: []models.GPU{
			{Slot: "#GPU6"},
			{Slot: "#GPU5"},
		},
	}

	events := []models.Event{
		{
			ID:          "17FFB002",
			Timestamp:   time.Now(),
			Description: "PCIe Present mismatch BIOS miss F_GPU6",
		},
	}

	applyGPUStatusFromEvents(hw, events)

	if hw.GPUs[0].Status != "Critical" {
		t.Fatalf("expected #GPU6 status Critical, got %q", hw.GPUs[0].Status)
	}
	if hw.GPUs[1].Status != "OK" {
		t.Fatalf("expected healthy GPU status OK, got %q", hw.GPUs[1].Status)
	}
}

func TestParseIDLLog_ParsesStructuredJSONLine(t *testing.T) {
	content := []byte(`{ "MESSAGE": "|2026-01-12T23:05:18+08:00|PCIE|Assert|Critical|17FFB002|PCIe Present mismatch BIOS miss F_GPU6 - Assert|" }`)

	events := ParseIDLLog(content)
	if len(events) != 1 {
		t.Fatalf("expected 1 event from JSON line, got %d", len(events))
	}
	if events[0].ID != "17FFB002" {
		t.Fatalf("expected event ID 17FFB002, got %q", events[0].ID)
	}
	if events[0].Source != "PCIE" {
		t.Fatalf("expected source PCIE, got %q", events[0].Source)
	}
}