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
|
// Copyright (C) 2019-2021 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 gen
import (
"fmt"
"io/ioutil"
"os"
"path/filepath"
"strings"
"sync"
"testing"
"github.com/algorand/go-algorand/config"
"github.com/algorand/go-algorand/data/account"
"github.com/algorand/go-algorand/protocol"
"github.com/algorand/go-algorand/util/db"
"github.com/algorand/go-algorand/test/partitiontest"
"github.com/stretchr/testify/require"
)
func TestLoadMultiRootKeyConcurrent(t *testing.T) {
t.Skip() // skip in auto-test mode
a := require.New(t)
tempDir, err := ioutil.TempDir("", "loadkey-test-")
a.NoError(err)
defer os.RemoveAll(tempDir)
const numThreads = 100
var wg sync.WaitGroup
wg.Add(numThreads)
for i := 0; i < numThreads; i++ {
go func(idx int) {
defer wg.Done()
wallet := filepath.Join(tempDir, fmt.Sprintf("wallet%d", idx+1))
rootDB, err := db.MakeErasableAccessor(wallet)
defer rootDB.Close()
a.NoError(err)
_, err = account.GenerateRoot(rootDB)
a.NoError(err)
}(i)
}
wg.Wait()
for r := 0; r < 1000; r++ {
var wg sync.WaitGroup
wg.Add(numThreads)
for i := 0; i < numThreads; i++ {
go func(idx int) {
defer wg.Done()
wallet := filepath.Join(tempDir, fmt.Sprintf("wallet%d", idx+1))
_, db, err := loadRootKey(wallet)
a.NoError(err)
db.Close()
}(i)
}
wg.Wait()
}
}
func TestLoadSingleRootKeyConcurrent(t *testing.T) {
t.Skip() // skip in auto-test mode
a := require.New(t)
tempDir, err := ioutil.TempDir("", "loadkey-test-")
a.NoError(err)
defer os.RemoveAll(tempDir)
wallet := filepath.Join(tempDir, "wallet1")
rootDB, err := db.MakeErasableAccessor(wallet)
a.NoError(err)
_, err = account.GenerateRoot(rootDB)
rootDB.Close()
a.NoError(err)
const numThreads = 10000
var wg sync.WaitGroup
wg.Add(numThreads)
for i := 0; i < numThreads; i++ {
go func(idx int) {
defer wg.Done()
wallet := filepath.Join(tempDir, "wallet1")
_, db, err := loadRootKey(wallet)
a.NoError(err)
db.Close()
}(i)
}
wg.Wait()
}
func TestGenesisRoundoff(t *testing.T) {
partitiontest.PartitionTest(t)
verbosity := strings.Builder{}
genesisData := DefaultGenesis
genesisData.NetworkName = "wat"
genesisData.ConsensusProtocol = protocol.ConsensusCurrentVersion // TODO: also check ConsensusFuture ?
genesisData.Wallets = make([]WalletData, 15)
for i := range genesisData.Wallets {
genesisData.Wallets[i].Name = fmt.Sprintf("w%d", i)
genesisData.Wallets[i].Stake = 100.0 / float64(len(genesisData.Wallets))
}
_, _, _, err := setupGenerateGenesisFiles(&genesisData, config.Consensus, &verbosity)
require.NoError(t, err)
require.True(t, strings.Contains(verbosity.String(), "roundoff"))
}
|
