clawgo/pkg/session/manager_test.go

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))
	}
}