toki
b4c6550eab
- Add node store implementation for edge app - Add adapters factory for node app - Update edge and node transport layers - Update domain rules for edge and node - Add bin scripts for edge and node - Update configs and documentation - Add agent-task node_centralized_mgmt directory
38 lines
853 B
Go
38 lines
853 B
Go
package node_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
toki "git.toki-labs.com/toki/common-proto-socket/go"
|
|
|
|
edgenode "iop/apps/edge/internal/node"
|
|
)
|
|
|
|
func TestRegistry_RegisterAndCount(t *testing.T) {
|
|
reg := edgenode.NewRegistry()
|
|
entry := &edgenode.NodeEntry{
|
|
NodeID: "node-001",
|
|
Alias: "local-node",
|
|
Client: &toki.TcpClient{},
|
|
}
|
|
reg.Register(entry)
|
|
if reg.Count() != 1 {
|
|
t.Fatalf("expected 1 node, got %d", reg.Count())
|
|
}
|
|
}
|
|
|
|
func TestRegistry_Pick_Empty(t *testing.T) {
|
|
reg := edgenode.NewRegistry()
|
|
if _, err := reg.Pick(); err == nil {
|
|
t.Fatal("expected error on empty registry")
|
|
}
|
|
}
|
|
|
|
func TestRegistry_Unregister(t *testing.T) {
|
|
reg := edgenode.NewRegistry()
|
|
reg.Register(&edgenode.NodeEntry{NodeID: "node-001"})
|
|
reg.Unregister("node-001")
|
|
if reg.Count() != 0 {
|
|
t.Fatalf("expected 0 nodes after unregister, got %d", reg.Count())
|
|
}
|
|
}
|