Files
Cyrene_v2/plugins/tools/io/tools_test.go
T
AskaEth 20c1edb36a
CI / Backend (build / vet / test / cross-compile) (push) Has been cancelled
CI / Frontend (tsc + vite build) (push) Has been cancelled
feat(tools): document-read 按需读取工具(design/24 S4)
- 只读 S3-A 的 sidecar 正文(首行元数据行不计入 text),按字符分页,带 truncated/nextOffset 续读语义与 extractTruncated(区分「正文到此为止」与「原文件到此为止」)。

- 越界三态不静默:offset<0 / offset>totalChars → ErrDocumentOffset(碰盘前拦下,文案带总字符数);offset==totalChars 合法(读到末尾)。名非法/无 sidecar/unreadable → 哨兵错误 + 中文文案,绝不返回空正文假装成功。

- 读侧出口收敛到 documents 包:新增 ReadSidecar(含 Sidecar/Readable/ErrSidecarUnavailable),sidecar.go 的 load 变薄壳且对外行为逐字不变 ⇒ sidecar 形状与名字形态校验全仓只有一份实现。

- 注册与既有 6 个 io 工具同一路径(tools.go),策略默认 owner-only(群聊/陌生人默认调不到)。

- 主会话独立验收:隔离 worktree(HEAD + 本切片 6 个路径)build/vet 绿、全树 59 包 0 FAIL、双平台交叉编译绿、io 包 60 PASS;变异独立复现(工具名常量改成 document_read ⇒ 注册名单两条用例红)。

- 报告里发现的真实断裂(清单游标恒等于 sidecar 正文长度 ⇒ 照它续读必空)已登记,并入 T1 以「sidecar 存全文 + 注入预算与抽取预算解耦」根治。
2026-10-04 17:36:56 +08:00

108 lines
2.7 KiB
Go

package io
import (
"errors"
"testing"
"git.yeij.top/AskaEth/Cyrene_v2/internal/capability/registry"
"git.yeij.top/AskaEth/Cyrene_v2/internal/config"
"git.yeij.top/AskaEth/Cyrene_v2/pkg/plugins/sdk"
)
func TestTools_ReturnsEveryTool(t *testing.T) {
t.Parallel()
set, err := New(&config.Config{DataDir: t.TempDir()})
if err != nil {
t.Fatalf("New: %v", err)
}
if len(set) != 7 {
t.Fatalf("expected 7 tools, got %d", len(set))
}
want := []string{"file", "image-crop", "http", "web-search", "web-fetch", "knowledge", "document-read"}
seen := map[string]bool{}
for _, tool := range set {
seen[tool.Name()] = true
if tool.Description() == "" {
t.Errorf("tool %q has an empty description", tool.Name())
}
params, err := tool.Parameters()
if err != nil {
t.Errorf("tool %q Parameters error: %v", tool.Name(), err)
}
if params == nil {
t.Errorf("tool %q has nil parameters", tool.Name())
}
}
for _, name := range want {
if !seen[name] {
t.Errorf("missing tool %q", name)
}
}
}
func TestNew_NilConfigErrors(t *testing.T) {
t.Parallel()
if _, err := New(nil); err == nil {
t.Fatal("expected error for nil config")
}
}
func TestRegister_RegistersEveryTool(t *testing.T) {
t.Parallel()
reg := registry.New()
if err := Register(reg); err != nil {
t.Fatalf("Register: %v", err)
}
for _, name := range []string{"file", "image-crop", "http", "web-search", "web-fetch", "knowledge", "document-read"} {
c, err := reg.Get(name)
if err != nil {
t.Fatalf("Get(%q): %v", name, err)
}
if c.Kind != registry.KindTool {
t.Errorf("kind of %q = %q, want tool", name, c.Kind)
}
}
}
func TestRegister_NilRegistrar(t *testing.T) {
t.Parallel()
if err := Register(nil); !errors.Is(err, sdk.ErrNilRegistrar) {
t.Fatalf("expected sdk.ErrNilRegistrar, got %v", err)
}
}
func TestRegister_DuplicateErrors(t *testing.T) {
t.Parallel()
reg := registry.New()
if err := Register(reg); err != nil {
t.Fatalf("first Register: %v", err)
}
if err := Register(reg); !errors.Is(err, registry.ErrDuplicate) {
t.Fatalf("expected registry.ErrDuplicate, got %v", err)
}
}
func TestTools_ConcurrentDescribe(t *testing.T) {
t.Parallel()
set, err := New(&config.Config{DataDir: t.TempDir()})
if err != nil {
t.Fatalf("New: %v", err)
}
// Exercise the read-only contract concurrently to catch any shared-state
// mutation (the dispatcher may call Parameters from many goroutines).
for _, tool := range set {
tool := tool
t.Run(tool.Name(), func(t *testing.T) {
t.Parallel()
for i := 0; i < 10; i++ {
if _, err := tool.Parameters(); err != nil {
t.Fatalf("Parameters: %v", err)
}
if tool.Name() == "" || tool.Description() == "" {
t.Fatalf("empty identity for %v", tool)
}
}
})
}
}