layer1: add namespace manager skeleton step 8

2026-04-26 10:00:45 +03:00
parent ac2638f17d
commit 114d5b99af
3 changed files with 264 additions and 0 deletions
@@ -0,0 +1,96 @@
+package utils
+
+import (
+	"reflect"
+	"testing"
+	"time"
+)
+
+func TestNamespaceManagerSnapshotAndGet(t *testing.T) {
+	manager := NewNamespaceManager()
+	now := time.Now().UTC()
+	manager.Upsert(NamespaceEvent{Type: NamespaceEventAdd, Name: "tenant-b", Source: NamespaceSourceWatcher, ObservedAt: now})
+	manager.Upsert(NamespaceEvent{Type: NamespaceEventAdd, Name: "tenant-a", Source: NamespaceSourceWatcher, ObservedAt: now})
+
+	expected := []string{"tenant-a", "tenant-b"}
+	if !reflect.DeepEqual(expected, manager.Snapshot()) {
+		t.Fatalf("expected snapshot %v, got %v", expected, manager.Snapshot())
+	}
+
+	record, ok := manager.Get("tenant-a")
+	if !ok {
+		t.Fatalf("expected tenant-a to exist")
+	}
+	if record.Name != "tenant-a" || record.Phase != NamespacePhaseDiscovered {
+		t.Fatalf("unexpected record returned: %+v", record)
+	}
+}
+
+func TestNamespaceManagerUpsertIncrementsGeneration(t *testing.T) {
+	manager := NewNamespaceManager()
+	now := time.Now().UTC()
+
+	first := manager.Upsert(NamespaceEvent{Type: NamespaceEventAdd, Name: "tenant-a", Source: NamespaceSourceWatcher, ObservedAt: now})
+	second := manager.Upsert(NamespaceEvent{Type: NamespaceEventUpdate, Name: "tenant-a", Source: NamespaceSourceWatcher, ObservedAt: now.Add(time.Second)})
+
+	if first.Generation != 1 {
+		t.Fatalf("expected first generation to be 1, got %d", first.Generation)
+	}
+	if second.Generation != 2 {
+		t.Fatalf("expected second generation to be 2, got %d", second.Generation)
+	}
+}
+
+func TestNamespaceManagerMarkPartState(t *testing.T) {
+	manager := NewNamespaceManager()
+	manager.Upsert(NamespaceEvent{Type: NamespaceEventAdd, Name: "tenant-a", Source: NamespaceSourceWatcher})
+
+	record, ok := manager.MarkPartState("tenant-a", "router", NamespacePartState{State: "active"})
+	if !ok {
+		t.Fatalf("expected mark part state to succeed")
+	}
+	if record.RegisteredParts["router"].State != "active" {
+		t.Fatalf("expected router part state to be stored")
+	}
+}
+
+func TestNamespaceManagerRemove(t *testing.T) {
+	manager := NewNamespaceManager()
+	manager.Upsert(NamespaceEvent{Type: NamespaceEventAdd, Name: "tenant-a", Source: NamespaceSourceWatcher})
+
+	if !manager.Remove("tenant-a") {
+		t.Fatalf("expected remove to succeed")
+	}
+	if _, ok := manager.Get("tenant-a"); ok {
+		t.Fatalf("expected record to be removed")
+	}
+	if manager.Remove("tenant-a") {
+		t.Fatalf("expected second remove to report false")
+	}
+}
+
+func TestNamespaceManagerSnapshotRecordsReturnsCopies(t *testing.T) {
+	manager := NewNamespaceManager()
+	manager.Upsert(NamespaceEvent{
+		Type:   NamespaceEventAdd,
+		Name:   "tenant-a",
+		Source: NamespaceSourceWatcher,
+		Labels: map[string]string{"fission.io/managed": "true"},
+	})
+	_, _ = manager.MarkPartState("tenant-a", "router", NamespacePartState{State: "active"})
+
+	records := manager.SnapshotRecords()
+	records[0].Labels["fission.io/managed"] = "false"
+	records[0].RegisteredParts["router"] = NamespacePartState{State: "failed"}
+
+	record, ok := manager.Get("tenant-a")
+	if !ok {
+		t.Fatalf("expected tenant-a to exist")
+	}
+	if record.Labels["fission.io/managed"] != "true" {
+		t.Fatalf("expected snapshot records to be detached copies")
+	}
+	if record.RegisteredParts["router"].State != "active" {
+		t.Fatalf("expected part states to be detached copies")
+	}
+}