message_decode_test.go RawHistoryFolder

1package core
2
3import (
4	"encoding/json"
5	"testing"
6
7	"github.com/bnema/zut/packages/provider"
8)
9
10func TestDecodeMessageJSONRestoresToolContent(t *testing.T) {
11	raw, err := json.Marshal(provider.Message{Role: provider.RoleAssistant, Content: []provider.Content{
12		provider.TextBlock{Text: "checking"}, provider.ToolCallBlock{ID: "call-1", Name: "bash", Arguments: json.RawMessage(`{"command":"pwd"}`)},
13	}})
14	if err != nil {
15		t.Fatal(err)
16	}
17	message, err := DecodeMessageJSON(raw)
18	if err != nil {
19		t.Fatal(err)
20	}
21	if len(message.Content) != 2 {
22		t.Fatalf("message = %#v", message)
23	}
24	call, ok := message.Content[1].(provider.ToolCallBlock)
25	if !ok || call.ID != "call-1" || string(call.Arguments) != `{"command":"pwd"}` {
26		t.Fatalf("call = %#v", message.Content[1])
27	}
28}