tool_prompt_test.go RawHistoryFolder

1package ext
2
3import (
4	"encoding/json"
5	"testing"
6	"time"
7
8	"github.com/patriceckhart/zot/packages/agent/extproto"
9)
10
11func TestToolPromptResponse(t *testing.T) {
12	h := newHarness("shortcut")
13	h.ext.Command("commit", "commit", func(args string) Response {
14		return ToolPrompt("skill", map[string]string{"name": "developer:commit"}, args)
15	})
16	done := make(chan error, 1)
17	go func() { done <- h.ext.Run() }()
18	t.Cleanup(func() {
19		_ = h.hostW.Close()
20		select {
21		case err := <-done:
22			if err != nil {
23				t.Error(err)
24			}
25		case <-time.After(time.Second):
26			t.Error("SDK did not exit")
27		}
28	})
29	h.drainUntil(t, "hello")
30	h.sendToExt(t, extproto.HelloAckFromHost{Type: "hello_ack"})
31	h.drainUntil(t, "ready")
32	h.sendToExt(t, extproto.CommandInvokedFromHost{Type: "command_invoked", ID: "1", Name: "commit", Args: "and push"})
33	frame := h.drainUntil(t, "command_response")
34	var decoded extproto.CommandResponseFromExt
35	if err := json.Unmarshal(frame.raw, &decoded); err != nil {
36		t.Fatal(err)
37	}
38	if decoded.ID != "1" || decoded.Action != "tool_prompt" || decoded.Prompt != "and push" || decoded.ToolName != "skill" || string(decoded.ToolArgs) != `{"name":"developer:commit"}` {
39		t.Fatalf("response = %+v", decoded)
40	}
41	if got := ToolPrompt("skill", []string{"not", "object"}, "request"); got.Error == "" {
42		t.Fatalf("invalid args accepted: %+v", got)
43	}
44}