X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/30fbeb13a74942c6d515cc2a51eab7407a78c325..d20d4c6b2aa7bceddced27be9b24ed171f38696f:/compactor_test.go?ds=inline diff --git a/compactor_test.go b/compactor_test.go index 7813c77..35e09eb 100644 --- a/compactor_test.go +++ b/compactor_test.go @@ -72,11 +72,19 @@ func TestCompactorRetriesALongLineWithTheNewestNote(t *testing.T) { writeEvents(w, completedEvent([]any{messageItem(long[call.number])}, 100, 0, 5)) } - line, err := newTestCompactor(t, server).Summarize(t.Context(), leafTask()) + compactor := newTestCompactor(t, server) + line, err := compactor.Summarize(t.Context(), leafTask()) require.NoError(t, err) require.Equal(t, long[2], line) require.Equal(t, 3, server.Calls()) + records := readUsage(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) + } + first := inputParts(t, server.Call(0)) require.Len(t, first, 2) for n, want := range []string{long[0], long[1]} { @@ -85,7 +93,7 @@ func TestCompactorRetriesALongLineWithTheNewestNote(t *testing.T) { require.Equal(t, first, parts[:2]) require.Equal(t, []string{tooLong(want)}, parts[2:]) require.Contains(t, parts[2], "was "+strconv.Itoa(len(want))+" bytes,") - require.Contains(t, parts[2], "\n"+want[:512]+"| ← LIMIT") + require.Contains(t, parts[2], "\n"+want[:512]+"| <- LIMIT") note, err := json.Marshal(map[string]string{ "type": "input_text",