mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-10-13 16:14:57 +00:00
Duy /zuey/
269e2618ae
Squash merge PR #113 after resolving the project changelog conflict with current dev. Local checks passed: Go store/http/gateway/agent/pipeline tests, SQLite-tagged tests, both Go builds, web Vitest, and web build. PR CI run 26705098712 passed release-versioning, go, and web.
108 lines
3.0 KiB
Go
108 lines
3.0 KiB
Go
package pg
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/nextlevelbuilder/goclaw/internal/store"
|
|
)
|
|
|
|
func TestPGRunTimelineStoreAppendAndListBySeq(t *testing.T) {
|
|
db := hooksTestDB(t)
|
|
timeline := NewPGRunTimelineStore(db)
|
|
tenantID, agentID := seedTenantAndAgent(t, db)
|
|
ctx := store.WithTenantID(context.Background(), tenantID)
|
|
|
|
items := []store.RunTimelineItem{
|
|
{
|
|
RunID: "pg-run-1",
|
|
SessionKey: "agent:default:direct:user-1",
|
|
AgentID: &agentID,
|
|
UserID: "user-1",
|
|
Channel: "web",
|
|
Seq: 2,
|
|
ItemType: store.RunTimelineItemTypeToolCall,
|
|
Status: store.RunTimelineStatusRunning,
|
|
Title: "exec_command",
|
|
Preview: `{"cmd":"date"}`,
|
|
ToolName: "exec_command",
|
|
ToolCallID: "call-1",
|
|
Metadata: json.RawMessage(`{"safe":true}`),
|
|
},
|
|
{
|
|
RunID: "pg-run-1",
|
|
SessionKey: "agent:default:direct:user-1",
|
|
UserID: "user-1",
|
|
Channel: "web",
|
|
Seq: 1,
|
|
ItemType: store.RunTimelineItemTypeRunStatus,
|
|
Status: store.RunTimelineStatusStarted,
|
|
Title: "Run started",
|
|
Preview: "hello",
|
|
},
|
|
}
|
|
for i := range items {
|
|
if err := timeline.AppendRunTimelineItem(ctx, &items[i]); err != nil {
|
|
t.Fatalf("AppendRunTimelineItem(%d): %v", i, err)
|
|
}
|
|
}
|
|
|
|
got, err := timeline.ListRunTimelineItems(ctx, store.RunTimelineListOpts{RunID: "pg-run-1", Limit: 10})
|
|
if err != nil {
|
|
t.Fatalf("ListRunTimelineItems: %v", err)
|
|
}
|
|
if len(got) != 2 {
|
|
t.Fatalf("len(items) = %d, want 2", len(got))
|
|
}
|
|
if got[0].Seq != 1 || got[1].Seq != 2 {
|
|
t.Fatalf("seq order = [%d,%d], want [1,2]", got[0].Seq, got[1].Seq)
|
|
}
|
|
if got[1].Content != "" {
|
|
t.Fatalf("Content persisted = %q, want empty preview-only archive", got[1].Content)
|
|
}
|
|
}
|
|
|
|
func TestPGRunTimelineStoreTenantScope(t *testing.T) {
|
|
db := hooksTestDB(t)
|
|
timeline := NewPGRunTimelineStore(db)
|
|
tenantID, _ := seedTenantAndAgent(t, db)
|
|
ctxA := store.WithTenantID(context.Background(), tenantID)
|
|
ctxB := store.WithTenantID(context.Background(), uuid.Must(uuid.NewV7()))
|
|
|
|
item := store.RunTimelineItem{
|
|
RunID: "pg-run-scope",
|
|
SessionKey: "agent:default:direct:user-1",
|
|
Seq: 1,
|
|
ItemType: store.RunTimelineItemTypeAssistantMessage,
|
|
Status: store.RunTimelineStatusCompleted,
|
|
Title: "Assistant",
|
|
Preview: "Visible only to tenant A",
|
|
Content: `{"raw":"must not persist"}`,
|
|
}
|
|
if err := timeline.AppendRunTimelineItem(ctxA, &item); err != nil {
|
|
t.Fatalf("AppendRunTimelineItem: %v", err)
|
|
}
|
|
|
|
gotA, err := timeline.ListRunTimelineItems(ctxA, store.RunTimelineListOpts{RunID: "pg-run-scope"})
|
|
if err != nil {
|
|
t.Fatalf("List tenant A: %v", err)
|
|
}
|
|
if len(gotA) != 1 {
|
|
t.Fatalf("tenant A len = %d, want 1", len(gotA))
|
|
}
|
|
if gotA[0].Content != "" {
|
|
t.Fatalf("Content = %q, want empty", gotA[0].Content)
|
|
}
|
|
|
|
gotB, err := timeline.ListRunTimelineItems(ctxB, store.RunTimelineListOpts{RunID: "pg-run-scope"})
|
|
if err != nil {
|
|
t.Fatalf("List tenant B: %v", err)
|
|
}
|
|
if len(gotB) != 0 {
|
|
t.Fatalf("tenant B len = %d, want 0", len(gotB))
|
|
}
|
|
}
|