metis/pkg/facts/aggregate_test.go

package facts

import (
	"testing"

	"metis/pkg/inventory"
)

func TestAggregateGroupsByClass(t *testing.T) {
	inv := &inventory.Inventory{
		Classes: []inventory.NodeClass{{Name: "c1"}, {Name: "c2"}},
		Nodes: []inventory.NodeSpec{
			{Name: "n1", Class: "c1", USBScratch: &inventory.USBScratchDisk{Mountpoint: "/mnt/scratch", Label: "scratch-1", BindTargets: []string{"/var/lib/rancher"}}},
			{Name: "n2", Class: "c2"},
		},
	}
	snaps := []Snapshot{
		{Hostname: "n1", Kernel: "k1", PackageSample: map[string]string{"containerd": "2.0"}, USBScratch: &USBScratch{Mountpoint: "/mnt/scratch", Label: "scratch-1", MountHealthy: true, LabelHealthy: true, BindHealthy: true, BindTargets: []USBBindTarget{{Path: "/var/lib/rancher", Healthy: true}}}},
		{Hostname: "n2", Kernel: "k2", PackageSample: map[string]string{"containerd": "1.7"}},
		{Hostname: "n1", Kernel: "k1"},
	}
	sum := Aggregate(inv, snaps)
	if len(sum) != 2 {
		t.Fatalf("expected 2 classes, got %d", len(sum))
	}
	c1 := sum["c1"]
	if c1 == nil || c1.Kernels["k1"] != 2 {
		t.Fatalf("expected k1 count 2, got %#v", c1)
	}
	if c1.PackageStats["containerd"]["2.0"] != 1 {
		t.Fatalf("package stats not tallied: %#v", c1.PackageStats)
	}
	if c1.USBMountHealth["ok"] != 1 || c1.USBLabelHealth["ok"] != 1 || c1.USBBindHealth["ok"] != 1 {
		t.Fatalf("usb health not tallied: %#v", c1)
	}
}

func TestAggregateKeepsUnknownHostnames(t *testing.T) {
	sum := Aggregate(nil, []Snapshot{{Hostname: "ghost", Kernel: "k"}})
	if sum["unknown"].Nodes[0] != "ghost" {
		t.Fatalf("unexpected unknown aggregate: %#v", sum["unknown"])
	}
}

func TestChooseTargetsHandlesTiesAndEmptyValues(t *testing.T) {
	sum := &ClassSummary{
		Kernels: map[string]int{"k1": 2, "k2": 2},
		OSImages: map[string]int{
			"img": 1,
		},
		PackageStats: map[string]map[string]int{
			"p": {"": 3, "1": 1},
		},
	}
	targets := ChooseTargets(sum)
	if targets.Kernel != "" {
		t.Fatalf("expected kernel tie to return empty, got %q", targets.Kernel)
	}
	if targets.OSImage != "img" {
		t.Fatalf("expected OS image img, got %q", targets.OSImage)
	}
	if _, ok := targets.Packages["p"]; ok {
		t.Fatalf("expected empty package version to be skipped: %+v", targets.Packages)
	}
}