Implement audit enrichments, TUI workflows, and production ISO scaffold

audit/internal/collector/nvidia_test.go

@@ -0,0 +1,116 @@
+package collector
+
+import (
+	"bee/audit/internal/schema"
+	"testing"
+)
+
+func TestParseNVIDIASMIQuery(t *testing.T) {
+	raw := "0, 00000000:65:00.0, GPU-SERIAL-1, 96.00.1F.00.02, 54, 210.33, 0, 5, Not Active\n"
+	byBDF, err := parseNVIDIASMIQuery(raw)
+	if err != nil {
+		t.Fatalf("parse failed: %v", err)
+	}
+
+	gpu, ok := byBDF["0000:65:00.0"]
+	if !ok {
+		t.Fatalf("gpu by normalized bdf not found")
+	}
+	if gpu.Serial != "GPU-SERIAL-1" {
+		t.Fatalf("serial: got %q", gpu.Serial)
+	}
+	if gpu.VBIOS != "96.00.1F.00.02" {
+		t.Fatalf("vbios: got %q", gpu.VBIOS)
+	}
+	if gpu.ECCUncorrected == nil || *gpu.ECCUncorrected != 0 {
+		t.Fatalf("ecc uncorrected: got %v", gpu.ECCUncorrected)
+	}
+	if gpu.HWSlowdown == nil || *gpu.HWSlowdown {
+		t.Fatalf("hw slowdown: got %v, want false", gpu.HWSlowdown)
+	}
+}
+
+func TestNormalizePCIeBDF(t *testing.T) {
+	tests := []struct {
+		in   string
+		want string
+	}{
+		{"00000000:17:00.0", "0000:17:00.0"},
+		{"0000:17:00.0", "0000:17:00.0"},
+		{"17:00.0", "0000:17:00.0"},
+	}
+	for _, tt := range tests {
+		got := normalizePCIeBDF(tt.in)
+		if got != tt.want {
+			t.Fatalf("normalizePCIeBDF(%q)=%q want %q", tt.in, got, tt.want)
+		}
+	}
+}
+
+func TestEnrichPCIeWithNVIDIAData_driverLoaded(t *testing.T) {
+	vendorID := nvidiaVendorID
+	bdf := "0000:65:00.0"
+	manufacturer := "NVIDIA Corporation"
+	status := "OK"
+	devices := []schema.HardwarePCIeDevice{
+		{
+			VendorID:     &vendorID,
+			BDF:          &bdf,
+			Manufacturer: &manufacturer,
+			Status:       &status,
+		},
+	}
+
+	byBDF := map[string]nvidiaGPUInfo{
+		"0000:65:00.0": {
+			BDF:            "0000:65:00.0",
+			Serial:         "GPU-ABC",
+			VBIOS:          "96.00.1F.00.02",
+			ECCUncorrected: ptrInt64(2),
+			ECCCorrected:   ptrInt64(10),
+			TemperatureC:   ptrFloat(55.5),
+			PowerW:         ptrFloat(230.2),
+		},
+	}
+
+	out := enrichPCIeWithNVIDIAData(devices, byBDF, "BOARD-001", true)
+	if out[0].SerialNumber == nil || *out[0].SerialNumber != "GPU-ABC" {
+		t.Fatalf("serial: got %v", out[0].SerialNumber)
+	}
+	if out[0].Firmware == nil || *out[0].Firmware != "96.00.1F.00.02" {
+		t.Fatalf("firmware: got %v", out[0].Firmware)
+	}
+	if out[0].Status == nil || *out[0].Status != "WARNING" {
+		t.Fatalf("status: got %v", out[0].Status)
+	}
+	if out[0].Telemetry == nil {
+		t.Fatal("expected telemetry")
+	}
+	if got, ok := out[0].Telemetry["ecc_uncorrected_total"].(int64); !ok || got != 2 {
+		t.Fatalf("ecc_uncorrected_total: got %#v", out[0].Telemetry["ecc_uncorrected_total"])
+	}
+}
+
+func TestEnrichPCIeWithNVIDIAData_driverMissingFallback(t *testing.T) {
+	vendorID := nvidiaVendorID
+	bdf := "0000:17:00.0"
+	manufacturer := "NVIDIA Corporation"
+	devices := []schema.HardwarePCIeDevice{
+		{
+			VendorID:     &vendorID,
+			BDF:          &bdf,
+			Manufacturer: &manufacturer,
+		},
+	}
+
+	out := enrichPCIeWithNVIDIAData(devices, nil, "BOARD-123", false)
+	if out[0].SerialNumber == nil || *out[0].SerialNumber != "BOARD-123-PCIE-0000:17:00.0" {
+		t.Fatalf("fallback serial: got %v", out[0].SerialNumber)
+	}
+	if out[0].Status == nil || *out[0].Status != "UNKNOWN" {
+		t.Fatalf("fallback status: got %v", out[0].Status)
+	}
+}
+
+func ptrInt64(v int64) *int64     { return &v }
+func ptrFloat(v float64) *float64 { return &v }