Function TestChildTrackerBasicOperations

server/parent_test.go:102–177

(t *testing.T)

	}
	if cfg2.Users["admin"] != "supersecret" {
		t.Error("Password not properly serialized/deserialized")
	}
}

func TestChildTrackerBasicOperations(t *testing.T) {
	tracker := NewChildTracker()

	// Initially empty
	if tracker.Count() != 0 {
		t.Errorf("Expected 0 children, got %d", tracker.Count())
	}

	// Add a child
	child1 := &ChildProcess{
		PID:        1001,
		Username:   "user1",
		RemoteAddr: "192.168.1.1:1111",
		BackendKey: BackendKey{Pid: 1001, SecretKey: 111},
		StartTime:  time.Now(),
		done:       make(chan struct{}),
	}
	tracker.Add(child1)

	if tracker.Count() != 1 {
		t.Errorf("Expected 1 child, got %d", tracker.Count())
	}

	// Get by PID
	got := tracker.Get(1001)
	if got == nil {
		t.Fatal("Expected to find child by PID")
		return
	} else if got.Username != "user1" {
		t.Errorf("Username mismatch: got %q, want %q", got.Username, "user1")
	}

	// Add another child
	child2 := &ChildProcess{
		PID:        1002,
		Username:   "user2",
		RemoteAddr: "192.168.1.2:2222",
		BackendKey: BackendKey{Pid: 1002, SecretKey: 222},
		StartTime:  time.Now(),
		done:       make(chan struct{}),
	}
	tracker.Add(child2)

	if tracker.Count() != 2 {
		t.Errorf("Expected 2 children, got %d", tracker.Count())
	}

	// Find by backend key
	found := tracker.FindByBackendKey(BackendKey{Pid: 1002, SecretKey: 222})
	if found == nil {
		t.Fatal("Expected to find child by backend key")
		return
	} else if found.PID != 1002 {
		t.Errorf("PID mismatch: got %d, want %d", found.PID, 1002)