mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-06-10 04:10:26 +00:00
viettranx
ee9f5b8aaf
Team members dispatched via team tasks can now read the leader's memory files (MEMORY.md, memory/*.md) as a fallback when they have no own memory. Memory writes are blocked for members — only the leader can save memory files. This follows the existing team_workspace context propagation pattern: dispatch metadata → RunRequest → context → tools. Affected tools: read_file (interceptor), memory_get, memory_search, list_files (interceptor). All include leader fallback with per-user → global scope cascade.
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestToolContextKeys_Channel(t *testing.T) {
|
|
ctx := context.Background()
|
|
if v := ToolChannelFromCtx(ctx); v != "" {
|
|
t.Errorf("expected empty, got %q", v)
|
|
}
|
|
|
|
ctx = WithToolChannel(ctx, "telegram")
|
|
if v := ToolChannelFromCtx(ctx); v != "telegram" {
|
|
t.Errorf("expected telegram, got %q", v)
|
|
}
|
|
}
|
|
|
|
func TestToolContextKeys_ChatID(t *testing.T) {
|
|
ctx := context.Background()
|
|
if v := ToolChatIDFromCtx(ctx); v != "" {
|
|
t.Errorf("expected empty, got %q", v)
|
|
}
|
|
|
|
ctx = WithToolChatID(ctx, "chat-123")
|
|
if v := ToolChatIDFromCtx(ctx); v != "chat-123" {
|
|
t.Errorf("expected chat-123, got %q", v)
|
|
}
|
|
}
|
|
|
|
func TestToolContextKeys_PeerKind(t *testing.T) {
|
|
ctx := context.Background()
|
|
ctx = WithToolPeerKind(ctx, "group")
|
|
if v := ToolPeerKindFromCtx(ctx); v != "group" {
|
|
t.Errorf("expected group, got %q", v)
|
|
}
|
|
}
|
|
|
|
func TestToolContextKeys_SandboxKey(t *testing.T) {
|
|
ctx := context.Background()
|
|
ctx = WithToolSandboxKey(ctx, "agent:main:telegram:direct:123")
|
|
if v := ToolSandboxKeyFromCtx(ctx); v != "agent:main:telegram:direct:123" {
|
|
t.Errorf("expected sandbox key, got %q", v)
|
|
}
|
|
}
|
|
|
|
func TestToolContextKeys_AsyncCB(t *testing.T) {
|
|
ctx := context.Background()
|
|
if v := ToolAsyncCBFromCtx(ctx); v != nil {
|
|
t.Error("expected nil callback")
|
|
}
|
|
|
|
called := false
|
|
cb := AsyncCallback(func(ctx context.Context, result *Result) {
|
|
called = true
|
|
})
|
|
|
|
ctx = WithToolAsyncCB(ctx, cb)
|
|
got := ToolAsyncCBFromCtx(ctx)
|
|
if got == nil {
|
|
t.Fatal("expected non-nil callback")
|
|
}
|
|
got(ctx, nil)
|
|
if !called {
|
|
t.Error("callback was not invoked")
|
|
}
|
|
}
|
|
|
|
func TestToolContextKeys_LeaderAgentID(t *testing.T) {
|
|
ctx := context.Background()
|
|
if v := LeaderAgentIDFromCtx(ctx); v != "" {
|
|
t.Errorf("expected empty, got %q", v)
|
|
}
|
|
|
|
ctx = WithLeaderAgentID(ctx, "leader-uuid-123")
|
|
if v := LeaderAgentIDFromCtx(ctx); v != "leader-uuid-123" {
|
|
t.Errorf("expected leader-uuid-123, got %q", v)
|
|
}
|
|
}
|
|
|
|
func TestToolContextKeys_MultipleValues(t *testing.T) {
|
|
ctx := context.Background()
|
|
ctx = WithToolChannel(ctx, "slack")
|
|
ctx = WithToolChatID(ctx, "C123")
|
|
ctx = WithToolPeerKind(ctx, "direct")
|
|
ctx = WithToolSandboxKey(ctx, "sandbox-1")
|
|
|
|
if v := ToolChannelFromCtx(ctx); v != "slack" {
|
|
t.Errorf("channel: expected slack, got %q", v)
|
|
}
|
|
if v := ToolChatIDFromCtx(ctx); v != "C123" {
|
|
t.Errorf("chatID: expected C123, got %q", v)
|
|
}
|
|
if v := ToolPeerKindFromCtx(ctx); v != "direct" {
|
|
t.Errorf("peerKind: expected direct, got %q", v)
|
|
}
|
|
if v := ToolSandboxKeyFromCtx(ctx); v != "sandbox-1" {
|
|
t.Errorf("sandboxKey: expected sandbox-1, got %q", v)
|
|
}
|
|
}
|