1package vt
2
3import (
4 "testing"
5
6 renderer "github.com/bnema/vev-vt/core"
7 "github.com/stretchr/testify/require"
8)
9
10func TestHistoryScalarReadsAndCopyDoNotAllocate(t *testing.T) {
11 h := NewHistory(HistoryConfig{MaxRows: 4, ChunkRows: 1})
12 require.NoError(t, h.Append(historyRow("abc"), LineBound{End: 3}))
13 require.NoError(t, h.Append(historyRow("de"), LineBound{End: 2}))
14 view := h.View()
15 require.Equal(t, 3, view.RowWidth(0))
16 require.Equal(t, 2, view.RowWidth(1))
17 for _, y := range []int{-1, 2} {
18 require.Zero(t, view.RowWidth(y))
19 require.Equal(t, renderer.BlankCell(), view.Cell(0, y))
20 }
21 require.Equal(t, renderer.BlankCell(), view.Cell(-1, 0))
22 require.Equal(t, renderer.BlankCell(), view.Cell(3, 0))
23 dst := make([]renderer.Cell, 2)
24 require.Equal(t, 2, view.CopyRow(0, dst))
25 require.Equal(t, 'a', dst[0].Rune)
26 dst[0].Rune = 'X'
27 require.Equal(t, 'a', view.Cell(0, 0).Rune)
28 require.Zero(t, testing.AllocsPerRun(100, func() {
29 _ = view.RowWidth(1)
30 _ = view.Cell(1, 1)
31 view.CopyRow(0, dst)
32 }))
33}
34
35func TestSnapshotTailReusesItsCompactCapture(t *testing.T) {
36 h := NewHistory(HistoryConfig{MaxRows: 4, ChunkRows: 4})
37 require.NoError(t, h.Append(historyRow("ab"), LineBound{End: 2}))
38 snapshot := h.SnapshotView()
39 tail := snapshot.Tail()
40 require.Same(t, tail.Chunk(0), snapshot.Tail().Chunk(0))
41 require.Zero(t, testing.AllocsPerRun(100, func() { _ = snapshot.Tail() }))
42 require.NoError(t, h.Append(historyRow("cd"), LineBound{End: 2}))
43 require.Equal(t, 1, tail.Len())
44 require.Equal(t, 'a', tail.Cell(0, 0).Rune)
45 require.NoError(t, tail.Chunk(0).CheckInvariants())
46}
47
48func BenchmarkRecoveryTranscriptCompactCapture(b *testing.B) {
49 s := NewScreen(120, 40)
50 for y := range s.Rows() {
51 s.frame.Set(0, y, renderer.Cell{Rune: 'x', Style: renderer.DefaultStyle()})
52 s.buffer.boundaries[y] = LineBound{End: 1}
53 }
54 b.ReportAllocs()
55 for b.Loop() {
56 recoveryTranscriptSnapshotSink = s.RecoveryTranscriptSnapshot()
57 }
58}