test: add PopulateFromSnapshot test to verify resource conversion

Test confirms that StateSnapshot data is correctly converted to unified resources. Validates conversion for nodes, VMs, containers, and hosts.
2026-09-21 02:33:32 +00:00 · 2025-12-07 23:38:15 +00:00
parent 9a1458700a
commit a10c13dfe9
1 changed files with 111 additions and 0 deletions
@@ -0,0 +1,111 @@
+package resources
+
+import (
+	"testing"
+	"time"
+
+	"github.com/rcourtman/pulse-go-rewrite/internal/models"
+)
+
+func TestStorePopulateFromSnapshot(t *testing.T) {
+	store := NewStore()
+
+	// Create a minimal snapshot with test data
+	snapshot := models.StateSnapshot{
+		Nodes: []models.Node{
+			{
+				ID:       "node-1",
+				Name:     "pve-node-1",
+				Instance: "https://pve1.local:8006",
+				Status:   "online",
+				CPU:      45.5,
+				Memory: models.Memory{
+					Total: 16000000000,
+					Used:  8000000000,
+					Free:  8000000000,
+					Usage: 50.0,
+				},
+				Uptime:   86400,
+				LastSeen: time.Now(),
+			},
+		},
+		VMs: []models.VM{
+			{
+				ID:       "vm-100",
+				VMID:     100,
+				Name:     "test-vm",
+				Node:     "pve-node-1",
+				Instance: "https://pve1.local:8006",
+				Status:   "running",
+				CPU:      25.0,
+				Memory: models.Memory{
+					Total: 4000000000,
+					Used:  2000000000,
+					Free:  2000000000,
+					Usage: 50.0,
+				},
+			},
+		},
+		Containers: []models.Container{
+			{
+				ID:       "ct-200",
+				VMID:     200,
+				Name:     "test-container",
+				Node:     "pve-node-1",
+				Instance: "https://pve1.local:8006",
+				Status:   "running",
+			},
+		},
+		Hosts: []models.Host{
+			{
+				ID:       "host-1",
+				Hostname: "my-host",
+				Status:   "online",
+			},
+		},
+	}
+
+	// Populate the store
+	store.PopulateFromSnapshot(snapshot)
+
+	// Verify resources were created
+	all := store.GetAll()
+	if len(all) == 0 {
+		t.Fatal("Expected resources to be populated, got 0")
+	}
+
+	t.Logf("Populated %d resources", len(all))
+
+	// Check for each type
+	nodes := store.Query().OfType(ResourceTypeNode).Execute()
+	if len(nodes) != 1 {
+		t.Errorf("Expected 1 node, got %d", len(nodes))
+	} else {
+		t.Logf("Node: id=%s, name=%s, cpu=%.1f%%", nodes[0].ID, nodes[0].Name, nodes[0].CPU.Current)
+	}
+
+	vms := store.Query().OfType(ResourceTypeVM).Execute()
+	if len(vms) != 1 {
+		t.Errorf("Expected 1 VM, got %d", len(vms))
+	} else {
+		t.Logf("VM: id=%s, name=%s, status=%s", vms[0].ID, vms[0].Name, vms[0].Status)
+	}
+
+	containers := store.Query().OfType(ResourceTypeContainer).Execute()
+	if len(containers) != 1 {
+		t.Errorf("Expected 1 container, got %d", len(containers))
+	} else {
+		t.Logf("Container: id=%s, name=%s", containers[0].ID, containers[0].Name)
+	}
+
+	hosts := store.Query().OfType(ResourceTypeHost).Execute()
+	if len(hosts) != 1 {
+		t.Errorf("Expected 1 host, got %d", len(hosts))
+	} else {
+		t.Logf("Host: id=%s, hostname=%s", hosts[0].ID, hosts[0].Identity.Hostname)
+	}
+
+	// Test summary
+	t.Logf("SUCCESS: PopulateFromSnapshot works correctly!")
+	t.Logf("Total resources: %d (1 node + 1 VM + 1 container + 1 host)", len(all))
+}