/
ibft.go
111 lines (89 loc) · 2.17 KB
/
ibft.go
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
package framework
import (
"context"
"fmt"
"os"
"testing"
"github.com/Coinmaxify/Coinmaxify/types"
)
type IBFTServersManager struct {
t *testing.T
servers []*TestServer
}
type IBFTServerConfigCallback func(index int, config *TestServerConfig)
func NewIBFTServersManager(
t *testing.T,
numNodes int,
ibftDirPrefix string,
callback IBFTServerConfigCallback,
) *IBFTServersManager {
t.Helper()
dataDir, err := tempDir()
if err != nil {
t.Fatal(err)
}
srvs := make([]*TestServer, 0, numNodes)
t.Cleanup(func() {
for _, s := range srvs {
s.Stop()
}
if err := os.RemoveAll(dataDir); err != nil {
t.Log(err)
}
})
bootnodes := make([]string, 0, numNodes)
genesisValidators := make([]string, 0, numNodes)
for i := 0; i < numNodes; i++ {
srv := NewTestServer(t, dataDir, func(config *TestServerConfig) {
config.SetConsensus(ConsensusIBFT)
config.SetIBFTDirPrefix(ibftDirPrefix)
config.SetIBFTDir(fmt.Sprintf("%s%d", ibftDirPrefix, i))
callback(i, config)
})
res, err := srv.SecretsInit()
if err != nil {
t.Fatal(err)
}
libp2pAddr := ToLocalIPv4LibP2pAddr(srv.Config.LibP2PPort, res.NodeID)
srvs = append(srvs, srv)
bootnodes = append(bootnodes, libp2pAddr)
genesisValidators = append(genesisValidators, res.Address)
}
srv := srvs[0]
srv.Config.SetBootnodes(bootnodes)
// Set genesis staking balance for genesis validators
for i, v := range genesisValidators {
addr := types.StringToAddress(v)
conf := srvs[i].Config
if conf.GenesisValidatorBalance != nil {
srv.Config.Premine(addr, conf.GenesisValidatorBalance)
}
}
if err := srv.GenerateGenesis(); err != nil {
t.Fatal(err)
}
return &IBFTServersManager{t, srvs}
}
func (m *IBFTServersManager) StartServers(ctx context.Context) {
for _, srv := range m.servers {
if err := srv.Start(ctx); err != nil {
m.t.Fatal(err)
}
}
for _, srv := range m.servers {
if err := srv.WaitForReady(ctx); err != nil {
m.t.Fatal(err)
}
}
}
func (m *IBFTServersManager) StopServers() {
for _, srv := range m.servers {
srv.Stop()
}
}
func (m *IBFTServersManager) GetServer(i int) *TestServer {
if i >= len(m.servers) {
return nil
}
return m.servers[i]
}