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

79 lines
3.0 KiB
Go

package resolver
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/zzet/gortex/internal/graph"
)
func mediatrHandler(g *graph.Graph, id, file, reqType, kind string) {
g.AddNode(&graph.Node{ID: id, Kind: graph.KindMethod, Name: "Handle", FilePath: file, Language: "csharp",
Meta: map[string]any{"mediatr_request_type": reqType, "mediatr_kind": kind}})
}
func mediatrSend(g *graph.Graph, fromID, file, reqType, kind string) {
if g.GetNode(fromID) == nil {
g.AddNode(&graph.Node{ID: fromID, Kind: graph.KindMethod, Name: lastSeg(fromID), FilePath: file, Language: "csharp"})
}
g.AddEdge(&graph.Edge{From: fromID, To: "unresolved::*.Handle", Kind: graph.EdgeCalls, FilePath: file,
Meta: map[string]any{"via": mediatrVia, "mediatr_request_type": reqType, "mediatr_kind": kind}})
}
func synthMediatREdge(g graph.Store, from, to string) *graph.Edge {
for e := range g.EdgesByKind(graph.EdgeCalls) {
if e == nil || e.From != from || e.To != to || e.Meta == nil {
continue
}
if by, _ := e.Meta[MetaSynthesizedBy].(string); by == SynthMediatR {
return e
}
}
return nil
}
func TestResolveMediatRCalls_SendBindsSingleHandler(t *testing.T) {
g := graph.New()
mediatrHandler(g, "App.cs::CreateOrderHandler.Handle", "App.cs", "CreateOrder", "request")
mediatrSend(g, "App.cs::Controller.Place", "App.cs", "CreateOrder", "request")
n := ResolveMediatRCalls(g)
require.Equal(t, 1, n)
e := synthMediatREdge(g, "App.cs::Controller.Place", "App.cs::CreateOrderHandler.Handle")
require.NotNil(t, e)
assert.Equal(t, ConfidenceTyped, e.Confidence)
assert.Equal(t, ProvenanceFramework, e.Meta[MetaProvenance])
}
func TestResolveMediatRCalls_PublishFansOut(t *testing.T) {
g := graph.New()
mediatrHandler(g, "App.cs::EmailHandler.Handle", "App.cs", "OrderPlaced", "notification")
mediatrHandler(g, "App.cs::SmsHandler.Handle", "App.cs", "OrderPlaced", "notification")
mediatrSend(g, "App.cs::Controller.Place", "App.cs", "OrderPlaced", "notification")
n := ResolveMediatRCalls(g)
require.Equal(t, 2, n, "a notification fans out to every handler")
assert.NotNil(t, synthMediatREdge(g, "App.cs::Controller.Place", "App.cs::EmailHandler.Handle"))
assert.NotNil(t, synthMediatREdge(g, "App.cs::Controller.Place", "App.cs::SmsHandler.Handle"))
}
func TestResolveMediatRCalls_RequestKindIsolated(t *testing.T) {
// A Send (request) must not bind to a same-typed notification handler.
g := graph.New()
mediatrHandler(g, "App.cs::NotifHandler.Handle", "App.cs", "Thing", "notification")
mediatrSend(g, "App.cs::C.M", "App.cs", "Thing", "request")
assert.Equal(t, 0, ResolveMediatRCalls(g))
assert.Nil(t, synthMediatREdge(g, "App.cs::C.M", "App.cs::NotifHandler.Handle"))
}
func TestResolveMediatRCalls_UnknownRequestStaysPlaceholder(t *testing.T) {
g := graph.New()
mediatrHandler(g, "App.cs::H.Handle", "App.cs", "Known", "request")
mediatrSend(g, "App.cs::C.M", "App.cs", "Ghost", "request")
assert.Equal(t, 0, ResolveMediatRCalls(g))
}