package collector

import (
	"bee/audit/internal/schema"
	"testing"
)

func TestParseMDStatArrays(t *testing.T) {
	raw := `Personalities : [raid1]
md126 : active raid1 nvme0n1[0] nvme1n1[1]
      976630464 blocks super external:/md127/0 [2/2] [UU]

md125 : active raid1 nvme2n1[0] nvme3n1[1]
      976630464 blocks super external:/md127/1 [2/1] [U_]
`
	arrays := parseMDStatArrays(raw)
	if len(arrays) != 2 {
		t.Fatalf("expected 2 arrays, got %d", len(arrays))
	}
	if arrays[0].Name != "md126" || arrays[0].Degraded {
		t.Fatalf("unexpected array0: %+v", arrays[0])
	}
	if len(arrays[0].Members) != 2 || arrays[0].Members[0] != "nvme0n1" {
		t.Fatalf("unexpected members array0: %+v", arrays[0].Members)
	}
	if arrays[1].Name != "md125" || !arrays[1].Degraded {
		t.Fatalf("unexpected array1: %+v", arrays[1])
	}
}

func TestParseMDAdmPlatformLicense(t *testing.T) {
	premium := `Platform : Intel(R) Virtual RAID on CPU
Version  : 1.3.0.1138
RAID Levels : raid0 raid1 raid5 raid10
Total Disks : 4
License  : Premium
`
	got := parseMDAdmPlatformLicense(premium)
	if got == nil || *got != "premium" {
		t.Fatalf("expected 'premium', got %v", got)
	}

	standard := `Platform : Intel(R) Virtual RAID on CPU
License  : Standard
`
	got = parseMDAdmPlatformLicense(standard)
	if got == nil || *got != "standard" {
		t.Fatalf("expected 'standard', got %v", got)
	}

	noLicense := `Platform : Intel(R) Virtual RAID on CPU
Version  : 1.0.0
`
	got = parseMDAdmPlatformLicense(noLicense)
	if got != nil {
		t.Fatalf("expected nil, got %v", *got)
	}
}

func TestHasVROCController(t *testing.T) {
	intel := vendorIntel
	model := "Volume Management Device NVMe RAID Controller"
	class := "MassStorageController"
	tests := []struct {
		name string
		pcie []schema.HardwarePCIeDevice
		want bool
	}{
		{
			name: "intel vroc",
			pcie: []schema.HardwarePCIeDevice{{VendorID: &intel, Model: &model, DeviceClass: &class}},
			want: true,
		},
		{
			name: "non-intel raid",
			pcie: []schema.HardwarePCIeDevice{{}},
			want: false,
		},
	}
	for _, tt := range tests {
		got := hasVROCController(tt.pcie)
		if got != tt.want {
			t.Fatalf("%s: got %v want %v", tt.name, got, tt.want)
		}
	}
}