42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func TestAliasRepository_EnsureSystemAliases(t *testing.T) {
|
|
dbURL := os.Getenv("TEST_DATABASE_URL")
|
|
if dbURL == "" {
|
|
t.Skip("TEST_DATABASE_URL not set - skipping Postgres integration test")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
pool, err := NewPool(ctx, dbURL)
|
|
if err != nil {
|
|
t.Fatalf("connect pool: %v", err)
|
|
}
|
|
defer pool.Close()
|
|
|
|
domainID := uuid.New()
|
|
if _, err := pool.Exec(ctx, `INSERT INTO domains (id, name, punycode_name) VALUES ($1, 'dnsalias.test', 'dnsalias.test')`, domainID); err != nil {
|
|
t.Fatalf("seed domain: %v", err)
|
|
}
|
|
defer pool.Exec(ctx, `DELETE FROM domains WHERE id = $1`, domainID)
|
|
|
|
repo := NewAliasRepository(pool)
|
|
|
|
if err := repo.EnsureSystemAliases(ctx, "dnsalias.test", "admin@dnsalias.test"); err != nil {
|
|
t.Fatalf("EnsureSystemAliases: %v", err)
|
|
}
|
|
|
|
// Verify postmaster, abuse, dmarc, tlsrpt aliases exist
|
|
var count int
|
|
if err := pool.QueryRow(ctx, `SELECT count(*) FROM aliases WHERE domain_id = $1 AND is_system = true`, domainID).Scan(&count); err != nil || count != 4 {
|
|
t.Fatalf("expected 4 system aliases in DB, got count=%d, err=%v", count, err)
|
|
}
|
|
}
|