rpc_alert_test.go RawHistoryFolder

1package agent
2
3import (
4	"bytes"
5	"encoding/json"
6	"strings"
7	"testing"
8
9	"github.com/bnema/zut/packages/agent/extproto"
10)
11
12func TestRPCExtAlertIsJSONOnly(t *testing.T) {
13	var out bytes.Buffer
14	hooks := &rpcExtHooks{server: &rpcServer{out: &out, started: true, authed: true}}
15	hooks.Alert("question-ext", extproto.AlertRequest{Kind: extproto.AlertKindBell, Reason: "question_ready"})
16
17	if strings.ContainsRune(out.String(), '\a') {
18		t.Fatalf("RPC alert contains raw BEL: %q", out.String())
19	}
20	var frame map[string]any
21	if err := json.Unmarshal(out.Bytes(), &frame); err != nil {
22		t.Fatalf("decode RPC alert: %v", err)
23	}
24	if frame["type"] != "ext_alert" || frame["extension"] != "question-ext" || frame["kind"] != extproto.AlertKindBell || frame["reason"] != "question_ready" {
25		t.Fatalf("RPC alert frame = %#v", frame)
26	}
27}
28
29func TestRPCExtAlertWaitsForAuthentication(t *testing.T) {
30	t.Setenv("ZUTCORE_RPC_TOKEN", "synthetic-token")
31
32	var out bytes.Buffer
33	server := &rpcServer{out: &out, provider: "test", model: "test-model", version: "test-version"}
34	hooks := &rpcExtHooks{server: server}
35	hooks.Alert("question-ext", extproto.AlertRequest{Kind: extproto.AlertKindBell, Reason: "question_ready"})
36	if out.Len() != 0 {
37		t.Fatalf("pre-auth RPC output = %q, want empty", out.String())
38	}
39
40	if err := server.run(strings.NewReader(`{"type":"hello","id":"1","token":"synthetic-token"}` + "\n")); err != nil {
41		t.Fatalf("run RPC server: %v", err)
42	}
43	lines := strings.Split(strings.TrimSpace(out.String()), "\n")
44	if len(lines) != 2 || !strings.Contains(lines[0], `"type":"response"`) || !strings.Contains(lines[1], `"type":"ext_alert"`) {
45		t.Fatalf("authenticated RPC frames = %q, want response followed by ext_alert", out.String())
46	}
47}