mirror of
https://github.com/Tencent/WeKnora.git
synced 2026-08-30 16:53:21 +08:00
nullkey
6f8b825f5f
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.
85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package doc
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/Tencent/WeKnora/cli/internal/cmdutil"
|
|
"github.com/Tencent/WeKnora/cli/internal/iostreams"
|
|
sdk "github.com/Tencent/WeKnora/client"
|
|
)
|
|
|
|
type fakeReparseSvc struct {
|
|
k *sdk.Knowledge
|
|
err error
|
|
called bool
|
|
}
|
|
|
|
func (f *fakeReparseSvc) ReparseKnowledge(_ context.Context, id string) (*sdk.Knowledge, error) {
|
|
f.called = true
|
|
if f.err != nil {
|
|
return nil, f.err
|
|
}
|
|
k := *f.k
|
|
k.ID = id
|
|
return &k, nil
|
|
}
|
|
|
|
func TestDocReparse_Text(t *testing.T) {
|
|
out, _ := iostreams.SetForTest(t)
|
|
svc := &fakeReparseSvc{k: &sdk.Knowledge{FileName: "runbook.md", ParseStatus: "processing"}}
|
|
require.NoError(t, runReparse(context.Background(), &ReparseOptions{}, &cmdutil.FormatOptions{Mode: cmdutil.FormatText}, svc, "doc_abc"))
|
|
assert.True(t, svc.called)
|
|
got := out.String()
|
|
assert.Contains(t, got, "doc_abc")
|
|
assert.Contains(t, got, "runbook.md")
|
|
assert.Contains(t, got, "processing")
|
|
assert.Contains(t, got, "doc wait doc_abc")
|
|
}
|
|
|
|
func TestDocReparse_JSON(t *testing.T) {
|
|
out, _ := iostreams.SetForTest(t)
|
|
svc := &fakeReparseSvc{k: &sdk.Knowledge{ParseStatus: "processing"}}
|
|
require.NoError(t, runReparse(context.Background(), &ReparseOptions{}, &cmdutil.FormatOptions{Mode: cmdutil.FormatJSON}, svc, "doc_abc"))
|
|
var env struct {
|
|
OK bool `json:"ok"`
|
|
Data sdk.Knowledge `json:"data"`
|
|
}
|
|
require.NoError(t, json.Unmarshal(out.Bytes(), &env), "got %q", out.String())
|
|
assert.True(t, env.OK)
|
|
assert.Equal(t, "doc_abc", env.Data.ID)
|
|
assert.Equal(t, "processing", env.Data.ParseStatus)
|
|
}
|
|
|
|
func TestDocReparse_NotFound(t *testing.T) {
|
|
_, _ = iostreams.SetForTest(t)
|
|
svc := &fakeReparseSvc{err: errors.New("HTTP error 404: not found")}
|
|
err := runReparse(context.Background(), &ReparseOptions{}, &cmdutil.FormatOptions{Mode: cmdutil.FormatText}, svc, "missing")
|
|
require.Error(t, err)
|
|
assert.True(t, cmdutil.IsNotFound(err))
|
|
}
|
|
|
|
// TestDocReparse_DryRun_NoServerCall: --dry-run must emit a doc.reparse plan
|
|
// (exit 0) without reaching the server, honoring the mutation-preview contract.
|
|
func TestDocReparse_DryRun_NoServerCall(t *testing.T) {
|
|
out, _ := iostreams.SetForTest(t)
|
|
root := withRootHarnessDoc(NewCmdReparse(docDryRunFactory(t)), "doc_x", "--dry-run", "--format", "json")
|
|
require.NoError(t, root.Execute(), "dry-run must succeed without a client")
|
|
var env struct {
|
|
OK bool `json:"ok"`
|
|
Meta struct {
|
|
DryRun bool `json:"dry_run"`
|
|
Plan map[string]any `json:"plan"`
|
|
} `json:"meta"`
|
|
}
|
|
require.NoError(t, json.Unmarshal(out.Bytes(), &env), "got %q", out.String())
|
|
assert.True(t, env.OK)
|
|
assert.True(t, env.Meta.DryRun)
|
|
assert.Equal(t, "doc.reparse", env.Meta.Plan["action"])
|
|
}
|