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
69 lines
2.2 KiB
Go
69 lines
2.2 KiB
Go
package mcp
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/mark3labs/mcp-go/mcp"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/zzet/gortex/internal/config"
|
|
"github.com/zzet/gortex/internal/graph"
|
|
)
|
|
|
|
func newGuardsTestServer(t *testing.T) *Server {
|
|
t.Helper()
|
|
g := graph.New()
|
|
g.AddNode(&graph.Node{ID: "dom", Kind: graph.KindFunction, Name: "Dom", FilePath: "internal/domain/user.go"})
|
|
g.AddNode(&graph.Node{ID: "inf", Kind: graph.KindFunction, Name: "Inf", FilePath: "internal/infra/db.go"})
|
|
g.AddEdge(&graph.Edge{From: "dom", To: "inf", Kind: graph.EdgeCalls})
|
|
return &Server{
|
|
graph: g,
|
|
session: newSessionState(),
|
|
tokenStats: &tokenStats{},
|
|
symHistory: &symbolHistory{entries: make(map[string][]SymbolModification)},
|
|
sessions: newSessionMap(),
|
|
toolScopes: newScopeRegistry(),
|
|
}
|
|
}
|
|
|
|
func callCheckGuards(t *testing.T, s *Server, ids string) map[string]any {
|
|
t.Helper()
|
|
req := mcp.CallToolRequest{}
|
|
req.Params.Arguments = map[string]any{"ids": ids}
|
|
res, err := s.handleCheckGuards(context.Background(), req)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, res)
|
|
require.False(t, res.IsError)
|
|
tc, ok := res.Content[0].(mcp.TextContent)
|
|
require.True(t, ok)
|
|
var m map[string]any
|
|
require.NoError(t, json.Unmarshal([]byte(tc.Text), &m))
|
|
return m
|
|
}
|
|
|
|
func TestCheckGuards_ArchitectureLayerViolation(t *testing.T) {
|
|
s := newGuardsTestServer(t)
|
|
s.SetArchitecture(config.ArchitectureConfig{
|
|
Layers: map[string]config.LayerRule{
|
|
"domain": {Paths: []string{"internal/domain/**"}, Deny: []string{"*"}},
|
|
"infra": {Paths: []string{"internal/infra/**"}, Allow: []string{"domain"}},
|
|
},
|
|
})
|
|
out := callCheckGuards(t, s, "dom")
|
|
violations, _ := out["violations"].([]any)
|
|
require.Len(t, violations, 1)
|
|
v, _ := violations[0].(map[string]any)
|
|
require.Equal(t, "layer", v["kind"])
|
|
require.Equal(t, "domain", v["layer_from"])
|
|
require.Equal(t, "infra", v["layer_to"])
|
|
}
|
|
|
|
func TestCheckGuards_NoRulesNoArchitecture(t *testing.T) {
|
|
s := newGuardsTestServer(t)
|
|
out := callCheckGuards(t, s, "dom")
|
|
// No flat rules and no architecture block — the explicit
|
|
// "nothing configured" message, not a violation list.
|
|
require.Equal(t, "no guard rules configured", out["message"])
|
|
}
|