Files
wehub-resource-sync a06f331eb8
CI / benchmark (push) Has been skipped
install-script / posix-syntax (push) Successful in 6m1s
CI / build-onnx (push) Failing after 6m43s
init-smoke / dry-run (push) Failing after 15m57s
security / govulncheck (push) Has been cancelled
security / trivy-fs (push) Has been cancelled
CI / test (1.26, ubuntu-latest) (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
CI / test (1.26, macos-latest) (push) Has been cancelled
CI / build-windows (push) Has been cancelled
CI / lint (push) Has been cancelled
install-script / powershell-syntax (push) Has been cancelled
install-script / install (macos-14) (push) Has been cancelled
install-script / install (ubuntu-latest) (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 12:33:42 +08:00

70 lines
2.4 KiB
Go

package analysis
import (
"testing"
"github.com/stretchr/testify/require"
"github.com/zzet/gortex/internal/config"
"github.com/zzet/gortex/internal/graph"
)
// buildEventGraph wires one producer that emits topic "orders" and, optionally,
// a consumer that listens on it.
func buildEventGraph(withConsumer bool) (*graph.Graph, string) {
g := graph.New()
g.AddNode(&graph.Node{ID: "svc/pub.go::Publish", Name: "Publish", Kind: graph.KindFunction, FilePath: "svc/pub.go"})
g.AddNode(&graph.Node{ID: "topic::orders", Name: "orders", Kind: graph.KindEvent, FilePath: "svc/pub.go"})
g.AddEdge(&graph.Edge{From: "svc/pub.go::Publish", To: "topic::orders", Kind: graph.EdgeEmits})
if withConsumer {
g.AddNode(&graph.Node{ID: "worker/sub.go::Consume", Name: "Consume", Kind: graph.KindFunction, FilePath: "worker/sub.go"})
g.AddEdge(&graph.Edge{From: "worker/sub.go::Consume", To: "topic::orders", Kind: graph.EdgeListensOn})
}
return g, "svc/pub.go::Publish"
}
func TestEventBoundaryProducerPath(t *testing.T) {
g, pubID := buildEventGraph(true)
fam := EventBoundaryFamily{Rules: []config.EventRule{{
Name: "orders-producer", Topic: "orders",
Producer: "ingest/**", Severity: "error",
}}}
v := fam.Evaluate(g, []string{pubID})
require.Len(t, v, 1)
require.Equal(t, "event_boundary", v[0].Kind)
require.Equal(t, "error", v[0].Severity)
}
func TestEventBoundaryRequireConsumer(t *testing.T) {
// No consumer present -> require_consumer fires.
g, pubID := buildEventGraph(false)
fam := EventBoundaryFamily{Rules: []config.EventRule{{
Name: "orders-needs-consumer", Topic: "orders", RequireConsumer: true,
}}}
require.Len(t, fam.Evaluate(g, []string{pubID}), 1)
// With a consumer present -> no violation.
g2, pubID2 := buildEventGraph(true)
require.Empty(t, fam.Evaluate(g2, []string{pubID2}))
}
func TestEventBoundaryForbid(t *testing.T) {
g, pubID := buildEventGraph(true)
fam := EventBoundaryFamily{Rules: []config.EventRule{{
Name: "no-pub-from-svc", Topic: "*",
Forbid: []string{"svc/**"},
}}}
v := fam.Evaluate(g, []string{pubID})
require.Len(t, v, 1)
require.Equal(t, "warn", v[0].Severity) // default severity
}
func TestEventBoundaryAllowedProducerClean(t *testing.T) {
g, pubID := buildEventGraph(true)
fam := EventBoundaryFamily{Rules: []config.EventRule{{
Name: "orders-producer", Topic: "orders",
Producer: "svc/**", // pub.go is in svc/, so it's allowed
}}}
require.Empty(t, fam.Evaluate(g, []string{pubID}))
}