git.giacomolli.uk - onechat/blob - memory_test.go

   1 package main
   2
   3 import (
   4         "fmt"
   5         "os"
   6         "path/filepath"
   7         "testing"
   8
   9         "github.com/stretchr/testify/require"
  10 )
  11
  12 func TestMemoryRestartLoadsTheSameView(t *testing.T) {
  13         root := t.TempDir()
  14         clock := newTestClock()
  15         memory, err := openMemory(t.Context(), root, clock.Now)
  16         require.NoError(t, err)
  17
  18         memory.chat.high = 200
  19         memory.chat.low = 100
  20         memory.cview.high = 80
  21         memory.cview.low = 40
  22         for n := range 100 {
  23                 _, err = memory.Append("user", fmt.Sprint(n))
  24                 require.NoError(t, err)
  25         }
  26
  27         view := memory.View()
  28         context := memory.Context(0, 100)
  29         saved, err := os.ReadFile(filepath.Join(root, "view.json"))
  30         require.NoError(t, err)
  31         require.Less(t, len(memory.chat.lines), 100)
  32         require.NoError(t, memory.Close())
  33
  34         memory, err = openMemory(t.Context(), root, clock.Now)
  35         require.NoError(t, err)
  36
  37         defer func() {
  38                 require.NoError(t, memory.Close())
  39         }()
  40
  41         require.Equal(t, view, memory.View())
  42         require.Equal(t, context, memory.Context(0, 100))
  43
  44         loaded, err := os.ReadFile(filepath.Join(root, "view.json"))
  45         require.NoError(t, err)
  46         require.Equal(t, saved, loaded)
  47 }
  48
  49 func TestMemoryKeepsTheCompactionViewCoarser(t *testing.T) {
  50         memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now)
  51         require.NoError(t, err)
  52
  53         defer func() {
  54                 require.NoError(t, memory.Close())
  55         }()
  56
  57         memory.chat.high = 200
  58         memory.chat.low = 100
  59         memory.cview.high = 80
  60         memory.cview.low = 40
  61         for n := range 64 {
  62                 _, err = memory.Append("user", fmt.Sprint(n%10))
  63                 require.NoError(t, err)
  64
  65                 coarse := 0
  66                 for _, line := range memory.chat.lines {
  67                         outer := memory.cview.lines[coarse]
  68                         for nodeEnd(outer.level, outer.index) <= line.index<<line.level {
  69                                 coarse++
  70                                 outer = memory.cview.lines[coarse]
  71                         }
  72
  73                         require.GreaterOrEqual(t, outer.level, line.level, "message %d", n)
  74                 }
  75         }
  76 }
  77
  78 func TestMemoryBuildsFreeParents(t *testing.T) {
  79         memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now)
  80         require.NoError(t, err)
  81
  82         defer func() {
  83                 require.NoError(t, memory.Close())
  84         }()
  85
  86         for _, text := range []string{"a", "b", "c", "d"} {
  87                 _, err = memory.Append("user", text)
  88                 require.NoError(t, err)
  89         }
  90
  91         node, built := memory.tree.Node(2, 0)
  92         require.True(t, built)
  93         require.Equal(t, "user: a\nuser: b\nuser: c\nuser: d", node.Text)
  94 }
  95
  96 func TestMemoryKeepsALongMessageOutOfTheView(t *testing.T) {
  97         memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now)
  98         require.NoError(t, err)
  99
 100         defer func() {
 101                 require.NoError(t, memory.Close())
 102         }()
 103
 104         _, err = memory.Append("echo", string(make([]byte, nodeBytes)))
 105         require.NoError(t, err)
 106
 107         require.Equal(t, "<chat>\n0+1|"+pending+"\n</chat>", memory.View())
 108 }
 109
 110 func TestMemoryCoversMessagesAfterTheSavedView(t *testing.T) {
 111         root := t.TempDir()
 112         clock := newTestClock()
 113         memory, err := openMemory(t.Context(), root, clock.Now)
 114         require.NoError(t, err)
 115
 116         _, err = memory.Append("user", "a")
 117         require.NoError(t, err)
 118
 119         _, err = memory.log.Append("reply", "b")
 120         require.NoError(t, err)
 121         require.NoError(t, memory.Close())
 122
 123         memory, err = openMemory(t.Context(), root, clock.Now)
 124         require.NoError(t, err)
 125
 126         defer func() {
 127                 require.NoError(t, memory.Close())
 128         }()
 129
 130         require.Equal(t, "<chat>\n0+1|user: a\n1+1|reply: b\n</chat>", memory.View())
 131 }
 132
 133 func TestMemoryRefusesToRebuildAMissingView(t *testing.T) {
 134         root := t.TempDir()
 135         clock := newTestClock()
 136         memory, err := openMemory(t.Context(), root, clock.Now)
 137         require.NoError(t, err)
 138
 139         _, err = memory.Append("user", "a")
 140         require.NoError(t, err)
 141         require.NoError(t, memory.Close())
 142         require.NoError(t, os.Remove(filepath.Join(root, "view.json")))
 143
 144         _, err = openMemory(t.Context(), root, clock.Now)
 145         require.ErrorIs(t, err, errNoView)
 146 }