spaxel/mothership/internal/api/simulator_test.go
jedarden c69b0f7ca1 test(bf-95tx): integration test tracing positions through simulator→registry→fusion
bf-95tx wants a single demonstrable verification that node positions flow
through every runtime stage the position-propagation work touched. bf-69ym
locked in simulator→fleet-registry; bf-u7ds locked in simulator→fusion. This
adds the missing end-to-end lock-in that ties both together in one run.

TestPositionPropagation_FullPipeline wires the production chain behind the
SimulatorHandler (handler → bridge.SyncToRegistry → adapter → fleet.Registry
SQLite, with adapter forwardPos → fleet.Manager.ForwardNodePosition → sink →
fusion.Engine.SetNodePosition), posts one node with an explicit position,
then reads that SAME position back from all three stages and asserts they
agree — at both AddNode and UpdateNode. A pipelineStages helper logs each
stage's value so `go test -v` visibly traces the value's path:

  [create] position propagation  simulator={1.5 2.5 1}  registry={1.5 2.5 1}  fusion={1.5 2.5 1}
  [update] position propagation  simulator={3.5 4.5 2}  registry={3.5 4.5 2}  fusion={3.5 4.5 2}

The update leg also asserts no stage still holds the stale pre-update value
and that a single node stays a single node at every stage (no orphan/dup).

Co-Authored-By: Claude <noreply@anthropic.com>
2026-07-06 13:13:28 -04:00

782 lines
27 KiB
Go

// Package api tests the simulator REST API
package api
import (
"bytes"
"context"
"encoding/json"
"fmt"
"net/http"
"net/http/httptest"
"path/filepath"
"sync"
"testing"
"github.com/go-chi/chi/v5"
"github.com/spaxel/mothership/internal/fleet"
"github.com/spaxel/mothership/internal/fusion"
"github.com/spaxel/mothership/internal/simulator"
)
// TestAddNode_SpreadsDefaultOrigin tests that nodes created without explicit
// positions (at origin {0,0,0}) are assigned spread-out positions instead of
// remaining co-located, which would collapse Fresnel excess path length toward
// 0 and prevent blob formation (bf-18yn, bf-4q5w).
func TestAddNode_SpreadsDefaultOrigin(t *testing.T) {
handler := NewSimulatorHandler(nil, nil, nil)
space := simulator.DefaultSpace()
handler.mu.Lock()
handler.space = space
handler.mu.Unlock()
// Create multiple nodes without positions (all at origin)
var createdNodes []simulator.Node
for i := 0; i < 4; i++ {
nodeJSON := map[string]interface{}{
"id": string(rune('a' + i)),
"name": "Node Name",
}
body, _ := json.Marshal(nodeJSON)
req := httptest.NewRequest("POST", "/api/simulator/nodes", bytes.NewReader(body))
req.Header.Set("Content-Type", "application/json")
w := httptest.NewRecorder()
handler.AddNode(w, req)
if w.Code != http.StatusCreated {
t.Fatalf("expected status %d, got %d", http.StatusCreated, w.Code)
}
var node simulator.Node
if err := json.Unmarshal(w.Body.Bytes(), &node); err != nil {
t.Fatalf("failed to decode response: %v", err)
}
createdNodes = append(createdNodes, node)
}
// Verify all nodes have distinct positions (not co-located at origin)
positions := make(map[[3]float64]int)
for _, node := range createdNodes {
pos := [3]float64{node.Position.X, node.Position.Y, node.Position.Z}
// Check not at origin
if node.Position.X == 0 && node.Position.Y == 0 && node.Position.Z == 0 {
t.Errorf("node %s still at origin (0,0,0)", node.ID)
}
// Check distinct
positions[pos]++
if positions[pos] > 1 {
t.Errorf("duplicate position %v for node %s", pos, node.ID)
}
}
// Verify positions span the room (not all at same point)
_, _, _, _, _, _ = space.Bounds()
var minXSeen, maxXSeen, minYSeen, maxYSeen float64
for i, node := range createdNodes {
if i == 0 {
minXSeen, maxXSeen = node.Position.X, node.Position.X
minYSeen, maxYSeen = node.Position.Y, node.Position.Y
}
if node.Position.X < minXSeen {
minXSeen = node.Position.X
}
if node.Position.X > maxXSeen {
maxXSeen = node.Position.X
}
if node.Position.Y < minYSeen {
minYSeen = node.Position.Y
}
if node.Position.Y > maxYSeen {
maxYSeen = node.Position.Y
}
}
// For 4 nodes, we should see some spread across the room
// (not strictly required for correctness, but verifies the fix works)
if minXSeen == maxXSeen && minYSeen == maxYSeen {
t.Errorf("nodes don't span room: all at (%.2f, %.2f)", minXSeen, minYSeen)
}
t.Logf("Successfully spread %d nodes across room: X=[%.2f,%.2f] Y=[%.2f,%.2f]",
len(createdNodes), minXSeen, maxXSeen, minYSeen, maxYSeen)
}
// TestAddNode_KeepsExplicitPosition tests that nodes with explicit positions
// are not modified.
func TestAddNode_KeepsExplicitPosition(t *testing.T) {
handler := NewSimulatorHandler(nil, nil, nil)
nodeJSON := map[string]interface{}{
"id": "test-node",
"position": map[string]interface{}{
"x": 1.5,
"y": 2.3,
"z": 1.0,
},
}
body, _ := json.Marshal(nodeJSON)
req := httptest.NewRequest("POST", "/api/simulator/nodes", bytes.NewReader(body))
req.Header.Set("Content-Type", "application/json")
w := httptest.NewRecorder()
handler.AddNode(w, req)
if w.Code != http.StatusCreated {
t.Fatalf("expected status %d, got %d", http.StatusCreated, w.Code)
}
var node simulator.Node
if err := json.Unmarshal(w.Body.Bytes(), &node); err != nil {
t.Fatalf("failed to decode response: %v", err)
}
// Verify explicit position was preserved
if node.Position.X != 1.5 || node.Position.Y != 2.3 || node.Position.Z != 1.0 {
t.Errorf("explicit position not preserved: got (%.2f, %.2f, %.2f)",
node.Position.X, node.Position.Y, node.Position.Z)
}
}
// recordingRegistry is a simulator.RegistryNodeAdapter that records the side
// effects of SyncToRegistry so tests can assert the simulator API drove an
// immediate write into the fleet registry (rather than only logging).
type recordingRegistry struct {
mu sync.Mutex
added []simulator.NodeRecord
posUpdates []simulator.NodeRecord
nodesByMAC map[string]simulator.NodeRecord
}
func newRecordingRegistry() *recordingRegistry {
return &recordingRegistry{nodesByMAC: make(map[string]simulator.NodeRecord)}
}
func (r *recordingRegistry) AddVirtualNode(mac, name string, x, y, z float64) error {
r.mu.Lock()
defer r.mu.Unlock()
rec := simulator.NodeRecord{MAC: mac, Name: name, PosX: x, PosY: y, PosZ: z, Virtual: true, Enabled: true}
r.nodesByMAC[mac] = rec
r.added = append(r.added, rec)
return nil
}
func (r *recordingRegistry) SetNodePosition(mac string, x, y, z float64) error {
r.mu.Lock()
defer r.mu.Unlock()
rec := r.nodesByMAC[mac]
rec.PosX, rec.PosY, rec.PosZ = x, y, z
r.nodesByMAC[mac] = rec
r.posUpdates = append(r.posUpdates, simulator.NodeRecord{MAC: mac, PosX: x, PosY: y, PosZ: z})
return nil
}
func (r *recordingRegistry) SetNodeRole(mac, role string) error { return nil }
func (r *recordingRegistry) DeleteNode(mac string) error {
r.mu.Lock()
defer r.mu.Unlock()
delete(r.nodesByMAC, mac)
return nil
}
func (r *recordingRegistry) GetNode(mac string) (*simulator.NodeRecord, error) {
r.mu.Lock()
defer r.mu.Unlock()
rec, ok := r.nodesByMAC[mac]
if !ok {
return nil, fmt.Errorf("not found: %s", mac)
}
return &rec, nil
}
func (r *recordingRegistry) GetAllNodes() ([]simulator.NodeRecord, error) {
r.mu.Lock()
defer r.mu.Unlock()
out := make([]simulator.NodeRecord, 0, len(r.nodesByMAC))
for _, rec := range r.nodesByMAC {
out = append(out, rec)
}
return out, nil
}
// newWiredHandler builds a SimulatorHandler backed by a real VirtualNodeStore
// (in a temp dir), a FleetRegistryBridge over that store, and the given adapter.
// This mirrors the wiring in cmd/mothership/main.go that drives immediate syncs.
func newWiredHandler(t *testing.T, adapter simulator.RegistryNodeAdapter) *SimulatorHandler {
t.Helper()
store, err := simulator.NewVirtualNodeStore(simulator.StoreConfig{
DataDir: filepath.Join(t.TempDir(), "simulator"),
Space: simulator.DefaultSpace(),
})
if err != nil {
t.Fatalf("create virtual store: %v", err)
}
t.Cleanup(func() { _ = store.Close() })
bridge := simulator.NewFleetRegistryBridge(store)
return NewSimulatorHandler(store, bridge, adapter)
}
// postNode issues a POST /api/simulator/nodes for the given node body.
func postNode(t *testing.T, h *SimulatorHandler, body map[string]interface{}) simulator.Node {
t.Helper()
raw, _ := json.Marshal(body)
req := httptest.NewRequest("POST", "/api/simulator/nodes", bytes.NewReader(raw))
req.Header.Set("Content-Type", "application/json")
w := httptest.NewRecorder()
h.AddNode(w, req)
if w.Code != http.StatusCreated {
t.Fatalf("AddNode status = %d, want %d (body: %s)", w.Code, http.StatusCreated, w.Body.String())
}
var node simulator.Node
if err := json.Unmarshal(w.Body.Bytes(), &node); err != nil {
t.Fatalf("decode node: %v", err)
}
return node
}
// putNode issues a PUT /api/simulator/nodes/{nodeID} with chi's URL param set.
func putNode(t *testing.T, h *SimulatorHandler, nodeID string, body map[string]interface{}) {
t.Helper()
raw, _ := json.Marshal(body)
req := httptest.NewRequest("PUT", "/api/simulator/nodes/"+nodeID, bytes.NewReader(raw))
req.Header.Set("Content-Type", "application/json")
rctx := chi.NewRouteContext()
rctx.URLParams.Add("nodeID", nodeID)
req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
w := httptest.NewRecorder()
h.UpdateNode(w, req)
if w.Code != http.StatusOK {
t.Fatalf("UpdateNode status = %d, want %d (body: %s)", w.Code, http.StatusOK, w.Body.String())
}
}
// TestImmediateSyncOnAddAndUpdate verifies the bf-5lii wiring: AddNode and
// UpdateNode must drive an immediate SyncToRegistry into the fleet registry
// instead of only logging that the node will sync on the next periodic tick.
func TestImmediateSyncOnAddAndUpdate(t *testing.T) {
cases := []struct {
name string
adapter simulator.RegistryNodeAdapter // nil exercises the no-op path
}{
{"wired adapter triggers immediate sync", newRecordingRegistry()},
{"nil adapter is a safe no-op", nil},
}
for _, c := range cases {
t.Run(c.name, func(t *testing.T) {
h := newWiredHandler(t, c.adapter)
rec, _ := c.adapter.(*recordingRegistry)
// AddNode: node must reach the registry synchronously, before any ticker.
node := postNode(t, h, map[string]interface{}{
"id": "node-A",
"name": "Node A",
"position": map[string]interface{}{"x": 1.0, "y": 1.0, "z": 1.0},
})
if rec == nil {
return // no-op path: nothing further to assert, just no panic
}
if len(rec.added) == 0 {
t.Fatalf("AddNode did not trigger an immediate SyncToRegistry: no AddVirtualNode calls recorded")
}
added := rec.added[len(rec.added)-1]
if added.Name != "Node A" {
t.Errorf("registry saw name %q, want %q", added.Name, "Node A")
}
if added.PosX != node.Position.X || added.PosY != node.Position.Y || added.PosZ != node.Position.Z {
t.Errorf("registry saw position (%.2f,%.2f,%.2f), want node position (%.2f,%.2f,%.2f)",
added.PosX, added.PosY, added.PosZ, node.Position.X, node.Position.Y, node.Position.Z)
}
addedBeforeUpdate := len(rec.posUpdates)
// UpdateNode: position change must push through immediately.
putNode(t, h, "node-A", map[string]interface{}{
"id": "node-A",
"name": "Node A",
"position": map[string]interface{}{"x": 2.5, "y": 3.0, "z": 1.5},
})
if len(rec.posUpdates) <= addedBeforeUpdate {
t.Fatalf("UpdateNode did not trigger an immediate SyncToRegistry: no new SetNodePosition call")
}
upd := rec.posUpdates[len(rec.posUpdates)-1]
if upd.MAC != added.MAC {
t.Errorf("position update MAC %q != add MAC %q", upd.MAC, added.MAC)
}
if upd.PosX != 2.5 || upd.PosY != 3.0 || upd.PosZ != 1.5 {
t.Errorf("position update saw (%.2f,%.2f,%.2f), want (2.50,3.00,1.50)", upd.PosX, upd.PosY, upd.PosZ)
}
})
}
}
// fleetRegistryTestAdapter mirrors cmd/mothership/main.go's fleetRegistryAdapter:
// it exposes a real, SQLite-backed *fleet.Registry through the
// simulator.RegistryNodeAdapter interface. The production adapter lives in
// package main (not importable from tests), so this re-creates the identical
// thin wrapper — write-side methods delegate directly, and GetNode/GetAllNodes
// convert fleet.NodeRecord -> simulator.NodeRecord (Enabled follows the fleet's
// role-based model: role "idle" means disabled). Routing through the real
// registry is what makes the tests below genuinely end-to-end: a position
// written by the bridge is round-tripped through SQLite, not merely recorded by
// a fake. (bf-69ym)
//
// bf-u7ds: AddVirtualNode/SetNodePosition also forward the new position through
// forwardPos (mirroring the production adapter) so a simulator-driven registry
// write reaches the blob-producing fusion engine at runtime, not only at
// startup seeding. Tests set forwardPos to fleet.Manager.ForwardNodePosition,
// exactly as cmd/mothership/main.go does.
type fleetRegistryTestAdapter struct {
reg *fleet.Registry
forwardPos func(mac string, x, y, z float64)
}
func (a *fleetRegistryTestAdapter) AddVirtualNode(mac, name string, x, y, z float64) error {
if err := a.reg.AddVirtualNode(mac, name, x, y, z); err != nil {
return err
}
if a.forwardPos != nil {
a.forwardPos(mac, x, y, z)
}
return nil
}
func (a *fleetRegistryTestAdapter) SetNodePosition(mac string, x, y, z float64) error {
if err := a.reg.SetNodePosition(mac, x, y, z); err != nil {
return err
}
if a.forwardPos != nil {
a.forwardPos(mac, x, y, z)
}
return nil
}
func (a *fleetRegistryTestAdapter) SetNodeRole(mac, role string) error {
return a.reg.SetNodeRole(mac, role)
}
func (a *fleetRegistryTestAdapter) DeleteNode(mac string) error {
return a.reg.DeleteNode(mac)
}
func (a *fleetRegistryTestAdapter) GetNode(mac string) (*simulator.NodeRecord, error) {
rec, err := a.reg.GetNode(mac)
if err != nil {
return nil, err
}
return fleetNodeRecordToSimulatorView(rec), nil
}
func (a *fleetRegistryTestAdapter) GetAllNodes() ([]simulator.NodeRecord, error) {
nodes, err := a.reg.GetAllNodes()
if err != nil {
return nil, err
}
out := make([]simulator.NodeRecord, 0, len(nodes))
for i := range nodes {
out = append(out, *fleetNodeRecordToSimulatorView(&nodes[i]))
}
return out, nil
}
// fleetNodeRecordToSimulatorView mirrors fleetNodeRecordToSimulator in
// cmd/mothership/main.go.
func fleetNodeRecordToSimulatorView(n *fleet.NodeRecord) *simulator.NodeRecord {
return &simulator.NodeRecord{
MAC: n.MAC,
Name: n.Name,
Role: n.Role,
PosX: n.PosX,
PosY: n.PosY,
PosZ: n.PosZ,
Virtual: n.Virtual,
Enabled: n.Role != "idle",
}
}
// newFleetRegistry opens a real fleet.Registry backed by a fresh temp SQLite
// DB, closing it automatically when the test ends.
func newFleetRegistry(t *testing.T) *fleet.Registry {
t.Helper()
reg, err := fleet.NewRegistry(filepath.Join(t.TempDir(), "fleet.db"))
if err != nil {
t.Fatalf("create fleet registry: %v", err)
}
t.Cleanup(func() { _ = reg.Close() })
return reg
}
// registryNodesByName indexes the live fleet registry's nodes by Name for
// assertion lookups. The bridge's virtualMAC is unexported, so tests key on the
// human-readable name they supplied when creating the node.
func registryNodesByName(t *testing.T, reg *fleet.Registry) map[string]*fleet.NodeRecord {
t.Helper()
nodes, err := reg.GetAllNodes()
if err != nil {
t.Fatalf("GetAllNodes: %v", err)
}
out := make(map[string]*fleet.NodeRecord, len(nodes))
for i := range nodes {
n := nodes[i]
out[n.Name] = &n
}
return out
}
// TestSimulatorHandlerToRegistry_CreateAndUpdate is the end-to-end lock-in for
// the simulator→fleet-registry position flow (bf-69ym, split from bf-4oiz). It
// wires a real SQLite-backed fleet.Registry behind the SimulatorHandler — the
// same wiring cmd/mothership/main.go uses — and drives the complete path:
//
// AddNode (simulator handler) -> VirtualNodeStore.CreateVirtualNode ->
// FleetRegistryBridge.SyncToRegistry -> adapter.AddVirtualNode -> fleet.Registry
//
// and then a position change via UpdateNode -> SetNodePosition. The create leg
// asserts the node lands in the live registry at the spread geometry
// DefaultNodePositions assigns (never the degenerate (0,0,0) / DB origin); the
// update leg asserts the new position flows through to the same registry row.
func TestSimulatorHandlerToRegistry_CreateAndUpdate(t *testing.T) {
reg := newFleetRegistry(t)
adapter := &fleetRegistryTestAdapter{reg: reg}
h := newWiredHandler(t, adapter) // real VirtualNodeStore + bridge + adapter
space := simulator.DefaultSpace()
// Create: post a node with no position. AddNode assigns it a spread slot
// from DefaultNodePositions and the immediate sync writes that into the live
// fleet registry (not just the in-memory NodeSet).
node := postNode(t, h, map[string]interface{}{
"id": "node-A",
"name": "Node A",
})
wantSpread := simulator.DefaultNodePositions(space, 1)
if len(wantSpread) != 1 {
t.Fatalf("DefaultNodePositions(1) returned %d slots, want 1", len(wantSpread))
}
wantPos := wantSpread[0]
byName := registryNodesByName(t, reg)
rec, ok := byName["Node A"]
if !ok {
t.Fatalf("node \"Node A\" never reached the fleet registry; have %d nodes", len(byName))
}
if !rec.Virtual {
t.Errorf("registry node not marked virtual: %+v", rec)
}
gotPos := simulator.Point{X: rec.PosX, Y: rec.PosY, Z: rec.PosZ}
if gotPos != wantPos {
t.Errorf("create: registry position %v != DefaultNodePositions spread %v", gotPos, wantPos)
}
if gotPos.X == 0 && gotPos.Y == 0 && gotPos.Z == 0 {
t.Errorf("create: registry node landed at (0,0,0)")
}
// The handler-returned node and the registry must agree.
if node.Position != wantPos {
t.Errorf("create: handler position %v != spread position %v", node.Position, wantPos)
}
// Update: change the position; the new value must reach the live registry
// via SetNodePosition, not just the in-memory NodeSet.
newPos := simulator.Point{X: 2.5, Y: 3.0, Z: 1.5}
putNode(t, h, "node-A", map[string]interface{}{
"id": "node-A",
"name": "Node A",
"position": map[string]interface{}{
"x": newPos.X, "y": newPos.Y, "z": newPos.Z,
},
})
byName = registryNodesByName(t, reg)
rec, ok = byName["Node A"]
if !ok {
t.Fatalf("node \"Node A\" disappeared from fleet registry after update")
}
upd := simulator.Point{X: rec.PosX, Y: rec.PosY, Z: rec.PosZ}
if upd != newPos {
t.Errorf("update: registry position %v != new position %v", upd, newPos)
}
// Still exactly one registry row — an update must not duplicate.
if len(byName) != 1 {
t.Errorf("update: expected 1 registry node, got %d (%+v)", len(byName), byName)
}
}
// singleEnginePosition returns the fusion engine's single registered node
// position, failing the test unless exactly one node is positioned. Used to
// assert what the accumulation grid will localize against.
func singleEnginePosition(t *testing.T, engine *fusion.Engine) simulator.Point {
t.Helper()
positions := engine.NodePositions()
if len(positions) != 1 {
t.Fatalf("expected exactly 1 fusion engine node position, got %d: %+v", len(positions), positions)
}
p := positions[0]
return simulator.Point{X: p.X, Y: p.Y, Z: p.Z}
}
// TestSimulatorHandlerToRegistry_PositionsReachFusionEngine (bf-u7ds) is the
// end-to-end lock-in for the third registry-write path: a simulator-driven node
// create/update must reach the blob-producing 3D fusion engine, not just the
// fleet DB. The production wiring (cmd/mothership/main.go) builds the adapter
// with forwardPos = fleetMgr.ForwardNodePosition, and the manager's
// nodePositionSink is wired to fusion.Engine.SetNodePosition; this test
// recreates that exact chain behind the SimulatorHandler and asserts the
// engine's nodePos mirror updates on both AddNode and UpdateNode.
//
// Without the forwardPos hook, SyncToRegistry lands the position in SQLite but
// the engine mirror stays stale and Fuse localizes against wrong/missing
// geometry — the regression this bead guards against. Mirrors the two bf-3p6g
// lock-ins (PATCH /position and OnNodeConnected) for the simulator path.
func TestSimulatorHandlerToRegistry_PositionsReachFusionEngine(t *testing.T) {
reg := newFleetRegistry(t)
mgr := fleet.NewManager(reg)
engine := fusion.NewEngine(nil)
mgr.SetNodePositionSink(func(mac string, x, y, z float64) {
engine.SetNodePosition(mac, x, y, z)
})
adapter := &fleetRegistryTestAdapter{
reg: reg,
forwardPos: mgr.ForwardNodePosition,
}
h := newWiredHandler(t, adapter)
space := simulator.DefaultSpace()
// Sanity: before any node is added the engine holds nothing to localize on.
if engine.NodeCount() != 0 {
t.Fatalf("engine should hold no positions before AddNode, got %d", engine.NodeCount())
}
// AddNode with no position: the bridge assigns spread geometry and the
// immediate sync must push it through to the engine's nodePos mirror.
postNode(t, h, map[string]interface{}{
"id": "node-A",
"name": "Node A",
})
wantSpread := simulator.DefaultNodePositions(space, 1)
if len(wantSpread) != 1 {
t.Fatalf("DefaultNodePositions(1) returned %d slots, want 1", len(wantSpread))
}
got := singleEnginePosition(t, engine)
if got != wantSpread[0] {
t.Fatalf("AddNode: fusion engine position %v != spread %v", got, wantSpread[0])
}
if got.X == 0 && got.Y == 0 && got.Z == 0 {
t.Fatalf("AddNode: fusion engine landed at (0,0,0) — engine mirror not seeded")
}
// UpdateNode with a new explicit position: the engine mirror must move with it.
newPos := simulator.Point{X: 2.5, Y: 3.0, Z: 1.5}
putNode(t, h, "node-A", map[string]interface{}{
"id": "node-A",
"name": "Node A",
"position": map[string]interface{}{
"x": newPos.X, "y": newPos.Y, "z": newPos.Z,
},
})
got = singleEnginePosition(t, engine)
if got != newPos {
t.Fatalf("UpdateNode: fusion engine position %v != new position %v", got, newPos)
}
// Still a single positioned node — an update must not orphan the old entry.
if engine.NodeCount() != 1 {
t.Errorf("UpdateNode: expected 1 fusion engine node, got %d", engine.NodeCount())
}
}
// TestSimulatorHandlerToRegistry_OriginNodesGetSpreadGeometry is the
// end-to-end analogue of the bridge-level
// TestSyncToRegistry_OriginNodesGetSpreadGeometry, but driven through the REST
// handler and persisted in SQLite. Nodes created at the DB default origin
// ({0,0,1}) must land in the live fleet registry at exactly the distinct,
// spread-out geometry DefaultNodePositions assigns — never co-located at the
// origin (which would collapse Fresnel excess path toward zero and prevent blob
// formation; bf-18yn / bf-4q5w).
func TestSimulatorHandlerToRegistry_OriginNodesGetSpreadGeometry(t *testing.T) {
const count = 4
reg := newFleetRegistry(t)
adapter := &fleetRegistryTestAdapter{reg: reg}
h := newWiredHandler(t, adapter)
space := simulator.DefaultSpace()
origin := simulator.Point{X: 0, Y: 0, Z: 1} // DefaultNodeOrigin
for _, id := range []string{"n-a", "n-b", "n-c", "n-d"} {
postNode(t, h, map[string]interface{}{
"id": id,
"name": id,
"position": map[string]interface{}{
"x": origin.X, "y": origin.Y, "z": origin.Z,
},
})
}
nodes, err := reg.GetAllNodes()
if err != nil {
t.Fatalf("GetAllNodes: %v", err)
}
if len(nodes) != count {
t.Fatalf("expected %d registry nodes, got %d", count, len(nodes))
}
want := simulator.DefaultNodePositions(space, count)
wantSet := make(map[simulator.Point]bool, len(want))
for _, p := range want {
wantSet[p] = true
}
gotSet := make(map[simulator.Point]bool, len(nodes))
for _, n := range nodes {
if !n.Virtual {
t.Errorf("registry node %s not marked virtual", n.MAC)
}
p := simulator.Point{X: n.PosX, Y: n.PosY, Z: n.PosZ}
if p == origin || (p.X == 0 && p.Y == 0 && p.Z == 0) {
t.Errorf("registry node %s still at origin/zero: %v", n.MAC, p)
}
if gotSet[p] {
t.Errorf("duplicate registry position %v", p)
}
gotSet[p] = true
}
for p := range gotSet {
if !wantSet[p] {
t.Errorf("registry position %v is not a DefaultNodePositions slot", p)
}
}
for _, p := range want {
if !gotSet[p] {
t.Errorf("expected DefaultNodePositions slot %v missing from registry", p)
}
}
}
// TestPositionPropagation_FullPipeline (bf-95tx) is the end-to-end lock-in for
// the COMPLETE simulator → fleet registry → fusion engine position path in a
// single test. One node position, posted once through the simulator REST API,
// must arrive — unchanged and in order — at every stage the runtime cares about:
//
// 1. simulator VirtualNodeStore — the source of truth for virtual nodes
// 2. fleet.Registry (SQLite) — persisted; read by the dashboard/REST
// 3. fusion.Engine nodePos mirror — what Fuse localizes against at runtime
//
// bf-69ym locked in stages 1→2; bf-u7ds locked in stages 1→3. This test ties
// them together: it traces ONE position value through all three stages in a
// single run and asserts they agree at both AddNode and UpdateNode, with
// per-stage logging so the full data path is observable under `go test -v`.
// That is the demonstrable full-pipeline verification the position-propagation
// bead (bf-95tx) requires.
func TestPositionPropagation_FullPipeline(t *testing.T) {
// Wire the full chain exactly as cmd/mothership/main.go does:
// handler → bridge.SyncToRegistry → adapter → fleet.Registry (SQLite),
// with the adapter's forwardPos → fleet.Manager.ForwardNodePosition →
// nodePositionSink → fusion.Engine.SetNodePosition.
reg := newFleetRegistry(t)
mgr := fleet.NewManager(reg)
engine := fusion.NewEngine(nil)
mgr.SetNodePositionSink(func(mac string, x, y, z float64) {
engine.SetNodePosition(mac, x, y, z)
})
adapter := &fleetRegistryTestAdapter{
reg: reg,
forwardPos: mgr.ForwardNodePosition,
}
h := newWiredHandler(t, adapter) // real VirtualNodeStore + bridge + adapter
// pipelineStages reads the posted node's position back from all three stages
// and logs each, so a `go test -v` run visibly traces the value's path:
// simulator → registry → fusion. nodeID keys the simulator store; name keys
// the fleet registry row (the bridge's virtualMAC is unexported); the single
// positioned node keys the fusion engine mirror.
pipelineStages := func(label, nodeID, name string) (simPos, regPos, fusPos simulator.Point) {
t.Helper()
// Stage 1 — simulator VirtualNodeStore (the source).
vs, err := h.virtualStore.GetNode(nodeID)
if err != nil {
t.Fatalf("%s: simulator store lookup %q: %v", label, nodeID, err)
}
simPos = simulator.Point{X: vs.Position.X, Y: vs.Position.Y, Z: vs.Position.Z}
// Stage 2 — fleet.Registry (SQLite), matched by Name.
byName := registryNodesByName(t, reg)
rec, ok := byName[name]
if !ok {
t.Fatalf("%s: node %q missing from fleet registry (have %d nodes)", label, name, len(byName))
}
regPos = simulator.Point{X: rec.PosX, Y: rec.PosY, Z: rec.PosZ}
// Stage 3 — fusion.Engine nodePos mirror (what Fuse localizes against).
fusPos = singleEnginePosition(t, engine)
t.Logf("[%s] position propagation simulator=%v registry=%v fusion=%v",
label, simPos, regPos, fusPos)
return simPos, regPos, fusPos
}
// --- Create leg: post one node with an explicit position. ---
const nodeID, nodeName = "node-A", "Node A"
createPos := simulator.Point{X: 1.5, Y: 2.5, Z: 1.0}
postNode(t, h, map[string]interface{}{
"id": nodeID,
"name": nodeName,
"position": map[string]interface{}{
"x": createPos.X, "y": createPos.Y, "z": createPos.Z,
},
})
simPos, regPos, fusPos := pipelineStages("create", nodeID, nodeName)
if simPos != createPos {
t.Errorf("create: simulator stage %v != posted %v", simPos, createPos)
}
if regPos != createPos {
t.Errorf("create: registry stage %v != posted %v", regPos, createPos)
}
if fusPos != createPos {
t.Errorf("create: fusion stage %v != posted %v", fusPos, createPos)
}
// --- Update leg: move the node and re-trace the full pipeline. ---
updatePos := simulator.Point{X: 3.5, Y: 4.5, Z: 2.0}
putNode(t, h, nodeID, map[string]interface{}{
"id": nodeID,
"name": nodeName,
"position": map[string]interface{}{
"x": updatePos.X, "y": updatePos.Y, "z": updatePos.Z,
},
})
simPos, regPos, fusPos = pipelineStages("update", nodeID, nodeName)
if simPos != updatePos {
t.Errorf("update: simulator stage %v != posted %v", simPos, updatePos)
}
if regPos != updatePos {
t.Errorf("update: registry stage %v != posted %v", regPos, updatePos)
}
if fusPos != updatePos {
t.Errorf("update: fusion stage %v != posted %v", fusPos, updatePos)
}
// No stage may still hold the pre-update value after the position moved.
if simPos == createPos || regPos == createPos || fusPos == createPos {
t.Errorf("update: a stage still holds the stale pre-update position %v "+
"(sim=%v reg=%v fus=%v)", createPos, simPos, regPos, fusPos)
}
// Cross-stage consistency: a single node must stay a single node at every
// stage — an update must not orphan or duplicate it anywhere in the path.
if got := len(registryNodesByName(t, reg)); got != 1 {
t.Errorf("update: expected 1 fleet registry node, got %d", got)
}
if got := engine.NodeCount(); got != 1 {
t.Errorf("update: expected 1 fusion engine node, got %d", got)
}
}