Files
EvoBGP/internal/repository/module_snapshot_context_integration_test.go
DenozordecandCursor f8e0520817 feat(httpapi): harden control plane and speed up prefix refresh
Очередь job получила lease и fencing, телеметрия и прогресс deploy сохраняются, ревизия пишет manifest входов. Обновление префиксов не переписывает неизменённый снимок, большие наборы идут через COPY, а устаревшая ревизия не попадает в deploy. Бандл ноды кэшируется, prune и постановка job больше не держат горячий путь.

Co-authored-by: Cursor <[email protected]>
2026-11-02 17:23:59 +07:00

45 lines
1.6 KiB
Go

package repository
import (
"context"
"errors"
"testing"
"evobgp/internal/store"
"github.com/google/uuid"
)
func TestPostgresModuleSnapshotMethodsRespectCancellationIntegration(t *testing.T) {
p, tenant := queueIntegration(t)
module := uuid.NewString()
if _, err := p.pool.Exec(context.Background(), `INSERT INTO module(id,tenant_id,type,name) VALUES($1,$2,'CDN_CIDRS','cancel-snapshot')`, module, tenant); err != nil {
t.Fatal(err)
}
ctx, cancel := context.WithCancel(context.Background())
cancel()
if _, _, err := p.GetModulePrefixSnapshot(ctx, tenant, module); !errors.Is(err, context.Canceled) {
t.Fatalf("get: %v", err)
}
if err := p.SetModulePrefixSnapshot(ctx, tenant, module, "hash", []store.PrefixRow{{Prefix: "192.0.2.0/24"}}); !errors.Is(err, context.Canceled) {
t.Fatalf("set: %v", err)
}
if err := p.DeleteModulePrefixSnapshot(ctx, tenant, module); !errors.Is(err, context.Canceled) {
t.Fatalf("delete: %v", err)
}
if err := p.SetModuleInputHash(ctx, tenant, module, "hash"); !errors.Is(err, context.Canceled) {
t.Fatalf("input hash: %v", err)
}
rev := uuid.NewString()
err := p.CreateRenderRevision(ctx, rev, tenant, module, nil, "cancel-hash", map[string]string{"bird.conf": "router id 192.0.2.1;\n"}, []store.PrefixRow{{Prefix: "203.0.113.0/24", Source: "test"}})
if !errors.Is(err, context.Canceled) {
t.Fatalf("revision: %v", err)
}
var n int
if err := p.pool.QueryRow(context.Background(), `SELECT count(*)::int FROM config_revision WHERE id=$1::uuid`, rev).Scan(&n); err != nil {
t.Fatal(err)
}
if n != 0 {
t.Fatal("cancelled revision write committed")
}
}