玉澜
7afd4139fc
- NewSchemaCommand / NewMCPCommand / newCatalogCommand no longer accept a DiscoveryCatalogLoader they always discarded; schema's no-discovery property is now structural, retiring the panic-loader test guard along with the trivial root.go wrappers and the unused buildMCPCommandFn seam. - Delete the lazy sync.Once / atomic counter around the retired MCP pin: runtimeMCPMetadata only existed so a diagnostic counter could observe a loader that always returns the constant empty pin. Assembly now calls emptyPinnedMCPMetadata directly and SchemaMetadataLoadCounts loses the dead MCPMetadata field (the policy bans keep the retired names from reappearing).
79 lines
2.6 KiB
Go
79 lines
2.6 KiB
Go
// Copyright 2026 Alibaba Group
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
|
|
package cli
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func TestSchemaUsesDeliveryCatalogWithoutRuntimeLoad(t *testing.T) {
|
|
root := &cobra.Command{Use: "dws"}
|
|
root.AddCommand(NewSchemaCommand())
|
|
var stdout bytes.Buffer
|
|
root.SetOut(&stdout)
|
|
root.SetArgs([]string{"schema"})
|
|
if err := root.Execute(); err != nil {
|
|
t.Fatalf("schema execute: %v", err)
|
|
}
|
|
var payload struct {
|
|
Count int `json:"count"`
|
|
ToolCount int `json:"tool_count"`
|
|
}
|
|
if err := json.Unmarshal(stdout.Bytes(), &payload); err != nil {
|
|
t.Fatalf("decode schema: %v\n%s", err, stdout.String())
|
|
}
|
|
loaded := deliverySchemaCatalog()
|
|
if payload.Count != len(loaded.Registry.Products) || payload.ToolCount != len(loaded.Index.CanonicalPaths()) {
|
|
t.Fatalf("schema counts = %d/%d, want %d/%d", payload.Count, payload.ToolCount, len(loaded.Registry.Products), len(loaded.Index.CanonicalPaths()))
|
|
}
|
|
}
|
|
|
|
func TestSchemaAllReturnsCompleteDeliveryLeafSchemas(t *testing.T) {
|
|
root := &cobra.Command{Use: "dws"}
|
|
root.AddCommand(NewSchemaCommand())
|
|
var stdout bytes.Buffer
|
|
root.SetOut(&stdout)
|
|
root.SetArgs([]string{"schema", "--all"})
|
|
if err := root.Execute(); err != nil {
|
|
t.Fatalf("schema --all execute: %v", err)
|
|
}
|
|
var payload map[string]any
|
|
if err := json.Unmarshal(stdout.Bytes(), &payload); err != nil {
|
|
t.Fatalf("decode schema --all: %v", err)
|
|
}
|
|
expanded := 0
|
|
for _, product := range schemaMapSlice(payload["products"]) {
|
|
for _, tool := range schemaMapSlice(product["tools"]) {
|
|
if _, ok := tool["parameters"].(map[string]any); !ok {
|
|
t.Fatalf("schema --all tool %q has no parameters object", schemaString(tool["canonical_path"]))
|
|
}
|
|
expanded++
|
|
}
|
|
}
|
|
want := len(deliverySchemaCatalog().Index.CanonicalPaths())
|
|
if expanded != want {
|
|
t.Fatalf("schema --all tools = %d, want %d", expanded, want)
|
|
}
|
|
}
|
|
|
|
func TestWalkLeafCommandsTraversesAnnotatedHiddenSubtree(t *testing.T) {
|
|
root := &cobra.Command{Use: "dws"}
|
|
group := &cobra.Command{Use: "compat", Hidden: true, Run: func(*cobra.Command, []string) {}}
|
|
leaf := &cobra.Command{Use: "legacy", Hidden: true, Run: func(*cobra.Command, []string) {}}
|
|
AttachRuntimeSchema(leaf, "compat", "legacy", "test")
|
|
group.AddCommand(leaf)
|
|
root.AddCommand(group)
|
|
|
|
var got []*cobra.Command
|
|
walkLeafCommands(root, func(command *cobra.Command) { got = append(got, command) })
|
|
if len(got) != 1 || got[0] != leaf {
|
|
t.Fatalf("walked commands = %#v, want annotated hidden leaf", got)
|
|
}
|
|
}
|