38 lines
787 B
Go
38 lines
787 B
Go
package inventory
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestLoadAndFindNodeBranches(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, "inventory.yaml")
|
|
if err := os.WriteFile(path, []byte(`
|
|
classes:
|
|
- name: ${CLASS}
|
|
arch: arm64
|
|
os: armbian
|
|
image: file:///tmp/base.img
|
|
nodes:
|
|
- name: node1
|
|
class: ${CLASS}
|
|
hostname: node1
|
|
k3s_role: agent
|
|
`), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Setenv("CLASS", "rpi4")
|
|
inv, err := Load(path)
|
|
if err != nil {
|
|
t.Fatalf("Load: %v", err)
|
|
}
|
|
if _, _, err := inv.FindNode("missing"); err == nil {
|
|
t.Fatal("expected missing node error")
|
|
}
|
|
if _, cls, err := inv.FindNode("node1"); err != nil || cls == nil {
|
|
t.Fatalf("expected class lookup for node1, got class=%#v err=%v", cls, err)
|
|
}
|
|
}
|