dione/node/node_test.go

package node

import (
	"context"
	"fmt"
	"testing"
	"time"

	"github.com/Secured-Finance/dione/config"
	"github.com/sirupsen/logrus"
)

func TestConsensus(t *testing.T) {
	logrus.SetLevel(logrus.DebugLevel)
	//log.SetAllLoggers(log.LevelDebug)

	cfg := &config.Config{
		ListenPort: "1234",
		ListenAddr: "0.0.0.0",
		Rendezvous: "dione",
		PubSub: config.PubSubConfig{
			ProtocolID: "/test/1.0",
		},
	}

	//cfg.BootstrapNodes = "/dnsaddr/bootstrap.libp2p.io/p2p/QmNnooDu7bfjPFoTZYxMNLWUQJyrVwtbZg5gBMjTezGAJN"

	// setup first node
	node1 := newNode(cfg)

	// setup second node
	cfg.ListenPort = "1235"
	cfg.BootstrapNodes = []string{node1.Host.Addrs()[0].String() + fmt.Sprintf("/p2p/%s", node1.Host.ID().String())}
	node2 := newNode(cfg)

	// setup third node
	cfg.ListenPort = "1236"
	node3 := newNode(cfg)

	cfg.ListenPort = "1237"
	node4 := newNode(cfg)
	cfg.ListenPort = "1238"
	node5 := newNode(cfg)
	cfg.ListenPort = "1239"
	node6 := newNode(cfg)

	time.Sleep(10 * time.Second)
	go node2.ConsensusManager.NewTestConsensus("test", "123")
	go node1.ConsensusManager.NewTestConsensus("test1", "123")
	go node3.ConsensusManager.NewTestConsensus("test", "123")
	go node4.ConsensusManager.NewTestConsensus("test1", "123")
	go node5.ConsensusManager.NewTestConsensus("test", "123")
	go node6.ConsensusManager.NewTestConsensus("test2", "123")
	select {}
}

func newNode(cfg *config.Config) *Node {
	privKey, err := generatePrivateKey()
	if err != nil {
		logrus.Fatal(err)
	}
	ctx, ctxCancel := context.WithCancel(context.Background())

	node := &Node{
		OracleTopic:     "dione",
		Config:          cfg,
		GlobalCtx:       ctx,
		GlobalCtxCancel: ctxCancel,
	}
	node.setupNode(ctx, privKey, 1*time.Second, 3)
	return node
}