Files
Enterprise-Knowledge-Base/cli/internal/mcp/server_test.go
T
nullkey 6f8b825f5f feat(cli): agent-first WeKnora CLI
The `weknora` command-line client for the WeKnora RAG server. Full command
surface: kb / doc / chunk / session / message / agent / model / search / chat /
auth / profile / link / api / mcp / skills / config / doctor / schema /
exit-codes.

Agent-first wire contract: JSON envelope by default, typed error codes mapped to
stable exit codes, error.retry_argv / retryable / partial-failure status,
global --dry-run preview, the exit-10 confirmation protocol for writes, and
control/ANSI/Bidi input hygiene. Stateless env-credential auth
(WEKNORA_API_KEY / WEKNORA_TOKEN + WEKNORA_HOST) for headless/CI/agent use.
A curated read-only MCP tool surface (`mcp serve`) and embedded Agent Skills
(`skills install`). Extensive unit + contract-golden tests and CI.
2026-07-06 22:53:01 +08:00

38 lines
929 B
Go

package mcp
import (
"context"
"errors"
"fmt"
"io"
"testing"
)
// TestIsCleanShutdown pins that an MCP host disconnecting (stdin EOF) or a
// cancelled context is treated as normal termination (exit 0), while a genuine
// transport fault is not.
func TestIsCleanShutdown(t *testing.T) {
clean := []error{
io.EOF,
context.Canceled,
context.DeadlineExceeded,
fmt.Errorf("jsonrpc: %w", io.EOF), // wrapped EOF
errors.New("server is closing: EOF"), // go-sdk formatted string (no unwrap)
}
for _, e := range clean {
if !isCleanShutdown(e) {
t.Errorf("isCleanShutdown(%v) = false, want true", e)
}
}
notClean := []error{
errors.New("connection reset by peer"),
errors.New("write: broken pipe"),
errors.New("EOF while reading header but more expected"), // EOF not at end
}
for _, e := range notClean {
if isCleanShutdown(e) {
t.Errorf("isCleanShutdown(%v) = true, want false", e)
}
}
}