nullkey
cf84bf2a38
5 new leaf commands wired into the root tree:
whoami — simplified `auth status` (user_id + tenant_id only)
doctor — 4-item self-check (base_url / auth / server_version / cred_storage)
with --offline / --no-cache flags + skip cascade + summary.all_passed
防 agent 看到 envelope.ok=true 误判命令整体 success
kb list — list KBs (default updated_at desc; 0 KB → "(no knowledge bases)")
tabwriter 4-col (ID/NAME/DOCS/UPDATED), display-width truncation
kb get — show single KB details (KEY: VALUE, suppress empty fields)
context use — switch default context (writes config.current_context),
带 levenshtein distance ≤ 2 的 did-you-mean hint
Each command uses the v0.0 narrow Service interface pattern (testable via
fakes), agent.SetAgentHelp for AI-friendly hints, and ClassifyHTTPError
for stable error code mapping.
cmdutil/errors.go: 新增 CodeLocalContextNotFound for `context use`,加入 AllCodes() 注册集.
cli/cmd/root.go: NewRootCmd 改为 exported (acceptance/contract 测试需要),
注册 4 个新命令 + 1 parent group; root_test.go 跟随更新.
104 lines
2.7 KiB
Go
104 lines
2.7 KiB
Go
package contextcmd
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Tencent/WeKnora/cli/internal/cmdutil"
|
|
"github.com/Tencent/WeKnora/cli/internal/config"
|
|
"github.com/Tencent/WeKnora/cli/internal/iostreams"
|
|
)
|
|
|
|
func TestUse_OK(t *testing.T) {
|
|
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
|
|
out, _ := iostreams.SetForTest(t)
|
|
|
|
cfg := &config.Config{
|
|
CurrentContext: "staging",
|
|
Contexts: map[string]config.Context{
|
|
"staging": {Host: "https://staging.example.com"},
|
|
"production": {Host: "https://prod.example.com"},
|
|
},
|
|
}
|
|
if err := config.Save(cfg); err != nil {
|
|
t.Fatalf("Save initial config: %v", err)
|
|
}
|
|
|
|
if err := runUse("production"); err != nil {
|
|
t.Fatalf("runUse: %v", err)
|
|
}
|
|
|
|
got, err := config.Load()
|
|
if err != nil {
|
|
t.Fatalf("reload: %v", err)
|
|
}
|
|
if got.CurrentContext != "production" {
|
|
t.Errorf("CurrentContext = %q, want production", got.CurrentContext)
|
|
}
|
|
if !strings.Contains(out.String(), "production") {
|
|
t.Errorf("output should mention switched-to context, got %q", out.String())
|
|
}
|
|
}
|
|
|
|
func TestUse_NotFound_WithDidYouMean(t *testing.T) {
|
|
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
|
|
_, _ = iostreams.SetForTest(t)
|
|
|
|
cfg := &config.Config{Contexts: map[string]config.Context{
|
|
"production": {Host: "https://prod"},
|
|
"staging": {Host: "https://staging"},
|
|
}}
|
|
if err := config.Save(cfg); err != nil {
|
|
t.Fatalf("Save: %v", err)
|
|
}
|
|
|
|
err := runUse("prodution") // typo: missing 'c'
|
|
if err == nil {
|
|
t.Fatal("expected error")
|
|
}
|
|
cm, ok := err.(*cmdutil.Error)
|
|
if !ok {
|
|
t.Fatalf("expected *cmdutil.Error, got %T", err)
|
|
}
|
|
if cm.Code != cmdutil.CodeLocalContextNotFound {
|
|
t.Errorf("code = %q, want %q", cm.Code, cmdutil.CodeLocalContextNotFound)
|
|
}
|
|
if !strings.Contains(cm.Hint, "production") {
|
|
t.Errorf("hint should suggest 'production', got %q", cm.Hint)
|
|
}
|
|
}
|
|
|
|
func TestUse_NotFound_EmptyContexts(t *testing.T) {
|
|
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
|
|
_, _ = iostreams.SetForTest(t)
|
|
|
|
err := runUse("anything")
|
|
if err == nil {
|
|
t.Fatal("expected error")
|
|
}
|
|
cm := err.(*cmdutil.Error)
|
|
if !strings.Contains(cm.Hint, "auth login") {
|
|
t.Errorf("hint should mention `auth login` for empty Contexts, got %q", cm.Hint)
|
|
}
|
|
}
|
|
|
|
func TestUse_CaseSensitive(t *testing.T) {
|
|
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
|
|
_, _ = iostreams.SetForTest(t)
|
|
|
|
cfg := &config.Config{Contexts: map[string]config.Context{
|
|
"Production": {Host: "https://prod"},
|
|
}}
|
|
_ = config.Save(cfg)
|
|
|
|
err := runUse("production") // lowercase — must NOT match "Production"
|
|
if err == nil {
|
|
t.Fatal("expected case-sensitive miss")
|
|
}
|
|
cm := err.(*cmdutil.Error)
|
|
// did-you-mean kicks in (distance 1 — "P"→"p")
|
|
if !strings.Contains(cm.Hint, "Production") {
|
|
t.Errorf("hint should suggest 'Production' (case-different), got %q", cm.Hint)
|
|
}
|
|
}
|