EvoBGP/internal/pipeline/asn_cache_test.go

package pipeline

import (
	"context"
	"net/http"
	"net/http/httptest"
	"strings"
	"sync/atomic"
	"testing"
	"time"

	"evobgp/internal/store"
)

func TestResolveASNForEntry_UsesTTLCache(t *testing.T) {
	m := store.NewMemory()
	t.Setenv("EVOBGP_ASN_CACHE_TTL_SEC", "3600")

	var calls atomic.Int32
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		calls.Add(1)
		if strings.Contains(r.URL.Path, "/announced") {
			_, _ = w.Write([]byte(`{"status":"ok","data":{"prefixes":[{"prefix":"203.0.113.0/24"}]}}`))
			return
		}
		_, _ = w.Write([]byte(`{"status":"ok","data":{"holder":"Test AS"}}`))
	}))
	defer srv.Close()
	t.Setenv("EVOBGP_RIPESTAT_ANNOUNCED_PREFIXES_URL", srv.URL+"/announced")
	t.Setenv("EVOBGP_RIPESTAT_AS_OVERVIEW_URL", srv.URL+"/overview")

	ctx := context.Background()
	hc := srv.Client()
	p1, h1, err := resolveASNForEntry(ctx, m, hc, 64512)
	if err != nil {
		t.Fatal(err)
	}
	if len(p1) != 1 || h1 != "Test AS" {
		t.Fatalf("unexpected first resolve: %+v holder=%q", p1, h1)
	}
	if calls.Load() < 2 {
		t.Fatalf("expected ripestat calls on first resolve, got %d", calls.Load())
	}
	firstCalls := calls.Load()
	_, _, err = resolveASNForEntry(ctx, m, hc, 64512)
	if err != nil {
		t.Fatal(err)
	}
	if calls.Load() != firstCalls {
		t.Fatalf("expected cache hit (no new HTTP), calls went from %d to %d", firstCalls, calls.Load())
	}
}

func TestModuleDueForScheduler_BucketRollover(t *testing.T) {
	mod := &store.Module{Enabled: true, Type: "CDN_CIDRS", RefreshIntervalSec: 300}
	boundary := time.Unix(300, 0)
	if !ModuleDueForScheduler(mod, boundary) {
		t.Fatal("expected due when refresh bucket rolls")
	}
	mid := time.Unix(330, 0)
	if ModuleDueForScheduler(mod, mid) {
		t.Fatal("expected not due within same bucket")
	}
}