woop/node/node_test.go

package node

import (
	"fmt"
	"github.com/harmony-one/harmony/utils"
	"os"
	"testing"
	"time"

	"github.com/harmony-one/harmony/crypto"
	"github.com/harmony-one/harmony/crypto/pki"

	"github.com/harmony-one/harmony/consensus"
	"github.com/harmony-one/harmony/p2p"

	proto_node "github.com/harmony-one/harmony/proto/node"
)

func TestNewNewNode(test *testing.T) {
	_, pubKey := utils.GenKey("1", "2")
	leader := p2p.Peer{IP: "1", Port: "2", PubKey: pubKey}
	validator := p2p.Peer{IP: "3", Port: "5"}
	consensus := consensus.New(leader, "0", []p2p.Peer{leader, validator}, leader)

	node := New(consensus, nil, leader)
	if node.Consensus == nil {
		test.Error("Consensus is not initialized for the node")
	}

	if node.blockchain == nil {
		test.Error("Blockchain is not initialized for the node")
	}

	if len(node.blockchain.Blocks) != 1 {
		test.Error("Genesis block is not initialized for the node")
	}

	if len(node.blockchain.Blocks[0].Transactions) != 1 {
		test.Error("Coinbase TX is not initialized for the node")
	}

	if node.UtxoPool == nil {
		test.Error("Utxo pool is not initialized for the node")
	}
}

func TestCountNumTransactionsInBlockchain(test *testing.T) {
	_, pubKey := utils.GenKey("1", "2")
	leader := p2p.Peer{IP: "1", Port: "2", PubKey: pubKey}
	validator := p2p.Peer{IP: "3", Port: "5"}
	consensus := consensus.New(leader, "0", []p2p.Peer{leader, validator}, leader)

	node := New(consensus, nil, leader)
	node.AddTestingAddresses(1000)
	if node.countNumTransactionsInBlockchain() != 1001 {
		test.Error("Count of transactions in the blockchain is incorrect")
	}
}

func TestAddPeers(test *testing.T) {
	priKey1 := crypto.Ed25519Curve.Scalar().SetInt64(int64(333))
	pubKey1 := pki.GetPublicKeyFromScalar(priKey1)

	priKey2 := crypto.Ed25519Curve.Scalar().SetInt64(int64(999))
	pubKey2 := pki.GetPublicKeyFromScalar(priKey2)

	peers1 := []*p2p.Peer{
		&p2p.Peer{
			IP:          "127.0.0.1",
			Port:        "8888",
			PubKey:      pubKey1,
			Ready:       true,
			ValidatorID: 1,
		},
		&p2p.Peer{
			IP:          "127.0.0.1",
			Port:        "9999",
			PubKey:      pubKey2,
			Ready:       false,
			ValidatorID: 2,
		},
	}
	_, pubKey := utils.GenKey("1", "2")
	leader := p2p.Peer{IP: "1", Port: "2", PubKey: pubKey}
	validator := p2p.Peer{IP: "3", Port: "5"}
	consensus := consensus.New(leader, "0", []p2p.Peer{leader, validator}, leader)

	node := New(consensus, nil, leader)
	r1 := node.AddPeers(peers1)
	e1 := 2
	if r1 != e1 {
		test.Errorf("Add %v peers, expectd %v", r1, e1)
	}
	r2 := node.AddPeers(peers1)
	e2 := 0
	if r2 != e2 {
		test.Errorf("Add %v peers, expectd %v", r2, e2)
	}
}

func sendPingMessage(leader p2p.Peer) {
	priKey1 := crypto.Ed25519Curve.Scalar().SetInt64(int64(333))
	pubKey1 := pki.GetPublicKeyFromScalar(priKey1)

	p1 := p2p.Peer{
		IP:     "127.0.0.1",
		Port:   "9999",
		PubKey: pubKey1,
	}

	ping1 := proto_node.NewPingMessage(p1)
	buf1 := ping1.ConstructPingMessage()

	fmt.Println("waiting for 5 seconds ...")
	time.Sleep(5 * time.Second)

	p2p.SendMessage(leader, buf1)
	fmt.Println("sent ping message ...")
}

func sendPongMessage(leader p2p.Peer) {
	priKey1 := crypto.Ed25519Curve.Scalar().SetInt64(int64(333))
	pubKey1 := pki.GetPublicKeyFromScalar(priKey1)
	p1 := p2p.Peer{
		IP:     "127.0.0.1",
		Port:   "9998",
		PubKey: pubKey1,
	}
	priKey2 := crypto.Ed25519Curve.Scalar().SetInt64(int64(999))
	pubKey2 := pki.GetPublicKeyFromScalar(priKey2)
	p2 := p2p.Peer{
		IP:     "127.0.0.1",
		Port:   "9999",
		PubKey: pubKey2,
	}

	pong1 := proto_node.NewPongMessage([]p2p.Peer{p1, p2}, nil)
	buf1 := pong1.ConstructPongMessage()

	fmt.Println("waiting for 10 seconds ...")
	time.Sleep(10 * time.Second)

	p2p.SendMessage(leader, buf1)
	fmt.Println("sent pong message ...")
}

func exitServer() {
	fmt.Println("wait 15 seconds to terminate the process ...")
	time.Sleep(15 * time.Second)

	os.Exit(0)
}

func TestPingPongHandler(test *testing.T) {
	leader := p2p.Peer{IP: "127.0.0.1", Port: "8881"}
	//   validator := p2p.Peer{IP: "127.0.0.1", Port: "9991"}
	consensus := consensus.New(leader, "0", []p2p.Peer{leader}, leader)
	node := New(consensus, nil, leader)
	//go sendPingMessage(leader)
	go sendPongMessage(leader)
	go exitServer()
	node.StartServer("8881")
}