X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/959c24478e2c3a596a81cca359fd784c897f46a2..e86df5fad9aa2f280aad845adeefd239b0e944ae:/tools_test.go?ds=inline diff --git a/tools_test.go b/tools_test.go index fbbea41..c03e7fc 100644 --- a/tools_test.go +++ b/tools_test.go @@ -185,7 +185,7 @@ func TestZoomWithANameGivesTheWholeChatOfTheAgent(t *testing.T) { output := box.run(t.Context(), "zoom", `{"id":0,"n":1,"name":"`+subagent.name+`"}`) require.Equal(t, "0|note: subagent\n\n1|user: Count the files.", output) - require.Equal(t, "error: no agent BlueFox", box.run(t.Context(), "zoom", `{"id":0,"n":1,"name":"BlueFox"}`)) + require.Equal(t, "error: open worker log: no agent BlueFox", box.run(t.Context(), "zoom", `{"id":0,"n":1,"name":"BlueFox"}`)) require.Equal(t, "error: line 0+1 goes past the last message, -1", box.run(t.Context(), "zoom", `{"id":0,"n":1,"name":null}`)) } @@ -244,6 +244,13 @@ func newTestToolbox(t *testing.T, log ToolLog, dir string) *Toolbox { } } +func (r *ToolRecordingLog) Events() []string { + r.mutex.Lock() + defer r.mutex.Unlock() + + return append([]string(nil), r.events...) +} + func (r *ToolRecordingLog) Append(kind, text string) (Message, error) { r.mutex.Lock() defer r.mutex.Unlock() @@ -267,10 +274,3 @@ func (r *ToolRecordingLog) Flush(ctx context.Context) error { return ctx.Err() } - -func (r *ToolRecordingLog) Events() []string { - r.mutex.Lock() - defer r.mutex.Unlock() - - return append([]string(nil), r.events...) -}