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         for n := range 100 {
  21                 _, err = memory.Append("user", fmt.Sprint(n))
  22                 require.NoError(t, err)
  23         }
  24
  25         view := memory.View()
  26         saved, err := os.ReadFile(filepath.Join(root, "view.json"))
  27         require.NoError(t, err)
  28         require.Less(t, len(memory.chat.lines), 100)
  29         require.NoError(t, memory.Close())
  30
  31         memory, err = openMemory(t.Context(), root, clock.Now)
  32         require.NoError(t, err)
  33
  34         defer func() {
  35                 require.NoError(t, memory.Close())
  36         }()
  37
  38         require.Equal(t, view, memory.View())
  39
  40         loaded, err := os.ReadFile(filepath.Join(root, "view.json"))
  41         require.NoError(t, err)
  42         require.Equal(t, saved, loaded)
  43 }
  44
  45 func TestMemoryBuildsFreeParents(t *testing.T) {
  46         memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now)
  47         require.NoError(t, err)
  48
  49         defer func() {
  50                 require.NoError(t, memory.Close())
  51         }()
  52
  53         for _, text := range []string{"a", "b", "c", "d"} {
  54                 _, err = memory.Append("user", text)
  55                 require.NoError(t, err)
  56         }
  57
  58         node, built := memory.tree.Node(2, 0)
  59         require.True(t, built)
  60         require.Equal(t, "user: a\nuser: b\nuser: c\nuser: d", node.Text)
  61 }
  62
  63 func TestMemoryKeepsALongMessageOutOfTheView(t *testing.T) {
  64         memory, err := openMemory(t.Context(), t.TempDir(), newTestClock().Now)
  65         require.NoError(t, err)
  66
  67         defer func() {
  68                 require.NoError(t, memory.Close())
  69         }()
  70
  71         _, err = memory.Append("echo", string(make([]byte, nodeBytes)))
  72         require.NoError(t, err)
  73
  74         require.Equal(t, "<chat>\n0+1|"+pending+"\n</chat>", memory.View())
  75 }
  76
  77 func TestMemoryCoversMessagesAfterTheSavedView(t *testing.T) {
  78         root := t.TempDir()
  79         clock := newTestClock()
  80         memory, err := openMemory(t.Context(), root, clock.Now)
  81         require.NoError(t, err)
  82
  83         _, err = memory.Append("user", "a")
  84         require.NoError(t, err)
  85
  86         _, err = memory.log.Append("reply", "b")
  87         require.NoError(t, err)
  88         require.NoError(t, memory.Close())
  89
  90         memory, err = openMemory(t.Context(), root, clock.Now)
  91         require.NoError(t, err)
  92
  93         defer func() {
  94                 require.NoError(t, memory.Close())
  95         }()
  96
  97         require.Equal(t, "<chat>\n0+1|user: a\n1+1|reply: b\n</chat>", memory.View())
  98 }
  99
 100 func TestMemoryRefusesToRebuildAMissingView(t *testing.T) {
 101         root := t.TempDir()
 102         clock := newTestClock()
 103         memory, err := openMemory(t.Context(), root, clock.Now)
 104         require.NoError(t, err)
 105
 106         _, err = memory.Append("user", "a")
 107         require.NoError(t, err)
 108         require.NoError(t, memory.Close())
 109         require.NoError(t, os.Remove(filepath.Join(root, "view.json")))
 110
 111         _, err = openMemory(t.Context(), root, clock.Now)
 112         require.ErrorIs(t, err, errNoView)
 113 }