-
Notifications
You must be signed in to change notification settings - Fork 8
/
boostrap_added_nodes.go
37 lines (30 loc) · 1.08 KB
/
boostrap_added_nodes.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
// (c) 2021, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.
package testhelpers
import (
"fmt"
"github.com/ava-labs/avalanchego-kurtosis/kurtosis/avalanche/libs/builder/networkbuilder"
"github.com/ava-labs/avalanchego-kurtosis/kurtosis/kurtosis/networksavalanche"
"github.com/kurtosis-tech/kurtosis-libs/golang/lib/networks"
"github.com/palantir/stacktrace"
"github.com/sirupsen/logrus"
)
func BootstrapAddedNodes(
network networks.Network,
definedNetwork *networkbuilder.Network,
avalancheImage string,
numNodes int) {
logrus.Infof("Adding %d additional nodes and waiting for them to bootstrap...", numNodes)
avalancheNetwork := networksavalanche.Cast(network)
for i := 1; i <= numNodes; i++ {
node := networkbuilder.NewNode(fmt.Sprintf("newNode-%d", i)).
Image(avalancheImage).
IsStaking(true)
definedNetwork.AddNode(node)
_, err := avalancheNetwork.CreateNode(definedNetwork, node)
if err != nil {
panic(stacktrace.Propagate(err, "Unable to create node %s", node.ID))
}
logrus.Infof("%s finished bootstrapping.", node.ID)
}
}