rpc_reasoning_test.go RawHistoryFolder

1package agent
2
3import (
4	"bytes"
5	"strings"
6	"testing"
7
8	"github.com/patriceckhart/zot/packages/core"
9)
10
11func TestRPCSetReasoningMax(t *testing.T) {
12	var out bytes.Buffer
13	s := &rpcServer{agent: &core.Agent{}, out: &out}
14	s.dispatch("set_reasoning", "1", []byte(`{"reasoning":"max"}`))
15
16	if s.agent.Reasoning != "max" {
17		t.Fatalf("reasoning = %q, want max", s.agent.Reasoning)
18	}
19	if !strings.Contains(out.String(), `"reasoning":"max"`) {
20		t.Fatalf("response = %q", out.String())
21	}
22}
23
24func TestRPCSetReasoningRejectsUnknownLevel(t *testing.T) {
25	var out bytes.Buffer
26	s := &rpcServer{agent: &core.Agent{}, out: &out}
27	s.dispatch("set_reasoning", "1", []byte(`{"reasoning":"extreme"}`))
28
29	if s.agent.Reasoning != "" {
30		t.Fatalf("reasoning = %q, want unchanged", s.agent.Reasoning)
31	}
32	if !strings.Contains(out.String(), `"success":false`) {
33		t.Fatalf("response = %q", out.String())
34	}
35}