mirror of
https://github.com/YspCoder/clawgo.git
synced 2026-04-29 11:07:29 +08:00
refactor: stabilize runtime and unify config
This commit is contained in:
64
pkg/session/manager_test.go
Normal file
64
pkg/session/manager_test.go
Normal file
@@ -0,0 +1,64 @@
|
||||
package session
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/YspCoder/clawgo/pkg/providers"
|
||||
)
|
||||
|
||||
func TestLoadSessionsReturnsScannerErrorForOversizedLine(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
storage := t.TempDir()
|
||||
line := `{"role":"user","content":"` + strings.Repeat("x", 2*1024*1024) + `"}`
|
||||
if err := os.WriteFile(filepath.Join(storage, "huge.jsonl"), []byte(line+"\n"), 0644); err != nil {
|
||||
t.Fatalf("write session file failed: %v", err)
|
||||
}
|
||||
|
||||
sm := &SessionManager{
|
||||
sessions: map[string]*Session{},
|
||||
storage: storage,
|
||||
}
|
||||
if err := sm.loadSessions(); err != nil {
|
||||
t.Fatalf("expected oversized line to load with expanded scanner buffer, got %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestFromJSONLLineParsesOpenClawToolResult(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
line := []byte(`{"type":"message","message":{"role":"toolResult","content":[{"type":"text","text":"done"}],"toolCallId":"call-1"}}`)
|
||||
msg, ok := fromJSONLLine(line)
|
||||
if !ok {
|
||||
t.Fatal("expected line to parse")
|
||||
}
|
||||
if msg.Role != "tool" || msg.ToolCallID != "call-1" || msg.Content != "done" {
|
||||
t.Fatalf("unexpected parsed message: %+v", msg)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRewriteSessionFileLockedPersistsMessages(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
storage := t.TempDir()
|
||||
sm := &SessionManager{storage: storage}
|
||||
session := &Session{
|
||||
Key: "abc",
|
||||
Messages: []providers.Message{
|
||||
{Role: "user", Content: "hello"},
|
||||
},
|
||||
}
|
||||
if err := sm.rewriteSessionFileLocked(session); err != nil {
|
||||
t.Fatalf("rewrite session failed: %v", err)
|
||||
}
|
||||
data, err := os.ReadFile(filepath.Join(storage, "abc.jsonl"))
|
||||
if err != nil {
|
||||
t.Fatalf("read rewritten session failed: %v", err)
|
||||
}
|
||||
if !strings.Contains(string(data), `"role":"user"`) {
|
||||
t.Fatalf("unexpected rewritten session contents: %s", string(data))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user