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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
// Copyright (C) 2019-2023 Algorand, Inc.
// This file is part of go-algorand
//
// go-algorand is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// go-algorand is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with go-algorand. If not, see <https://www.gnu.org/licenses/>.
package sqlitedriver
import (
"context"
"crypto/rand"
"testing"
"github.com/algorand/go-algorand/crypto"
"github.com/algorand/go-algorand/data/basics"
storetesting "github.com/algorand/go-algorand/ledger/store/testing"
"github.com/algorand/go-algorand/ledger/store/trackerdb"
"github.com/algorand/go-algorand/test/partitiontest"
"github.com/stretchr/testify/require"
)
// Test functions operating on catchpointfirststageinfo table.
func TestCatchpointFirstStageInfoTable(t *testing.T) {
partitiontest.PartitionTest(t)
dbs, _ := storetesting.DbOpenTest(t, true)
defer dbs.Close()
ctx := context.Background()
err := accountsCreateCatchpointFirstStageInfoTable(ctx, dbs.Wdb.Handle)
require.NoError(t, err)
crw := NewCatchpointSQLReaderWriter(dbs.Wdb.Handle)
for _, round := range []basics.Round{4, 6, 8} {
info := trackerdb.CatchpointFirstStageInfo{
TotalAccounts: uint64(round) * 10,
}
err = crw.InsertOrReplaceCatchpointFirstStageInfo(ctx, round, &info)
require.NoError(t, err)
}
for _, round := range []basics.Round{4, 6, 8} {
info, exists, err := crw.SelectCatchpointFirstStageInfo(ctx, round)
require.NoError(t, err)
require.True(t, exists)
infoExpected := trackerdb.CatchpointFirstStageInfo{
TotalAccounts: uint64(round) * 10,
}
require.Equal(t, infoExpected, info)
}
_, exists, err := crw.SelectCatchpointFirstStageInfo(ctx, 7)
require.NoError(t, err)
require.False(t, exists)
rounds, err := crw.SelectOldCatchpointFirstStageInfoRounds(ctx, 6)
require.NoError(t, err)
require.Equal(t, []basics.Round{4, 6}, rounds)
err = crw.DeleteOldCatchpointFirstStageInfo(ctx, 6)
require.NoError(t, err)
rounds, err = crw.SelectOldCatchpointFirstStageInfoRounds(ctx, 9)
require.NoError(t, err)
require.Equal(t, []basics.Round{8}, rounds)
}
func TestUnfinishedCatchpointsTable(t *testing.T) {
partitiontest.PartitionTest(t)
dbs, _ := storetesting.DbOpenTest(t, true)
defer dbs.Close()
cts := NewCatchpointSQLReaderWriter(dbs.Wdb.Handle)
err := accountsCreateUnfinishedCatchpointsTable(
context.Background(), dbs.Wdb.Handle)
require.NoError(t, err)
var d3 crypto.Digest
rand.Read(d3[:])
err = cts.InsertUnfinishedCatchpoint(context.Background(), 3, d3)
require.NoError(t, err)
var d5 crypto.Digest
rand.Read(d5[:])
err = cts.InsertUnfinishedCatchpoint(context.Background(), 5, d5)
require.NoError(t, err)
ret, err := cts.SelectUnfinishedCatchpoints(context.Background())
require.NoError(t, err)
expected := []trackerdb.UnfinishedCatchpointRecord{
{
Round: 3,
BlockHash: d3,
},
{
Round: 5,
BlockHash: d5,
},
}
require.Equal(t, expected, ret)
err = cts.DeleteUnfinishedCatchpoint(context.Background(), 3)
require.NoError(t, err)
ret, err = cts.SelectUnfinishedCatchpoints(context.Background())
require.NoError(t, err)
expected = []trackerdb.UnfinishedCatchpointRecord{
{
Round: 5,
BlockHash: d5,
},
}
require.Equal(t, expected, ret)
}
|