package api

import (
	"testing"

	"crypto-miner-server/internal/db"
	"crypto-miner-server/internal/models"
)

func TestBuildSpreadRouterInputErasureFlag(t *testing.T) {
	database, err := db.New(t.TempDir())
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { _ = database.Close() })

	hub := NewWSHub(database)
	hub.SetServerPolicy(ServerPolicy{ErasureLanesEnabled: true})
	if err := database.UpsertAgent(&models.Agent{ID: "router-seed", Name: "Seed", IP: "10.9.8.9", Status: "online"}); err != nil {
		t.Fatal(err)
	}
	connectTestAgent(t, hub, "router-seed")
	waitForHubAgents(t, hub, "router-seed")

	in := buildSpreadRouterInput(hub, nil, []string{"10.9.8"}, "do_peer")
	if !in.ErasureLanesEnabled {
		t.Fatal("expected erasure_lanes_enabled on spread router input")
	}
}