WIP for Session Tests

go.mod

@@ -18,6 +18,7 @@ require (
 	github.com/mark3labs/mcp-go v0.31.0
 	github.com/spf13/cobra v1.9.1
 	github.com/spf13/pflag v1.0.6
+	github.com/stretchr/testify v1.10.0
 	golang.org/x/sync v0.13.0
 	golang.org/x/term v0.31.0
 	k8s.io/klog/v2 v2.130.1
@@ -60,6 +61,7 @@ require (
 	github.com/charmbracelet/x/cellbuf v0.0.13 // indirect
 	github.com/charmbracelet/x/exp/slice v0.0.0-20250327172914-2fdc97757edf // indirect
 	github.com/charmbracelet/x/term v0.2.1 // indirect
+	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/dlclark/regexp2 v1.11.4 // indirect
 	github.com/erikgeiser/coninput v0.0.0-20211004153227-1c3628e74d0f // indirect
 	github.com/felixge/httpsnoop v1.0.4 // indirect
@@ -86,6 +88,7 @@ require (
 	github.com/ollama/ollama v0.6.5 // indirect
 	github.com/openai/openai-go v1.11.0 // indirect
 	github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/rivo/uniseg v0.4.7 // indirect
 	github.com/sahilm/fuzzy v0.1.1 // indirect
 	github.com/spf13/cast v1.7.1 // indirect
@@ -110,4 +113,5 @@ require (
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20250219182151-9fdb1cabc7b2 // indirect
 	google.golang.org/grpc v1.70.0 // indirect
 	google.golang.org/protobuf v1.36.5 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

pkg/sessions/session.go

@@ -15,6 +15,7 @@
 package sessions
 
 import (
+	"bufio"
 	"encoding/json"
 	"fmt"
 	"os"
@@ -146,11 +147,18 @@ func (s *Session) ChatMessages() []*api.Message {
 	}
 	defer f.Close()
 
-	scanner := json.NewDecoder(f)
-	for scanner.More() {
+	// Read file line by line instead of using json.Decoder
+	scanner := bufio.NewScanner(f)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if line == "" {
+			continue
+		}
+
 		var message api.Message
-		if err := scanner.Decode(&message); err != nil {
-			continue // skip malformed messages
+		if err := json.Unmarshal([]byte(line), &message); err != nil {
+			// Skip malformed messages
+			continue
 		}
 		messages = append(messages, &message)
 	}