Denozordec
f39df7c4bf
CI / changes (push) Successful in 8s
CI / commitlint (push) Has been skipped
CI / openapi (push) Successful in 25s
CI / web (push) Successful in 32s
CI / go (push) Successful in 57s
CI / bird2 (push) Successful in 15s
CI / release (push) Successful in 3m18s
Implemented new endpoints for estimating and pruning revisions, including detailed schemas for requests and responses. The `RevisionPruneEstimate` and `RevisionPruneResult` components were added to the OpenAPI documentation, enhancing the API's functionality for managing revision retention. Updated the backend to support these operations and integrated them into the tenant settings UI for improved user interaction.
84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"evobgp/internal/db"
|
|
"evobgp/internal/pipeline"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func TestPostgresPruneUnreferencedPrefixSnapshotsIntegration(t *testing.T) {
|
|
dsn := os.Getenv("EVOBGP_TEST_DATABASE_URL")
|
|
if dsn == "" {
|
|
t.Skip("EVOBGP_TEST_DATABASE_URL not set")
|
|
}
|
|
ctx := context.Background()
|
|
pool, err := db.OpenPostgresPool(ctx, dsn)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer pool.Close()
|
|
pg, err := NewPostgres(ctx, pool, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
snapID := uuid.NewString()
|
|
hash := normalizeSnapshotHash("sha256:test-orphan-" + snapID)
|
|
if _, err := pool.Exec(ctx, `INSERT INTO prefix_snapshot (id, content_hash) VALUES ($1::uuid, $2)`, snapID, hash); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := pool.Exec(ctx, `
|
|
INSERT INTO prefix_snapshot_row (snapshot_id, ord, prefix, source)
|
|
VALUES ($1::uuid, 0, '203.0.113.0/24', 'test')`, snapID); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
snaps, rows, err := pg.pruneUnreferencedPrefixSnapshots(ctx, 50)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if snaps < 1 || rows < 1 {
|
|
t.Fatalf("expected orphan cleanup, got snaps=%d rows=%d", snaps, rows)
|
|
}
|
|
|
|
var n int
|
|
if err := pool.QueryRow(ctx, `SELECT COUNT(*)::int FROM prefix_snapshot WHERE id = $1::uuid`, snapID).Scan(&n); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if n != 0 {
|
|
t.Fatalf("snapshot still exists")
|
|
}
|
|
}
|
|
|
|
func TestPostgresEstimateRevisionPruneIntegration(t *testing.T) {
|
|
dsn := os.Getenv("EVOBGP_TEST_DATABASE_URL")
|
|
if dsn == "" {
|
|
t.Skip("EVOBGP_TEST_DATABASE_URL not set")
|
|
}
|
|
ctx := context.Background()
|
|
pool, err := db.OpenPostgresPool(ctx, dsn)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer pool.Close()
|
|
pg, err := NewPostgres(ctx, pool, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
tenant := uuid.NewString()
|
|
cutoff := pipeline.RevisionCutoffFromMinutes(15)
|
|
est, err := pg.EstimateRevisionPrune(tenant, cutoff, 15)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if est.RevisionCount != 0 {
|
|
t.Fatalf("expected 0 revisions for empty tenant, got %d", est.RevisionCount)
|
|
}
|
|
}
|