package inspur

import (
	"strings"
	"testing"

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

func TestParsePCIeDevices_UsesDeviceNameAsModelWhenPartNumberMissing(t *testing.T) {
	content := []byte(`RESTful PCIE Device info:
[{"id":1,"present":1,"vendor_id":32902,"vendor_name":"Intel","device_id":5409,"device_name":"I350T4V2","bus_num":69,"dev_num":0,"func_num":0,"max_link_width":4,"max_link_speed":2,"current_link_width":4,"current_link_speed":2,"location":"#CPU0_PCIE4","dev_type":2,"dev_subtype":0,"part_num":"","serial_num":"","fw_ver":""}]
BMC sdr Info:`)

	devices := ParsePCIeDevices(content)
	if len(devices) != 1 {
		t.Fatalf("expected 1 device, got %d", len(devices))
	}
	if devices[0].PartNumber != "I350T4V2" {
		t.Fatalf("expected part/model I350T4V2, got %q", devices[0].PartNumber)
	}
	if devices[0].BDF != "45:00.0" {
		t.Fatalf("expected BDF 45:00.0, got %q", devices[0].BDF)
	}
}

func TestMergePCIeDevices_EnrichesGenericAssetEntry(t *testing.T) {
	base := []models.PCIeDevice{
		{
			Slot:         "#CPU1_PCIE9",
			BDF:          "98:00.0",
			VendorID:     0x9005,
			DeviceID:     0x028f,
			DeviceClass:  "SAS",
			Manufacturer: "Adaptec / Microsemi",
		},
	}
	rest := []models.PCIeDevice{
		{
			Slot:         "#CPU1_PCIE9",
			BDF:          "98:00.0",
			VendorID:     0x9005,
			DeviceID:     0x028f,
			DeviceClass:  "Storage Controller",
			Manufacturer: "Microchip",
			PartNumber:   "PM8222-SHBA",
		},
	}

	got := MergePCIeDevices(base, rest)
	if len(got) != 1 {
		t.Fatalf("expected 1 merged device, got %d", len(got))
	}
	if got[0].PartNumber != "PM8222-SHBA" {
		t.Fatalf("expected merged part number PM8222-SHBA, got %q", got[0].PartNumber)
	}
}

func TestParsePCIeDevices_ResolvesModelFromPCIIDsWhenDeviceNameIsRawHex(t *testing.T) {
	content := []byte(`RESTful PCIE Device info:
[{"id":5,"present":1,"vendor_id":36869,"vendor_name":"","device_id":655,"device_name":"0x028F","bus_num":152,"dev_num":0,"func_num":0,"max_link_width":8,"max_link_speed":3,"current_link_width":8,"current_link_speed":3,"location":"#CPU1_PCIE9","dev_type":1,"dev_subtype":7,"part_num":"","serial_num":"","fw_ver":""}]
BMC sdr Info:`)

	devices := ParsePCIeDevices(content)
	if len(devices) != 1 {
		t.Fatalf("expected 1 device, got %d", len(devices))
	}
	if devices[0].PartNumber == "" {
		t.Fatalf("expected part number resolved from pci.ids, got empty")
	}
	if strings.HasPrefix(strings.ToLower(strings.TrimSpace(devices[0].PartNumber)), "0x") {
		t.Fatalf("expected resolved name instead of raw hex, got %q", devices[0].PartNumber)
	}
	if devices[0].Manufacturer == "" {
		t.Fatalf("expected manufacturer resolved from pci.ids")
	}
}