sokv/storage_test.go at main · nyan233/sokv · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package sokv

import (
	"github.com/stretchr/testify/require"
	"path"
	"testing"
)

func TestStorage(t *testing.T) {
	initTest(t)
	s := newPageStorage(&pageStorageOption{
		DataPath:             path.Join("testdata", "test.storage.dat"),
		FreelistPath:         path.Join("testdata", "test.storage.dat.freelist"),
		MaxCacheSize:         1024 * 1024,
		FreelistMaxCacheSize: 1024 * 1024,
		PageCipher:           nil,
	})
	require.NoError(t, s.init())
	readTxHeader := newTxHeader()
	readTxHeader.seq = 0
	readTxHeader.isRead = true
	readTxHeader.isRollback = false
	readTxHeader.isCommit = false

	writeTxHeader := newTxHeader()
	writeTxHeader.seq = 1
	writeTxHeader.isRead = false
	writeTxHeader.isRollback = false
	writeTxHeader.isCommit = false
	s.cache.createShadowPage(writeTxHeader)

	p, err := s.readPage(writeTxHeader, createPageIdFromUint64(3))
	require.NoError(t, err)
	require.Equal(t, p.idOfU8(), uint64(0))
	p.Header.Header = pageHeaderDat
	p.Header.PgId = createPageIdFromUint64(3)
	err = s.modifyPage(writeTxHeader, p, func(p2 pageDesc) (pageDesc, error) {
		p2.Data[0] = 1
		p2.Data[1] = 2
		p2.Data[2] = 3
		p2.Data[3] = 4
		return p2, nil
	})
	require.NoError(t, err)
	readP, err := s.readPage(readTxHeader, createPageIdFromUint64(3))
	require.NoError(t, err)
	require.Equal(t, int(readP.Data[0]), 0)
	require.Equal(t, int(readP.Data[1]), 0)
	require.Equal(t, int(readP.Data[2]), 0)
	require.Equal(t, int(readP.Data[3]), 0)
	writeP, err := s.readPage(writeTxHeader, createPageIdFromUint64(3))
	require.NoError(t, err)
	require.Equal(t, int(writeP.Data[0]), 1)
	require.Equal(t, int(writeP.Data[1]), 2)
	require.Equal(t, int(writeP.Data[2]), 3)
	require.Equal(t, int(writeP.Data[3]), 4)
}