EvoBGP/internal/pipeline/refresh_aggregate_test.go

package pipeline

import (
	"context"
	"net/http"
	"testing"

	"evobgp/internal/store"
)

func TestRefreshModule_AggregatesAllEnabledModules(t *testing.T) {
	m := store.NewMemory()
	m.SeedDemo()
	tenant, _, modIP, _, _ := m.DemoIDs()

	mod2, err := m.CreateModule(tenant, &store.Module{Type: "IP_RANGES", Name: "extra-ip", Enabled: true, Priority: 30})
	if err != nil {
		t.Fatal(err)
	}
	if _, err := m.CreateIPRangeEntry(tenant, modIP, &store.IPRangeEntry{Prefix: "10.0.0.0/24"}); err != nil {
		t.Fatal(err)
	}
	if _, err := m.CreateIPRangeEntry(tenant, mod2.ID, &store.IPRangeEntry{Prefix: "192.168.0.0/24"}); err != nil {
		t.Fatal(err)
	}

	ctx := context.Background()
	if _, err := RefreshModule(ctx, m, http.DefaultClient, tenant, modIP); err != nil {
		t.Fatal(err)
	}
	revs, _, _ := m.ListRevisions(tenant, modIP, "", 1)
	if len(revs) == 0 {
		t.Fatal("no revision")
	}
	px, _, _ := m.ListRevisionPrefixes(tenant, revs[0].ID, "", 1000)
	if len(px) != 2 {
		t.Fatalf("first refresh: want 2 aggregated prefixes, got %d: %+v", len(px), px)
	}

	if _, err := RefreshModule(ctx, m, http.DefaultClient, tenant, mod2.ID); err != nil {
		t.Fatal(err)
	}
	revs2, _, _ := m.ListRevisions(tenant, mod2.ID, "", 1)
	if len(revs2) == 0 {
		t.Fatal("no revision for mod2")
	}
	px2, _, _ := m.ListRevisionPrefixes(tenant, revs2[0].ID, "", 1000)
	if len(px2) != 2 {
		t.Fatalf("second refresh: want 2 aggregated prefixes, got %d: %+v", len(px2), px2)
	}
}