X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/d302957c09b192d0fbc09287ba7dcb7c32426556..016d46008fc9726d2617e4e3ea098113f34c7636:/compactor_test.go?ds=inline diff --git a/compactor_test.go b/compactor_test.go index 8bdf45c..8c83594 100644 --- a/compactor_test.go +++ b/compactor_test.go @@ -167,6 +167,7 @@ func TestCompactorRetriesALongLineWithTheNewestNote(t *testing.T) { records := modelReadUsage(t, compactor.client.usage.path) require.Len(t, records, 3) + for n, record := range records { require.Equal(t, &[2]int{0, 0}, record.Node) require.Equal(t, len(long[n]), record.TextBytes) @@ -174,6 +175,7 @@ func TestCompactorRetriesALongLineWithTheNewestNote(t *testing.T) { first := compactorInputTexts(t, server.Call(0)) require.Len(t, first, 2) + for n, want := range []string{long[0], long[1]} { call := server.Call(n + 1) texts := compactorInputTexts(t, call) @@ -209,15 +211,16 @@ func TestCompactorFailsWithoutALine(t *testing.T) { }{ { name: "failed call", - reply: func(w http.ResponseWriter, call ModelFakeCall) { + reply: func(w http.ResponseWriter, _ ModelFakeCall) { w.WriteHeader(http.StatusInternalServerError) + _, _ = io.WriteString(w, "down") }, want: "HTTP 500", }, { name: "tool call and no text", - reply: func(w http.ResponseWriter, call ModelFakeCall) { + reply: func(w http.ResponseWriter, _ ModelFakeCall) { modelWriteEvents(w, modelCompletedEvent([]any{map[string]any{"type": "function_call", "name": "zoom", "arguments": "{}"}}, 100, 0, 5)) }, want: "no line", @@ -249,6 +252,7 @@ func TestCompactorBuildsNodesInTheCore(t *testing.T) { server.reply = func(w http.ResponseWriter, call ModelFakeCall) { if call.number == 0 { w.WriteHeader(http.StatusServiceUnavailable) + return } @@ -310,6 +314,7 @@ func compactorInputTexts(t *testing.T, call ModelFakeCall) []string { require.Equal(t, "user", message.Role) require.Len(t, message.Content, 1) require.Equal(t, "input_text", message.Content[0].Type) + texts[n] = message.Content[0].Text }