Function TestLogGrow

raftwal/encryption_test.go:110–144

TestLogGrow writes data of sufficient size to grow the log file.

(t *testing.T)

// TestLogGrow writes data of sufficient size to grow the log file.
func TestLogGrow(t *testing.T) {
	test := func(t *testing.T, key []byte) {
		dir := t.TempDir()
		ds, err := InitEncrypted(dir, key)
		require.NoError(t, err)

		const numEntries = (maxNumEntries * 3) / 2

		// 5KB * 30000 is ~ 150MB, this will cause the log file to grow.
		var entries []raftpb.Entry
		for i := range numEntries {
			data := make([]byte, 5<<10)
			rand.Read(data)
			entry := raftpb.Entry{Index: uint64(i + 1), Term: 1, Data: data}
			entries = append(entries, entry)
		}
		require.NoError(t, ds.wal.AddEntries(entries))

		// Reopen the file and retrieve all entries.
		ds, err = InitEncrypted(dir, key)
		require.NoError(t, err)
		readEntries := ds.wal.allEntries(0, math.MaxInt64, math.MaxInt64)
		require.Equal(t, numEntries, len(readEntries))

		for i, gotEntry := range readEntries {
			expEntry := entries[i]
			require.Equal(t, expEntry.Data, gotEntry.Data)
			require.Equal(t, expEntry.Index, gotEntry.Index)
			require.Equal(t, expEntry.Term, gotEntry.Term)
			require.Equal(t, expEntry.Type, gotEntry.Type)
		}
	}
	t.Run("without encryption", func(t *testing.T) { test(t, nil) })
	t.Run("with encryption", func(t *testing.T) { test(t, []byte("badger16byteskey")) })
}