Back up the daemon and restore its data

[onechat] / replay_test.go

diff --git a/replay_test.go b/replay_test.go
index 8dd54ab5b0f07a541edc87fefd070108864fc0e0..e6b8daa845e31a5bca9f6d967c33154757271de6 100644 (file)
--- a/replay_test.go
+++ b/replay_test.go
@@ -13,20 +13,32 @@ import (
        "github.com/stretchr/testify/require"
 )
 
        "github.com/stretchr/testify/require"
 )
 

-// replayMessages is the length of the synthetic log.
-const replayMessages = 100000
+const (
+       // replayMessages is the length of the synthetic log.
+       replayMessages = 100000
+
+       // replayShortMessages is the length of the synthetic log with -short.
+       // It is long enough for batches of the chat view.
+       replayShortMessages = 5000
+)

 
 // TestReplaySyntheticLog runs a synthetic log through the memory core with the
 // fake summarizer. After each message it waits for the builds, as a turn does.
 // It checks that each view stays within its sawtooth, that no view line holds a
 // whole long message or a part of an unbuilt one, and that a restart in the
 
 // TestReplaySyntheticLog runs a synthetic log through the memory core with the
 // fake summarizer. After each message it waits for the builds, as a turn does.
 // It checks that each view stays within its sawtooth, that no view line holds a
 // whole long message or a part of an unbuilt one, and that a restart in the

-// middle and at the end loads the same views byte for byte.
+// middle and at the end loads the same views byte for byte. With -short, the
+// log is shorter, and the test still requires a batch of the chat view.

 func TestReplaySyntheticLog(t *testing.T) {
        start := time.Now()
 func TestReplaySyntheticLog(t *testing.T) {
        start := time.Now()

+       messages := replayMessages
+       if testing.Short() {
+               messages = replayShortMessages
+       }
+

        root := t.TempDir()
        clock := newTestClock()
        random := rand.New(rand.NewPCG(1, 2))
        root := t.TempDir()
        clock := newTestClock()
        random := rand.New(rand.NewPCG(1, 2))

-       memory, err := openMemory(t.Context(), root, clock.Now, newFakeSummarizer())
+       memory, err := openMemory(t.Context(), root, clock.Now, newFakeSummarizer(), discardLog())

        require.NoError(t, err)
 
        defer func() {
        require.NoError(t, err)
 
        defer func() {


@@ -34,10 +46,11 @@ func TestReplaySyntheticLog(t *testing.T) {
        }()
 
        batches := 0
        }()
 
        batches := 0

+

        low := viewChatHigh
        low := viewChatHigh

-       long := make([]bool, replayMessages)
-       for id := range replayMessages {
-               if id == replayMessages/2 {
+       long := make([]bool, messages)
+       for id := range messages {
+               if id == messages/2 {

                        memory = replayRequireRestart(t, memory, root, clock)
                }
 
                        memory = replayRequireRestart(t, memory, root, clock)
                }
 


@@ -58,17 +71,18 @@ func TestReplaySyntheticLog(t *testing.T) {
        }
 
        memory = replayRequireRestart(t, memory, root, clock)
        }
 
        memory = replayRequireRestart(t, memory, root, clock)

-       require.Equal(t, replayMessages, memory.log.Len())
-       require.Greater(t, batches, 0)
+       require.Equal(t, messages, memory.log.Len())
+       require.Positive(t, batches)

 
        require.NoError(t, memory.Wait(t.Context()))
 
        require.NoError(t, memory.Wait(t.Context()))

-       t.Logf("%d messages, %d nodes, %d batches in %s", replayMessages, memory.tree.Count(), batches, time.Since(start).Round(time.Second))
+       t.Logf("%d messages, %d nodes, %d batches in %s", messages, memory.tree.Count(), batches, time.Since(start).Round(time.Second))

        t.Logf("chat view: %d lines, %d bytes, lowest after a batch %d bytes", len(memory.chat.lines), memory.chat.Size(), low)
        t.Logf("compaction view: %d lines, %d bytes", len(memory.cview.lines), memory.cview.Size())
 
        runtime.GC()
 
        var stats runtime.MemStats
        t.Logf("chat view: %d lines, %d bytes, lowest after a batch %d bytes", len(memory.chat.lines), memory.chat.Size(), low)
        t.Logf("compaction view: %d lines, %d bytes", len(memory.cview.lines), memory.cview.Size())
 
        runtime.GC()
 
        var stats runtime.MemStats

+

        runtime.ReadMemStats(&stats)
        t.Logf("heap in use: %d KB", stats.HeapInuse/1024)
 }
        runtime.ReadMemStats(&stats)
        t.Logf("heap in use: %d KB", stats.HeapInuse/1024)
 }


@@ -86,7 +100,7 @@ func replayRequireRestart(t *testing.T, memory *Memory, root string, clock *Test
        require.NoError(t, err)
        require.NoError(t, memory.Close())
 
        require.NoError(t, err)
        require.NoError(t, memory.Close())
 

-       memory, err = openMemory(t.Context(), root, clock.Now, newFakeSummarizer())
+       memory, err = openMemory(t.Context(), root, clock.Now, newFakeSummarizer(), discardLog())

        require.NoError(t, err)
        require.NoError(t, memory.Wait(t.Context()))
        require.Equal(t, view, memory.View())
        require.NoError(t, err)
        require.NoError(t, memory.Wait(t.Context()))
        require.Equal(t, view, memory.View())


@@ -198,6 +212,7 @@ func replayRequireOnlySummaries(t *testing.T, memory *Memory, long []bool, full
                }
 
                require.Equal(t, head+strings.ReplaceAll(node.Text, "\n", " "), rendered[n+1])
                }
 
                require.Equal(t, head+strings.ReplaceAll(node.Text, "\n", " "), rendered[n+1])

+

                if line.level == 0 && long[line.index] {
                        require.True(t, strings.HasPrefix(node.Text, "summary "), "message %d", line.index)
                }
                if line.level == 0 && long[line.index] {
                        require.True(t, strings.HasPrefix(node.Text, "summary "), "message %d", line.index)
                }