phase 1 review fixes: dedup edges, real export, validation, tests
Review of aa2ca0a found and fixed:
- re-ingest duplicated ALL edges (upsert conflicted on valid_from=now(),
never fired) — migration 007 dedupes + partial unique index on current
edges; upsert now targets it. Regression-tested.
- export was a stub that overwrote seeds/*.yaml with 11-byte "version: 1"
files — implemented real deterministic export (ontology/inventory/policy,
cognition-layer excluded); round-trip is byte-stable (tested)
- DB password leaked in startup logs (slog JSON bypasses String()) —
Config now implements slog.LogValuer; regression-tested
- docker-compose had literal '***' as DB password — env-interpolated
- uuid.New() (v4) → uuid.NewV7() per ADR-0005
- no ontology validation on ingest — internal/ontology TypeTree: abstract
instantiation rejected, relationship endpoints hierarchy-validated,
cardinality enforced in-transaction, lifecycle states checked, default
state applied (Phase 1 gate items, R3-1)
- getOrCreateEntityID swallowed non-ErrNoRows errors
- migration runner now holds a session advisory lock on one connection
- Makefile: hardcoded /opt/homebrew/bin/go → go; test-db target
Tests: 4 unit suites + 7 integration tests (env-guarded, throwaway DB per
run): migrate idempotent, seed idempotent + no dup edges, abstract/edge/
cardinality rejection, blast_radius cycle termination, export round-trip.
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
338
internal/db/integration_test.go
Normal file
338
internal/db/integration_test.go
Normal file
@@ -0,0 +1,338 @@
|
||||
package db
|
||||
|
||||
// Integration tests against a real TimescaleDB. Guarded by
|
||||
// OIKOS_TEST_DATABASE_URL — skipped when unset. Run with:
|
||||
//
|
||||
// docker compose up -d postgres
|
||||
// OIKOS_TEST_DATABASE_URL="postgres://oikos:oikos_dev@localhost:5432/oikos?sslmode=disable" go test ./internal/db/
|
||||
//
|
||||
// or `make test-db`. Each run creates a throwaway database and drops it.
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"math/rand"
|
||||
"os"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/dtoro/oikos/internal/domain"
|
||||
"github.com/jackc/pgx/v5"
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
func seedsDir() string { return "../../seeds" }
|
||||
|
||||
// newTestPool creates a throwaway database (dropped on cleanup), runs all
|
||||
// migrations, and returns a pool connected to it.
|
||||
func newTestPool(t *testing.T) *Pool {
|
||||
t.Helper()
|
||||
baseURL := os.Getenv("OIKOS_TEST_DATABASE_URL")
|
||||
if baseURL == "" {
|
||||
t.Skip("OIKOS_TEST_DATABASE_URL not set — skipping integration test")
|
||||
}
|
||||
ctx := context.Background()
|
||||
|
||||
admin, err := pgx.Connect(ctx, baseURL)
|
||||
if err != nil {
|
||||
t.Fatalf("connect admin: %v", err)
|
||||
}
|
||||
dbName := fmt.Sprintf("oikos_test_%08x", rand.Int63())
|
||||
if _, err := admin.Exec(ctx, "CREATE DATABASE "+dbName); err != nil {
|
||||
admin.Close(ctx)
|
||||
t.Fatalf("create test db: %v", err)
|
||||
}
|
||||
admin.Close(ctx)
|
||||
|
||||
testURL := swapDatabase(baseURL, dbName)
|
||||
pool, err := New(ctx, testURL)
|
||||
if err != nil {
|
||||
t.Fatalf("connect test db: %v", err)
|
||||
}
|
||||
t.Cleanup(func() {
|
||||
pool.Close()
|
||||
admin, err := pgx.Connect(ctx, baseURL)
|
||||
if err == nil {
|
||||
admin.Exec(ctx, "DROP DATABASE IF EXISTS "+dbName+" WITH (FORCE)")
|
||||
admin.Close(ctx)
|
||||
}
|
||||
})
|
||||
|
||||
if err := pool.Migrate(ctx); err != nil {
|
||||
t.Fatalf("migrate: %v", err)
|
||||
}
|
||||
return pool
|
||||
}
|
||||
|
||||
// swapDatabase replaces the database name in a postgres URL.
|
||||
func swapDatabase(url, db string) string {
|
||||
// postgres://user:pass@host:port/dbname?params
|
||||
qi := strings.Index(url, "?")
|
||||
params := ""
|
||||
base := url
|
||||
if qi >= 0 {
|
||||
base, params = url[:qi], url[qi:]
|
||||
}
|
||||
si := strings.LastIndex(base, "/")
|
||||
return base[:si+1] + db + params
|
||||
}
|
||||
|
||||
func seedAll(t *testing.T, pool *Pool, dir string) {
|
||||
t.Helper()
|
||||
for _, f := range []string{"ontology.yaml", "inventory.yaml", "policy.yaml"} {
|
||||
content, err := os.ReadFile(dir + "/" + f)
|
||||
if err != nil {
|
||||
t.Fatalf("read %s: %v", f, err)
|
||||
}
|
||||
ingestSeedContent(t, pool, f, content)
|
||||
}
|
||||
}
|
||||
|
||||
func ingestSeedContent(t *testing.T, pool *Pool, name string, content []byte) {
|
||||
t.Helper()
|
||||
ctx := context.Background()
|
||||
err := pool.SeedIngest(ctx, name, content,
|
||||
func(ctx context.Context, tx pgx.Tx, data map[string]any) error {
|
||||
var err error
|
||||
switch name {
|
||||
case "ontology.yaml":
|
||||
_, err = IngestOntologySeed(ctx, tx, data)
|
||||
case "inventory.yaml":
|
||||
_, err = IngestInventorySeed(ctx, tx, data)
|
||||
case "policy.yaml":
|
||||
_, err = IngestPolicySeed(ctx, tx, data)
|
||||
}
|
||||
return err
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("ingest %s: %v", name, err)
|
||||
}
|
||||
}
|
||||
|
||||
func count(t *testing.T, pool *Pool, query string) int {
|
||||
t.Helper()
|
||||
var n int
|
||||
if err := pool.QueryRow(context.Background(), query).Scan(&n); err != nil {
|
||||
t.Fatalf("count %q: %v", query, err)
|
||||
}
|
||||
return n
|
||||
}
|
||||
|
||||
func TestMigrateIdempotent(t *testing.T) {
|
||||
pool := newTestPool(t)
|
||||
// second run must be a clean no-op
|
||||
if err := pool.Migrate(context.Background()); err != nil {
|
||||
t.Fatalf("second migrate: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSeedIngestIdempotentAndNoDuplicateEdges(t *testing.T) {
|
||||
pool := newTestPool(t)
|
||||
seedAll(t, pool, seedsDir())
|
||||
|
||||
entities := count(t, pool, "SELECT count(*) FROM entities")
|
||||
edges := count(t, pool, "SELECT count(*) FROM relationships WHERE valid_to IS NULL")
|
||||
if entities == 0 || edges == 0 {
|
||||
t.Fatalf("seed produced empty graph: %d entities, %d edges", entities, edges)
|
||||
}
|
||||
|
||||
// Same content → hash no-op
|
||||
seedAll(t, pool, seedsDir())
|
||||
if got := count(t, pool, "SELECT count(*) FROM relationships WHERE valid_to IS NULL"); got != edges {
|
||||
t.Errorf("unchanged re-seed altered edges: %d → %d", edges, got)
|
||||
}
|
||||
|
||||
// Changed content (hash differs) → full re-ingest must NOT duplicate edges
|
||||
// (regression: the old upsert conflicted on valid_from and duplicated all
|
||||
// 144 edges on every re-ingest)
|
||||
content, err := os.ReadFile(seedsDir() + "/inventory.yaml")
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
touched := append(content, []byte("\n# touched for hash change\n")...)
|
||||
ingestSeedContent(t, pool, "inventory.yaml", touched)
|
||||
|
||||
if got := count(t, pool, "SELECT count(*) FROM relationships WHERE valid_to IS NULL"); got != edges {
|
||||
t.Errorf("touched re-seed duplicated edges: %d → %d", edges, got)
|
||||
}
|
||||
if dup := count(t, pool, `SELECT count(*) FROM (
|
||||
SELECT source_id, target_id, type FROM relationships
|
||||
WHERE valid_to IS NULL GROUP BY 1,2,3 HAVING count(*) > 1) d`); dup != 0 {
|
||||
t.Errorf("%d duplicated current edges", dup)
|
||||
}
|
||||
if got := count(t, pool, "SELECT count(*) FROM entities"); got != entities {
|
||||
t.Errorf("touched re-seed altered entity count: %d → %d", entities, got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestAbstractTypeRejected(t *testing.T) {
|
||||
pool := newTestPool(t)
|
||||
seedAll(t, pool, seedsDir())
|
||||
|
||||
bad := []byte(`
|
||||
version: 1
|
||||
entities:
|
||||
- {slug: "machine:ghost", type: machine, name: ghost}
|
||||
`)
|
||||
ctx := context.Background()
|
||||
err := pool.SeedIngest(ctx, "inventory.yaml", bad,
|
||||
func(ctx context.Context, tx pgx.Tx, data map[string]any) error {
|
||||
_, err := IngestInventorySeed(ctx, tx, data)
|
||||
return err
|
||||
})
|
||||
if !errors.Is(err, domain.ErrAbstractType) {
|
||||
t.Errorf("abstract instantiation = %v, want ErrAbstractType", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestEdgeEndpointValidation(t *testing.T) {
|
||||
pool := newTestPool(t)
|
||||
seedAll(t, pool, seedsDir())
|
||||
|
||||
// routes-to requires source ingress-route; a service source must fail
|
||||
bad := []byte(`
|
||||
version: 1
|
||||
relationships:
|
||||
- {source: "service:gitea", target: "service:caddy", type: routes-to}
|
||||
`)
|
||||
ctx := context.Background()
|
||||
err := pool.SeedIngest(ctx, "inventory.yaml", bad,
|
||||
func(ctx context.Context, tx pgx.Tx, data map[string]any) error {
|
||||
_, err := IngestInventorySeed(ctx, tx, data)
|
||||
return err
|
||||
})
|
||||
if !errors.Is(err, domain.ErrInvalidEdge) {
|
||||
t.Errorf("bad edge = %v, want ErrInvalidEdge", err)
|
||||
}
|
||||
|
||||
// hosts from a proxmox-host (is-a machine) to an lxc (is-a compute-entity)
|
||||
// must PASS via hierarchy walk — already covered by the seed itself, but
|
||||
// assert an explicit one for clarity
|
||||
good := []byte(`
|
||||
version: 1
|
||||
relationships:
|
||||
- {source: "host:strong", target: "lxc:jellyfin", type: hosts}
|
||||
`)
|
||||
err = pool.SeedIngest(ctx, "inventory.yaml", good,
|
||||
func(ctx context.Context, tx pgx.Tx, data map[string]any) error {
|
||||
_, err := IngestInventorySeed(ctx, tx, data)
|
||||
return err
|
||||
})
|
||||
if err != nil {
|
||||
t.Errorf("valid inherited edge rejected: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCardinalityEnforced(t *testing.T) {
|
||||
pool := newTestPool(t)
|
||||
seedAll(t, pool, seedsDir())
|
||||
|
||||
// routes-to is many-to-one: one ingress route cannot point at two services
|
||||
bad := []byte(`
|
||||
version: 1
|
||||
relationships:
|
||||
- {source: "ingress:git.hubris.network", target: "service:jellyfin", type: routes-to}
|
||||
`)
|
||||
ctx := context.Background()
|
||||
err := pool.SeedIngest(ctx, "inventory.yaml", bad,
|
||||
func(ctx context.Context, tx pgx.Tx, data map[string]any) error {
|
||||
_, err := IngestInventorySeed(ctx, tx, data)
|
||||
return err
|
||||
})
|
||||
if err == nil || !strings.Contains(err.Error(), "cardinality") {
|
||||
t.Errorf("cardinality violation = %v, want cardinality error", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBlastRadiusTerminatesOnCycles(t *testing.T) {
|
||||
pool := newTestPool(t)
|
||||
seedAll(t, pool, seedsDir())
|
||||
|
||||
// Build a dependency cycle: gitea → caddy → authentik → gitea
|
||||
cycle := []byte(`
|
||||
version: 1
|
||||
relationships:
|
||||
- {source: "service:gitea", target: "service:caddy", type: depends-on}
|
||||
- {source: "service:caddy", target: "service:authentik", type: depends-on}
|
||||
- {source: "service:authentik", target: "service:gitea", type: depends-on}
|
||||
`)
|
||||
ctx := context.Background()
|
||||
err := pool.SeedIngest(ctx, "inventory.yaml", cycle,
|
||||
func(ctx context.Context, tx pgx.Tx, data map[string]any) error {
|
||||
_, err := IngestInventorySeed(ctx, tx, data)
|
||||
return err
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("cycle ingest: %v", err)
|
||||
}
|
||||
|
||||
rows, err := pool.Query(ctx, `
|
||||
SELECT e.slug, b.depth
|
||||
FROM blast_radius((SELECT id FROM entities WHERE slug='service:gitea'), 5,
|
||||
ARRAY['depends-on']) b
|
||||
JOIN entities e ON e.id = b.entity_id ORDER BY b.depth`)
|
||||
if err != nil {
|
||||
t.Fatalf("blast_radius: %v", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
got := map[string]int{}
|
||||
for rows.Next() {
|
||||
var slug string
|
||||
var depth int
|
||||
if err := rows.Scan(&slug, &depth); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
got[slug] = depth
|
||||
}
|
||||
want := map[string]int{"service:gitea": 0, "service:caddy": 1, "service:authentik": 2}
|
||||
for slug, depth := range want {
|
||||
if got[slug] != depth {
|
||||
t.Errorf("blast_radius[%s] = %d, want %d (full: %v)", slug, got[slug], depth, got)
|
||||
}
|
||||
}
|
||||
if len(got) != len(want) {
|
||||
t.Errorf("blast_radius returned %d nodes, want %d: %v", len(got), len(want), got)
|
||||
}
|
||||
}
|
||||
|
||||
// TestExportRoundTripStable: export → ingest into a fresh DB → export again
|
||||
// must yield byte-identical YAML (the canonical-form fixpoint, plan D6).
|
||||
func TestExportRoundTripStable(t *testing.T) {
|
||||
pool := newTestPool(t)
|
||||
seedAll(t, pool, seedsDir())
|
||||
ctx := context.Background()
|
||||
|
||||
export1, err := ExportToYAML(ctx, pool)
|
||||
if err != nil {
|
||||
t.Fatalf("export 1: %v", err)
|
||||
}
|
||||
for name, content := range export1 {
|
||||
var doc map[string]any
|
||||
if err := yaml.Unmarshal(content, &doc); err != nil {
|
||||
t.Fatalf("export %s is not valid YAML: %v", name, err)
|
||||
}
|
||||
}
|
||||
|
||||
pool2 := newTestPool(t)
|
||||
for _, name := range []string{"ontology.yaml", "inventory.yaml", "policy.yaml"} {
|
||||
ingestSeedContent(t, pool2, name, export1[name])
|
||||
}
|
||||
export2, err := ExportToYAML(ctx, pool2)
|
||||
if err != nil {
|
||||
t.Fatalf("export 2: %v", err)
|
||||
}
|
||||
for _, name := range []string{"ontology.yaml", "inventory.yaml", "policy.yaml"} {
|
||||
if !bytes.Equal(export1[name], export2[name]) {
|
||||
t.Errorf("%s round-trip not byte-stable (len %d vs %d)",
|
||||
name, len(export1[name]), len(export2[name]))
|
||||
}
|
||||
}
|
||||
|
||||
// sanity: exported inventory carries the real graph, not a stub
|
||||
// (regression: export used to write 11-byte "version: 1" stubs)
|
||||
if len(export1["inventory.yaml"]) < 1000 {
|
||||
t.Errorf("inventory export suspiciously small: %d bytes", len(export1["inventory.yaml"]))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user