71 lines
2.2 KiB
Go
71 lines
2.2 KiB
Go
package kit
|
|
|
|
import "testing"
|
|
|
|
// EnsembleMinimal creates and starts an Ensemble with a single full node and a single miner.
|
|
// It does not interconnect nodes nor does it begin mining.
|
|
//
|
|
// This function supports passing both ensemble and node functional options.
|
|
// Functional options are applied to all nodes.
|
|
func EnsembleMinimal(t *testing.T, opts ...interface{}) (*TestFullNode, *TestMiner, *Ensemble) {
|
|
eopts, nopts := siftOptions(t, opts)
|
|
|
|
var (
|
|
full TestFullNode
|
|
miner TestMiner
|
|
)
|
|
ens := NewEnsemble(t, eopts...).FullNode(&full, nopts...).Miner(&miner, &full, nopts...).Start()
|
|
return &full, &miner, ens
|
|
}
|
|
|
|
// EnsembleTwoOne creates and starts an Ensemble with two full nodes and one miner.
|
|
// It does not interconnect nodes nor does it begin mining.
|
|
//
|
|
// This function supports passing both ensemble and node functional options.
|
|
// Functional options are applied to all nodes.
|
|
func EnsembleTwoOne(t *testing.T, opts ...interface{}) (*TestFullNode, *TestFullNode, *TestMiner, *Ensemble) {
|
|
eopts, nopts := siftOptions(t, opts)
|
|
|
|
var (
|
|
one, two TestFullNode
|
|
miner TestMiner
|
|
)
|
|
ens := NewEnsemble(t, eopts...).FullNode(&one, nopts...).FullNode(&two, nopts...).Miner(&miner, &one, nopts...).Start()
|
|
return &one, &two, &miner, ens
|
|
}
|
|
|
|
// EnsembleOneTwo creates and starts an Ensemble with one full node and two miners.
|
|
// It does not interconnect nodes nor does it begin mining.
|
|
//
|
|
// This function supports passing both ensemble and node functional options.
|
|
// Functional options are applied to all nodes.
|
|
func EnsembleOneTwo(t *testing.T, opts ...interface{}) (*TestFullNode, *TestMiner, *TestMiner, *Ensemble) {
|
|
eopts, nopts := siftOptions(t, opts)
|
|
|
|
var (
|
|
full TestFullNode
|
|
one, two TestMiner
|
|
)
|
|
ens := NewEnsemble(t, eopts...).
|
|
FullNode(&full, nopts...).
|
|
Miner(&one, &full, nopts...).
|
|
Miner(&two, &full, nopts...).
|
|
Start()
|
|
|
|
return &full, &one, &two, ens
|
|
}
|
|
|
|
func siftOptions(t *testing.T, opts []interface{}) (eopts []EnsembleOpt, nopts []NodeOpt) {
|
|
for _, v := range opts {
|
|
switch o := v.(type) {
|
|
case EnsembleOpt:
|
|
eopts = append(eopts, o)
|
|
case NodeOpt:
|
|
nopts = append(nopts, o)
|
|
default:
|
|
t.Fatalf("invalid option type: %T", o)
|
|
}
|
|
}
|
|
return eopts, nopts
|
|
}
|