X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/a3b9849a54a6cd8a4cc5b1ae03a39aab31322f7b..522d7fd117285e4e4ce3ea5ec9b5a1722521b681:/memory_test.go?ds=inline diff --git a/memory_test.go b/memory_test.go index 3930435..fe050c4 100644 --- a/memory_test.go +++ b/memory_test.go @@ -23,6 +23,7 @@ func TestMemoryRestartLoadsTheSameView(t *testing.T) { memory.chat.high = 200 memory.chat.low = 100 + memory.cview.high = 80 memory.cview.low = 40 for n := range 100 { @@ -89,6 +90,7 @@ func TestMemoryKeepsTheCompactionViewCoarser(t *testing.T) { memory.chat.high = 200 memory.chat.low = 100 + memory.cview.high = 80 memory.cview.low = 40 for n := range 64 { @@ -124,6 +126,7 @@ func TestMemoryRunsAtMostEightBuilds(t *testing.T) { } memory.mutex.Lock() + running := slices.Collect(maps.Keys(memory.queue.running)) memory.mutex.Unlock() slices.SortFunc(running, func(a, b NodeKey) int { @@ -179,6 +182,7 @@ func TestMemorySettleWaitsForThePendingLines(t *testing.T) { func TestMemorySettleFailsWhileABuildFailed(t *testing.T) { summarizer := newFakeSummarizer() summarizer.SetFailing(true) + logs := &SyncBuffer{} memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now, summarizer, log.New(logs, "", 0)) require.NoError(t, err) @@ -193,6 +197,7 @@ func TestMemorySettleFailsWhileABuildFailed(t *testing.T) { require.Equal(t, "summary failed, retry at next message: node (0, 0): fake failure\n", logs.String()) summarizer.SetFailing(false) + _, err = memory.Append("user", "retry") require.NoError(t, err) require.NoError(t, memory.Settle(t.Context())) @@ -220,6 +225,7 @@ func TestMemorySettleEndsWithItsContext(t *testing.T) { func TestMemoryStartsALeafAfterFewerThanEightUnbuilt(t *testing.T) { summarizer := newFakeSummarizer() summarizer.SetFailing(true) + memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now, summarizer, discardLog()) require.NoError(t, err) @@ -233,6 +239,7 @@ func TestMemoryStartsALeafAfterFewerThanEightUnbuilt(t *testing.T) { } require.NoError(t, memory.Wait(t.Context())) + for _, task := range summarizer.Tasks() { require.Less(t, task.Index, 8) } @@ -278,6 +285,7 @@ func TestMemoryRestartRestoresTheQueue(t *testing.T) { clock := newTestClock() summarizer := newFakeSummarizer() summarizer.SetFailing(true) + memory, err := openMemory(t.Context(), root, clock.Now, summarizer, discardLog()) require.NoError(t, err) @@ -320,6 +328,7 @@ func TestMemoryBuildsFreeParents(t *testing.T) { func TestMemoryKeepsALongMessageOutOfTheView(t *testing.T) { summarizer := newFakeSummarizer() summarizer.SetFailing(true) + memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now, summarizer, discardLog()) require.NoError(t, err)