66 lines
2.1 KiB
Go
66 lines
2.1 KiB
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)
|
|
}
|
|
}
|