EvoBGP/internal/pipeline/materialize_regression_test.go

package pipeline

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

	"evobgp/internal/birdfmt"
	"evobgp/internal/store"
)

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

	c1, err := m.CreateCommunity(tenant, &store.Community{Community: "65000:1", Title: "A", ValueJSON: "{}"})
	if err != nil {
		t.Fatal(err)
	}
	c2, err := m.CreateCommunity(tenant, &store.Community{Community: "65000:2", Title: "B", ValueJSON: "{}"})
	if err != nil {
		t.Fatal(err)
	}

	rows := []store.PrefixRow{
		{Prefix: "203.0.113.0/24", CommunityID: &c1.ID, Source: "ip_range"},
		{Prefix: "203.0.113.0/24", CommunityID: &c2.ID, Source: "cdn:test"},
	}
	out, err := buildPreviewFragments(m, tenant, moduleIP, "test-revision", rows)
	if err != nil {
		t.Fatal(err)
	}
	staticV4 := out[birdfmt.FragmentIncludePath(birdfmt.FragmentPrefixesV4)]
	if !strings.Contains(staticV4, "protocol static evobgp_prefixes_v4_"+communityProtocolSuffix(c1.ID)) {
		t.Fatalf("expected dedicated static protocol for first community, got:\n%s", staticV4)
	}
	if !strings.Contains(staticV4, "protocol static evobgp_prefixes_v4_"+communityProtocolSuffix(c2.ID)) {
		t.Fatalf("expected dedicated static protocol for second community, got:\n%s", staticV4)
	}
	out2, err := buildPreviewFragments(m, tenant, moduleIP, "test-revision", rows)
	if err != nil {
		t.Fatal(err)
	}
	staticV4Second := out2[birdfmt.FragmentIncludePath(birdfmt.FragmentPrefixesV4)]
	if staticV4 != staticV4Second {
		t.Fatalf("expected deterministic static preview text, got first:\n%s\nsecond:\n%s", staticV4, staticV4Second)
	}
}

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

	mod, err := m.CreateModule(tenant, &store.Module{
		Type:    "CDN_CIDRS",
		Name:    "cdn-refresh-force-full",
		Enabled: true,
	})
	if err != nil {
		t.Fatal(err)
	}

	var gotIfNoneMatch string
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		gotIfNoneMatch = strings.TrimSpace(r.Header.Get("If-None-Match"))
		w.Header().Set("ETag", "etag-new")
		_, _ = w.Write([]byte("198.51.100.0/24\n"))
	}))
	defer srv.Close()

	etag := "etag-old"
	if _, err := m.CreateCDNSource(tenant, mod.ID, &store.CDNSource{
		SourceKind: "txt",
		URL:        srv.URL,
		Etag:       etag,
	}); err != nil {
		t.Fatal(err)
	}

	collected, err := collectModulePrefixRows(context.Background(), m, srv.Client(), tenant, mod)
	if err != nil {
		t.Fatal(err)
	}
	if gotIfNoneMatch != "" {
		t.Fatalf("refresh path must not send If-None-Match, got %q", gotIfNoneMatch)
	}
	if len(collected) != 1 || collected[0].Prefix != "198.51.100.0/24" {
		t.Fatalf("unexpected collected rows: %+v", collected)
	}
}